Filter custom events from stats_for_run query (#6781)

Same as https://github.com/dagster-io/internal/commit/eaa4a10f978544ab22d1e18cc2f55e1303b50808
dagster-io · Feb 24, 2022 · 0ed9ad7 · 0ed9ad7
1 parent 5cec1c8
commit 0ed9ad7
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/python_modules/dagster/dagster/core/storage/event_log/sql_event_log.py b/python_modules/dagster/dagster/core/storage/event_log/sql_event_log.py
@@ -289,7 +289,12 @@ def get_stats_for_run(self, run_id):
                     db.func.max(SqlEventLogStorageTable.c.timestamp).label("last_event_timestamp"),
                 ]
             )
-            .where(SqlEventLogStorageTable.c.run_id == run_id)
+            .where(
+                db.and_(
+                    SqlEventLogStorageTable.c.run_id == run_id,
+                    SqlEventLogStorageTable.c.dagster_event_type != None,
+                )
+            )
             .group_by("dagster_event_type")
         )
 
@@ -301,9 +306,9 @@ def get_stats_for_run(self, run_id):
             times = {}
             for result in results:
                 (dagster_event_type, n_events_of_type, last_event_timestamp) = result
-                if dagster_event_type:
-                    counts[dagster_event_type] = n_events_of_type
-                    times[dagster_event_type] = last_event_timestamp
+                check.invariant(dagster_event_type is not None)
+                counts[dagster_event_type] = n_events_of_type
+                times[dagster_event_type] = last_event_timestamp
 
             enqueued_time = times.get(DagsterEventType.PIPELINE_ENQUEUED.value, None)
             launch_time = times.get(DagsterEventType.PIPELINE_STARTING.value, None)