sample requires integer number of samples, was being passed as a 1-tu…

…ple. This should fix failed checks.
rctn · Nov 12, 2024 · 21b0d67 · 21b0d67
1 parent 0830756
commit 21b0d67
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sparsecoding/data/datasets/bars.py b/sparsecoding/data/datasets/bars.py
@@ -48,7 +48,7 @@ def __init__(
         v_bars = v_bars.expand(self.P, self.P, self.P)
         self.basis = torch.cat((h_bars, v_bars), dim=0)  # [2*P, P, P]
 
-        self.weights = prior.sample([self.N])  # [N, 2*P]
+        self.weights = prior.sample(self.N)  # [N, 2*P]
 
         self.data = torch.einsum(
             "nd,dhw->nhw",