We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 3d2ae40 commit 93c8d35Copy full SHA for 93c8d35
scripts/evaluation/evaluate.py
@@ -239,7 +239,7 @@ def generate_sample_forecasts(
239
# Generate forecast samples
240
forecast_samples = []
241
for batch in tqdm(batcher(test_data_input, batch_size=batch_size)):
242
- context = [torch.tensor(entry["target"]) for entry in batch]
+ context = [torch.tensor(entry["target"], dtype=torch.float32) for entry in batch]
243
forecast_samples.append(
244
pipeline.predict(
245
context,
0 commit comments