Fix types

cohere-ai · Nov 7, 2024 · bd3041e · bd3041e
1 parent 715e261
commit bd3041e
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/src/cohere/manually_maintained/lazy_aws_deps.py b/src/cohere/manually_maintained/lazy_aws_deps.py
@@ -2,22 +2,22 @@
 
 def lazy_sagemaker():
     try:
-        import sagemaker as sage
+        import sagemaker as sage # type: ignore
         return sage
     except ImportError:
-        raise CohereError("Sagemaker not available. Please install sagemaker.")
+        raise ImportError("Sagemaker not available. Please install sagemaker.")
 
 def lazy_boto3():
     try:
-        import boto3
+        import boto3 # type: ignore
         return boto3
     except ImportError:
-        raise CohereError("Boto3 not available. Please install lazy_boto3().")
+        raise ImportError("Boto3 not available. Please install lazy_boto3().")
 
 def lazy_botocore():
     try:
-        import botocore
+        import botocore # type: ignore
         return botocore
     except ImportError:
-        raise CohereError("Botocore not available. Please install botocore.")
+        raise ImportError("Botocore not available. Please install botocore.")