call assert

mosaicml · Oct 10, 2023 · de602fe · de602fe
1 parent f54245d
commit de602fe
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_hf_mpt_gen.py b/tests/test_hf_mpt_gen.py
@@ -183,4 +183,4 @@ def test_mpt_generate_callback(callback_generate: Any, tmpdir: Path):
     trainer.logger.log_table = Mock()
     trainer.fit()
 
-    assert callback_generate.assert_called_once
+    callback_generate.assert_called_once()