Issue724 (#745)

* fixing issue724 * fixing issue724
2026-02-16 19:45:17 -05:00 · 2022-10-04 10:51:12 -04:00
parent 0df54ec3f3
commit ceb3e300cd
2 changed files with 8 additions and 2 deletions
--- a/test/nlp/test_autohf.py
+++ b/test/nlp/test_autohf.py
@@ -56,7 +56,7 @@ def test_hf_data():
        record_id=0,
        **automl_settings
    )
-    automl.predict(X_test)
+    automl.predict(X_test, **{"per_device_eval_batch_size": 2})
    automl.predict(["test test", "test test"])
    automl.predict(
        [