more linting

Signed-off-by: Francisco Javier Arceo <[email protected]>
feast-dev · Dec 20, 2024 · f324251 · f324251
1 parent 1c86e54
commit f324251
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/sdk/python/feast/infra/online_stores/milvus_online_store/milvus.py b/sdk/python/feast/infra/online_stores/milvus_online_store/milvus.py
@@ -9,6 +9,7 @@
     FieldSchema,
     connections,
 )
+from pymilvus.orm.connections import Connections
 
 from feast import Entity
 from feast.feature_view import FeatureView
@@ -102,7 +103,7 @@ class MilvusOnlineStore(OnlineStore):
         _collections: Dictionary to cache Milvus collections.
     """
 
-    _conn: Optional[connections] = None
+    _conn: Optional[Connections] = None
     _collections: Dict[str, Collection] = {}
 
     def _connect(self, config: RepoConfig) -> connections: