Attempt to fix tests (ATS is mad)

codecov · Sep 5, 2023 · 73a76ab · 73a76ab
1 parent f22e5f5
commit 73a76ab
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/conftest.py b/conftest.py
@@ -271,12 +271,10 @@ def mock_submit_fn(metric, start, end):
     mock_submit = mocker.Mock()
     mock_submit.side_effect = mock_submit_fn
 
-    import helpers
+    from helpers.checkpoint_logger import CheckpointLogger
 
     return mocker.patch.object(
-        helpers.checkpoint_logger.CheckpointLogger,
+        CheckpointLogger,
         "submit_subflow",
         mock_submit,
     )
-
-    return mock_submit