revert moving spy call

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