revert moving spy call

iterative · Feb 12, 2024 · 48a1f7c · 48a1f7c
1 parent 4043b8d
commit 48a1f7c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/frameworks/test_huggingface.py b/tests/frameworks/test_huggingface.py
@@ -117,8 +117,8 @@ def test_huggingface_integration(tmp_dir, model, args, data, mocker, callback):
     )
     callback = callback()
     live = callback.live
-    trainer.add_callback(callback)
     spy = mocker.spy(Live, "end")
+    trainer.add_callback(callback)
     trainer.train()
     spy.assert_called_once()