Skip to content

Commit cbf3318

Browse files
authored
Update data_factory.py: use training prompt for testing
1 parent 8239910 commit cbf3318

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

ltsm/data_provider/data_factory.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -277,7 +277,7 @@ def getDatasets(self)->Tuple[TSDataset, TSDataset, List[TSDataset]]:
277277

278278

279279
# Test Prompt
280-
test_prompt_data_path = self.prompt_data_path + '/test'
280+
test_prompt_data_path = self.prompt_data_path + '/train'
281281
sub_test_prompt_data, tmp = self.loadPrompts(data_path, test_prompt_data_path, buff)
282282
missing.update(set(tmp))
283283

@@ -421,4 +421,4 @@ def get_data_loaders(args):
421421
num_workers=0,
422422
)
423423

424-
return train_loader, val_loader, test_loader, processor
424+
return train_loader, val_loader, test_loader, processor

0 commit comments

Comments
 (0)