remove deci-common (#1887)

Deci-AI · Mar 5, 2024 · e7caf6c · e7caf6c
1 parent 2b601b8
commit e7caf6c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 7 deletions.
diff --git a/requirements.pro.txt b/requirements.pro.txt
@@ -1,2 +1 @@
 deci-platform-client>=5.0.0
-deci-common>=12.0.0
diff --git a/src/super_gradients/common/plugins/deci_client.py b/src/super_gradients/common/plugins/deci_client.py
@@ -33,7 +33,6 @@
         HardwareType,
         QuantizationLevel,
     )
-    from deci_common.data_interfaces.files_data_interface import FileDownloadFailedException
     from deci_platform_client.models import AutoNACFileName
     from deci_platform_client.exceptions import ApiException, ApiTypeError
 
@@ -51,10 +50,7 @@ class DeciClient:
 
     def __init__(self):
         if not client_enabled:
-            logger.error(
-                "deci-platform-client or deci-common are not installed. Model cannot be loaded from deci lab."
-                "Please install deci-platform-client>=5.0.0 and deci-common>=12.0.0"
-            )
+            logger.error("deci-platform-client is not installed. Model cannot be loaded from deci lab." "Please install deci-platform-client>=5.0.0")
             return
 
         self.lab_client = DeciPlatformClient()
@@ -111,7 +107,7 @@ def _download_file_to_cache_dir(self, file_url: str, file_path: str, timeout_sec
         try:
             urlretrieve(file_url, file_path)
         except urllib.error.ContentTooShortError as ex:
-            raise FileDownloadFailedException("File download did not finish correctly " + str(ex))
+            raise RuntimeError("File download did not finish correctly " + str(ex))
         return file_path
 
     def get_model_arch_params(self, model_name: str) -> Optional[DictConfig]: