sql

tchughesiv · tchughesiv · commit 3f784e2da129 · 2024-06-14T15:56:28.000-05:00
Signed-off-by: Tommy Hughes &lt;tohughes@redhat.com&gt;
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 []
 
