Fix test

yardenas · Oct 2, 2024 · 13f6c7e · 13f6c7e
1 parent 3a80617
commit 13f6c7e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_unsupervised_trainer.py b/tests/test_unsupervised_trainer.py
@@ -43,4 +43,4 @@ def test_epoch(trainer):
     with trainer as trainer:
         with patch.object(trainer.env, "reset", wraps=trainer.env.reset) as mock:
             trainer.train(1)
-    assert mock.call_count == 3
+    assert mock.call_count == 4