From 30a45bb59746a658e8f9fc970ab2467fdb00443d Mon Sep 17 00:00:00 2001
From: silur <silur@cryptall.co>
Date: Tue, 11 Oct 2022 13:17:21 +0200
Subject: [PATCH] add XGBoostRF models to freqai test interface

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

diff --git a/tests/freqai/test_freqai_interface.py b/tests/freqai/test_freqai_interface.py
index a61853c47..2f5ada90a 100644
--- a/tests/freqai/test_freqai_interface.py
+++ b/tests/freqai/test_freqai_interface.py
@@ -30,6 +30,7 @@ def is_mac() -> bool:
 @pytest.mark.parametrize('model', [
     'LightGBMRegressor',
     'XGBoostRegressor',
+    'XGBoostRFRegressor',
     'CatboostRegressor',
     ])
 def test_extract_data_and_train_model_Standard(mocker, freqai_conf, model):
@@ -111,6 +112,7 @@ def test_extract_data_and_train_model_MultiTargets(mocker, freqai_conf, model):
     'LightGBMClassifier',
     'CatboostClassifier',
     'XGBoostClassifier',
+    'XGBoostRFClassifier',
     ])
 def test_extract_data_and_train_model_Classifiers(mocker, freqai_conf, model):
     if is_arm() and model == 'CatboostClassifier':