more linting

franciscojavierarceo · franciscojavierarceo · commit fe518ed9cda3 · 2024-12-21T21:19:30.000-05:00
Signed-off-by: Francisco Javier Arceo &lt;farceo@redhat.com&gt;
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:
