mypy

cleanlab · Jun 17, 2024 · 7fdb8d4 · 7fdb8d4
1 parent d3ce900
commit 7fdb8d4
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/cleanlab_studio/internal/api/api.py b/cleanlab_studio/internal/api/api.py
@@ -534,7 +534,7 @@ def deploy_model(api_key: str, cleanset_id: str, model_name: str) -> str:
     return model_id
 
 
-def get_deployment_status(api_key: str, model_id: str) -> JSONDict:
+def get_deployment_status(api_key: str, model_id: str) -> str:
     """Gets status of model deployment."""
     check_uuid_well_formed(model_id, "model ID")
     res = requests.get(

diff --git a/cleanlab_studio/studio/studio.py b/cleanlab_studio/studio/studio.py
@@ -328,7 +328,7 @@ def deploy_model(self, cleanset_id: str, model_name: str) -> str:
         """
         return api.deploy_model(self._api_key, cleanset_id, model_name)
 
-    def wait_until_model_ready(self, model_id: str, timeout: Optional[float] = None) -> str:
+    def wait_until_model_ready(self, model_id: str, timeout: Optional[float] = None) -> None:
         """Blocks until a model is ready or the timeout is reached.
 
         Args: