From 7905888848e163177faa5915f7bcc2b75b588ac4 Mon Sep 17 00:00:00 2001 From: Kashif Rasul Date: Sun, 1 Dec 2019 10:05:36 +0100 Subject: [PATCH] use tuple --- test/model/deepar/test_auxillary_outputs.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/model/deepar/test_auxillary_outputs.py b/test/model/deepar/test_auxillary_outputs.py index 8f1bb68..2cdfeea 100644 --- a/test/model/deepar/test_auxillary_outputs.py +++ b/test/model/deepar/test_auxillary_outputs.py @@ -62,7 +62,7 @@ def test_distribution(): *[data_entry[k] for k in input_names] ) - assert distr.sample(torch.Size([num_samples])).shape == ( + assert distr.sample((num_samples,)).shape == ( num_samples, batch_size, seq_len,