Reorder data cleaning

2022-11-04 11:21:22 +01:00
parent c2130ed3dd
commit e08cd6ebff
5 changed files with 55 additions and 24 deletions
--- a/tests/freqai/test_freqai_datakitchen.py
+++ b/tests/freqai/test_freqai_datakitchen.py
@@ -157,5 +157,5 @@ def test_make_train_test_datasets(mocker, freqai_conf):
    data_dictionary = freqai.dk.make_train_test_datasets(features_filtered, labels_filtered)

    assert data_dictionary
-    assert len(data_dictionary) == 7
+    assert len(data_dictionary) == 9
    assert len(data_dictionary['train_features'].index) == 1916