From 366740885a85aa9111aed8cfa8a92d9c5b862e07 Mon Sep 17 00:00:00 2001
From: Yinon Polak <inon51@gmail.com>
Date: Mon, 13 Mar 2023 20:18:26 +0200
Subject: [PATCH] reduce mlp number of parameters for testing

---
 tests/freqai/test_freqai_interface.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/freqai/test_freqai_interface.py b/tests/freqai/test_freqai_interface.py
index 3b31012b2..f34659398 100644
--- a/tests/freqai/test_freqai_interface.py
+++ b/tests/freqai/test_freqai_interface.py
@@ -92,7 +92,7 @@ def test_extract_data_and_train_model_Standard(mocker, freqai_conf, model, pca,
             "max_iters": 1,
             "batch_size": 64,
             "learning_rate": 3e-4,
-            "max_n_eval_batches": None,
+            "max_n_eval_batches": 1,
             "model_kwargs": {
                 "hidden_dim": 32,
                 "dropout_percent": 0.2,