Fix mypy issue

pinecone-io · Sep 17, 2024 · b19cce2 · b19cce2
1 parent f3b2bab
commit b19cce2
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/pinecone/data/features/bulk_import.py b/pinecone/data/features/bulk_import.py
@@ -1,5 +1,5 @@
 from enum import Enum
-from typing import Optional, Literal, Iterator, List
+from typing import Optional, Literal, Iterator, List, Type, cast
 
 from pinecone.utils.decorators import prerelease_feature
 from pinecone.config.config import ConfigBuilder
@@ -20,7 +20,9 @@
 for m in [StartImportResponse, ImportListResponse, ImportModel]:
     install_json_repr_override(m)
 
-ImportErrorMode = Enum("ImportErrorMode", ImportErrorModeClass.allowed_values[("on_error",)])
+ImportErrorMode: Type[Enum] = cast(
+    Type[Enum], Enum("ImportErrorMode", ImportErrorModeClass.allowed_values[("on_error",)])
+)
 
 
 class ImportFeatureMixin: