Merge 80752a8 into 7f1e8c5

microsoft · Apr 26, 2023 · 9260626 · 9260626
2 parents 7f1e8c5 + 80752a8
commit 9260626
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/qlib/contrib/model/pytorch_alstm.py b/qlib/contrib/model/pytorch_alstm.py
@@ -218,8 +218,8 @@ def fit(
         save_path=None,
     ):
 
-        df_train, df_valid, df_test = dataset.prepare(
-            ["train", "valid", "test"],
+        df_train, df_valid = dataset.prepare(
+            ["train", "valid"],
             col_set=["feature", "label"],
             data_key=DataHandlerLP.DK_L,
         )