make pyright happy

dagster-io · Jan 29, 2024 · a0bec89 · a0bec89
1 parent 5bf8b32
commit a0bec89
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/python_modules/dagster/dagster_tests/definitions_tests/test_multi_partitions.py b/python_modules/dagster/dagster_tests/definitions_tests/test_multi_partitions.py
@@ -519,14 +519,18 @@ def test_context_partition_time_window():
 
     @asset(partitions_def=partitions_def)
     def my_asset(context: AssetExecutionContext):
+        time_partition = get_time_partitions_def(partitions_def)
+        if time_partition is None:
+            assert False, "expected a time component in the partitions definition"
+
         time_window = TimeWindow(
             start=pendulum.instance(
                 datetime(year=2020, month=1, day=1),
-                tz=get_time_partitions_def(partitions_def).timezone,
+                tz=time_partition.timezone,
             ),
             end=pendulum.instance(
                 datetime(year=2020, month=1, day=2),
-                tz=get_time_partitions_def(partitions_def).timezone,
+                tz=time_partition.timezone,
             ),
         )
         assert context.partition_time_window == time_window