fix test

dagster-io · Dec 1, 2023 · 73f5052 · 73f5052
1 parent dd44ea0
commit 73f5052
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/python_modules/dagster/dagster_tests/storage_tests/test_io_manager.py b/python_modules/dagster/dagster_tests/storage_tests/test_io_manager.py
@@ -1011,7 +1011,9 @@ def job1():
     job1.execute_in_process()
 
     assert my_io_manager.handle_output_calls == 1  # Nothing return type for op1 skips I/O manager
-    assert my_io_manager.handle_input_calls == 1
+    assert (
+        my_io_manager.handle_input_calls == 0
+    )  # Nothing return type for op1 means we skip I/O manager and directly provide a None
 
 
 def test_instance_set_on_input_context():