diff --git a/mlem/contrib/heroku/meta.py b/mlem/contrib/heroku/meta.py index 91e3d488..df86ab2c 100644 --- a/mlem/contrib/heroku/meta.py +++ b/mlem/contrib/heroku/meta.py @@ -73,6 +73,11 @@ def _get_client(self, state: HerokuState) -> Client: host=urlparse(state.ensured_app.web_url).netloc, port=80 ) + def _get_client(self, state: HerokuState) -> Client: + return HTTPClient( + host=urlparse(state.ensured_app.web_url).netloc, port=80 + ) + class HerokuEnv(MlemEnv[HerokuDeployment]): """Heroku Account""" diff --git a/tests/cli/test_deployment.py b/tests/cli/test_deployment.py index 587041e3..44d28c1c 100644 --- a/tests/cli/test_deployment.py +++ b/tests/cli/test_deployment.py @@ -43,6 +43,9 @@ class Config: def _get_client(self, state) -> Client: return HTTPClient(host="", port=None) + def _get_client(self, state) -> Client: + return HTTPClient(host="", port=None) + class MlemEnvMock(MlemEnv): """mock"""