use tuple

2026-09-10 12:37:53 +08:00 · 2019-12-01 10:05:36 +01:00
parent a01b5be848
commit 7905888848
1 changed files with 1 additions and 1 deletions
@@ -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,