fix: handle alias different to name and default meta (#7)

fal-ai · Dec 16, 2022 · 07a0fa3 · 07a0fa3
1 parent f18be27
commit 07a0fa3
Showing 1 changed file with 10 additions and 13 deletions.
diff --git a/macros/__f__load_meta.sql b/macros/__f__load_meta.sql
@@ -4,25 +4,22 @@
 {% endif %}
 
 {% if feature_store.__f__is_relation(obj) %}
+    {# Set default meta #}
+    {% set meta = {} %}
+
     {# Find the meta of the `obj` relation #}
-    {# Try in models #}
     {% set nodes = graph.nodes.values()
-            | selectattr("resource_type", "equalto", "model")
             | selectattr("database", "equalto", obj.table.database)
             | selectattr("schema", "equalto", obj.table.schema)
-            | selectattr("name", "equalto", obj.table.identifier)
+            | selectattr("alias", "equalto", obj.table.identifier)
             | list %}
+
     {% if nodes %}
-        {% set meta = nodes[0].config.meta %}
-    {% else %}
-        {# Try in sources #}
-        {% set nodes = graph.sources.values()
-            | selectattr("resource_type", "equalto", "source")
-            | selectattr("database", "equalto", obj.table.database)
-            | selectattr("schema", "equalto", obj.table.schema)
-            | selectattr("name", "equalto", obj.table.identifier)
-            | list %}
-        {% set meta = nodes[0].meta %}
+        {% if nodes[0].resource_type == "model" %}
+            {% set meta = nodes[0].config.meta %}
+        {% elif nodes[0].resource_type == "source" %}
+            {% set meta = nodes[0].meta %}
+        {% endif %}
     {% endif %}
 
     {# Populate any non-set properties available in the feature_store meta #}