From 7fdb8d4e99c56e0b0c747936d1297a021ab52f29 Mon Sep 17 00:00:00 2001 From: Angela Date: Mon, 17 Jun 2024 09:57:01 -0700 Subject: [PATCH] mypy --- cleanlab_studio/internal/api/api.py | 2 +- cleanlab_studio/studio/studio.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/cleanlab_studio/internal/api/api.py b/cleanlab_studio/internal/api/api.py index 1cea9f0c..b251d506 100644 --- 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 index 125ed6c3..1416e1b8 100644 --- 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: