Skip to content

Commit 0e9ac9d

Browse files
authored
Fix CI (#1529)
1 parent efffb28 commit 0e9ac9d

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

qlib/contrib/model/pytorch_sandwich.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -300,10 +300,15 @@ def test_epoch(self, data_x, data_y):
300300
return np.mean(losses), np.mean(scores)
301301

302302
def fit(
303-
self, dataset: DatasetH, evals_result=dict(), save_path=None,
303+
self,
304+
dataset: DatasetH,
305+
evals_result=dict(),
306+
save_path=None,
304307
):
305308
df_train, df_valid, df_test = dataset.prepare(
306-
["train", "valid", "test"], col_set=["feature", "label"], data_key=DataHandlerLP.DK_L,
309+
["train", "valid", "test"],
310+
col_set=["feature", "label"],
311+
data_key=DataHandlerLP.DK_L,
307312
)
308313
if df_train.empty or df_valid.empty:
309314
raise ValueError("Empty data from dataset, please check your dataset config.")

0 commit comments

Comments
 (0)