updating permissions

franciscojavierarceo · franciscojavierarceo · commit 815a35203e82 · 2024-09-29T15:06:29.000-04:00
Signed-off-by: Francisco Javier Arceo &lt;farceo@redhat.com&gt;
diff --git a/sdk/python/feast/infra/online_stores/sqlite.py b/sdk/python/feast/infra/online_stores/sqlite.py
@@ -153,8 +153,13 @@ def online_write_batch(
                         )
 
                     else:
-                        if table_name == 'test_on_demand_python_transformation_python_stored_writes_feature_view':
-                            print(f"writing online batch for {table_name} - {feature_name} = {val}")
+                        if (
+                            table_name
+                            == "test_on_demand_python_transformation_python_stored_writes_feature_view"
+                        ):
+                            print(
+                                f"writing online batch for {table_name} - {feature_name} = {val}"
+                            )
                         conn.execute(
                             f"""
                                 UPDATE {table_name}
@@ -174,20 +179,24 @@ def online_write_batch(
 
                         try:
                             conn.execute(
-                            f"""INSERT OR IGNORE INTO {table_name}
+                                f"""INSERT OR IGNORE INTO {table_name}
                                 (entity_key, feature_name, value, event_ts, created_ts)
                                 VALUES (?, ?, ?, ?, ?)""",
-                            (
-                                entity_key_bin,
-                                feature_name,
-                                val.SerializeToString(),
-                                timestamp,
-                                created_ts,
-                            ),
-                        )
+                                (
+                                    entity_key_bin,
+                                    feature_name,
+                                    val.SerializeToString(),
+                                    timestamp,
+                                    created_ts,
+                                ),
+                            )
                         except Exception as e:
-                            print(f"error writing online batch for {table_name} - {feature_name} = {val}\n {e}")
-                            print(f'querying all records for table: {conn.execute(f"select * from {table_name}").fetchall()}')
+                            print(
+                                f"error writing online batch for {table_name} - {feature_name} = {val}\n {e}"
+                            )
+                            print(
+                                f'querying all records for table: {conn.execute(f"select * from {table_name}").fetchall()}'
+                            )
                 if progress:
                     progress(1)
 
@@ -254,7 +263,7 @@ def update(
         project = config.project
 
         for table in tables_to_keep:
-            print(f'updating {_table_id(project, table)}')
+            print(f"updating {_table_id(project, table)}")
             conn.execute(
                 f"CREATE TABLE IF NOT EXISTS {_table_id(project, table)} (entity_key BLOB, feature_name TEXT, value BLOB, vector_value BLOB, event_ts timestamp, created_ts timestamp,  PRIMARY KEY(entity_key, feature_name))"
             )
