metadata methods

achals · achals · commit 7446c2081c15 · 2022-06-06T11:22:02.000-07:00
Signed-off-by: Achal Shah &lt;achals@gmail.com&gt;
diff --git a/sdk/python/feast/infra/registry_stores/sql.py b/sdk/python/feast/infra/registry_stores/sql.py
@@ -140,7 +140,6 @@
 
 
 class SqlRegistry(BaseRegistry):
-
     def __init__(
         self, registry_config: Optional[RegistryConfig], repo_path: Optional[Path]
     ):
@@ -475,7 +474,12 @@ def update_infra(self, infra: Infra, project: str, commit: bool = True):
     def get_infra(self, project: str, allow_cache: bool = False) -> Infra:
         pass
 
-    def apply_user_metadata(self, project: str, feature_view: BaseFeatureView, metadata_bytes: Optional[bytes]):
+    def apply_user_metadata(
+        self,
+        project: str,
+        feature_view: BaseFeatureView,
+        metadata_bytes: Optional[bytes],
+    ):
         if isinstance(feature_view, FeatureView):
             table = feature_views
         elif isinstance(feature_view, OnDemandFeatureView):
@@ -500,14 +504,16 @@ def apply_user_metadata(self, project: str, feature_view: BaseFeatureView, metad
                 }
                 update_stmt = (
                     update(table)
-                        .where(getattr(table.c, "feature_view_name") == name)
-                        .values(values,)
+                    .where(getattr(table.c, "feature_view_name") == name)
+                    .values(values,)
                 )
                 conn.execute(update_stmt)
             else:
                 raise FeatureViewNotFoundException(feature_view.name, project=project)
 
-    def get_user_metadata(self, project: str, feature_view: BaseFeatureView) -> Optional[bytes]:
+    def get_user_metadata(
+        self, project: str, feature_view: BaseFeatureView
+    ) -> Optional[bytes]:
         if isinstance(feature_view, FeatureView):
             table = feature_views
         elif isinstance(feature_view, OnDemandFeatureView):
@@ -583,7 +589,6 @@ def _apply_object(
                 insert_stmt = insert(table).values(values,)
                 conn.execute(insert_stmt)
 
-
     def _delete_object(self, table, name, project, id_field_name, not_found_exception):
         with self.engine.connect() as conn:
             stmt = delete(table).where(getattr(table.c, id_field_name) == name)
diff --git a/sdk/python/feast/registry.py b/sdk/python/feast/registry.py
@@ -632,11 +632,18 @@ def get_infra(self, project: str, allow_cache: bool = False) -> Infra:
         """
 
     @abstractmethod
-    def apply_user_metadata(self, project: str, feature_view: BaseFeatureView, metadata_bytes: Optional[bytes]):
+    def apply_user_metadata(
+        self,
+        project: str,
+        feature_view: BaseFeatureView,
+        metadata_bytes: Optional[bytes],
+    ):
         ...
 
     @abstractmethod
-    def get_user_metadata(self, project: str, feature_view: BaseFeatureView) -> Optional[bytes]:
+    def get_user_metadata(
+        self, project: str, feature_view: BaseFeatureView
+    ) -> Optional[bytes]:
         ...
 
     @abstractmethod
@@ -662,10 +669,17 @@ class Registry(BaseRegistry):
     Registry: A registry allows for the management and persistence of feature definitions and related metadata.
     """
 
-    def apply_user_metadata(self, project: str, feature_view: BaseFeatureView, metadata_bytes: Optional[bytes]):
+    def apply_user_metadata(
+        self,
+        project: str,
+        feature_view: BaseFeatureView,
+        metadata_bytes: Optional[bytes],
+    ):
         pass
 
-    def get_user_metadata(self, project: str, feature_view: BaseFeatureView) -> Optional[bytes]:
+    def get_user_metadata(
+        self, project: str, feature_view: BaseFeatureView
+    ) -> Optional[bytes]:
         pass
 
     # The cached_registry_proto object is used for both reads and writes. In particular,
