fix: psql facts

MatsMoll · Nov 2, 2023 · 55ec095 · 55ec095
1 parent 4008190
commit 55ec095
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/aligned/psql/jobs.py b/aligned/psql/jobs.py
@@ -370,7 +370,7 @@ def sql_aggregated_request(
         id_column = 'row_id'
         # id_column = window.group_by[0].name
         event_timestamp_clause: str | None = None
-        if request.event_timestamp_request:
+        if request.event_timestamp_request and request.event_timestamp_request.entity_column:
             timestamp = request.event_timestamp_request.event_timestamp
             entity_column = request.event_timestamp_request.entity_column
             group_by_names = {id_column}
@@ -490,7 +490,7 @@ def build_request_from_facts(self, facts: pl.LazyFrame) -> str:
                 entity_types[entity.name] = entity.dtype
                 all_entities.add(entity.name)
 
-            if request.event_timestamp_request:
+            if request.event_timestamp_request and request.event_timestamp_request.entity_column:
                 entity_column = request.event_timestamp_request.entity_column
                 has_event_timestamp = True
                 entity_types[entity_column] = FeatureType('').datetime
@@ -657,7 +657,7 @@ def build_sql_entity_query(self, sql_facts: PostgreSqlJob) -> str:
             final_select_names = final_select_names.union(
                 {f'entities.{entity}' for entity in request.entity_names}
             )
-            if request.event_timestamp_request:
+            if request.event_timestamp_request and request.event_timestamp_request.entity_column:
                 entity_column = request.event_timestamp_request.entity_column
 
                 if entity_column in sql_facts.query: