Skip to content

Commit 93c8d35

Browse files
committed
fix eval script
1 parent 3d2ae40 commit 93c8d35

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

scripts/evaluation/evaluate.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -239,7 +239,7 @@ def generate_sample_forecasts(
239239
# Generate forecast samples
240240
forecast_samples = []
241241
for batch in tqdm(batcher(test_data_input, batch_size=batch_size)):
242-
context = [torch.tensor(entry["target"]) for entry in batch]
242+
context = [torch.tensor(entry["target"], dtype=torch.float32) for entry in batch]
243243
forecast_samples.append(
244244
pipeline.predict(
245245
context,

0 commit comments

Comments
 (0)