Sagemaker deployments (#366)

* WIP * its alive (kinda) * it works but it's ugly * little less ugly * lil fix * fix lint * fix lint * fix tests * fix tests * fix windows bugs * fix tests * fix tests
iterative · Sep 14, 2022 · d836ba4 · d836ba4
1 parent 3c5dd97
commit d836ba4
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git 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
@@ -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"""