my pyright happy

dagster-io · Jan 29, 2024 · 5bf8b32 · 5bf8b32
1 parent d7edead
commit 5bf8b32
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/python_modules/dagster/dagster_tests/core_tests/test_op_invocation.py b/python_modules/dagster/dagster_tests/core_tests/test_op_invocation.py
@@ -1280,14 +1280,17 @@ def test_multipartitioned_time_window_asset_invocation():
 
     @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, "partitions def does not have a time component"
         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