sql

Signed-off-by: Tommy Hughes <tohughes@redhat.com>
feast-dev · Jun 14, 2024 · 3f784e2 · 3f784e2
1 parent 5d1b4fe
commit 3f784e2
Showing 1 changed file with 5 additions and 7 deletions.
diff --git a/sdk/python/feast/infra/registry/snowflake.py b/sdk/python/feast/infra/registry/snowflake.py
@@ -782,15 +782,13 @@ def _list_objects(
             """
             df = execute_snowflake_statement(conn, query).fetch_pandas_all()
             if not df.empty:
-                objects = [
-                    python_class.from_proto(
+                objects = []
+                for row in df.iterrows():
+                    obj = python_class.from_proto(
                         proto_class.FromString(row[1][proto_field_name])
                     )
-                    for row in df.iterrows()
-                ]
-                for obj in objects:
-                    if not utils.has_all_tags(obj.tags, tags):
-                        objects.remove(obj)
+                    if utils.has_all_tags(obj.tags, tags):
+                        objects.append(obj)
                 return objects
         return []