fix test

PrefectHQ · Jul 19, 2024 · 801d8b3 · 801d8b3
1 parent ba246ce
commit 801d8b3
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/test_deployments.py b/tests/test_deployments.py
@@ -1627,4 +1627,6 @@ def pretend_flow():
         result = await load_flow_from_flow_run(flow_run, client=prefect_client)
 
         assert result == pretend_flow
-        load_flow_from_entrypoint.assert_called_once_with("my.module.pretend_flow")
+        load_flow_from_entrypoint.assert_called_once_with(
+            "my.module.pretend_flow", True
+        )