Fix bug

wesselb · Aug 27, 2021 · 98f8f20 · 98f8f20
1 parent d329b0d
commit 98f8f20
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/stheno/model/measure.py b/stheno/model/measure.py
@@ -435,7 +435,7 @@ def sample(self, state: B.RandomState, *fdds: FDD):
 
     @_dispatch
     def sample(self, *fdds: FDD):
-        return B.squeeze(1, *fdds)
+        return self.sample(1, *fdds)
 
     @_dispatch
     def logpdf(self, *pairs: Union[list, tuple]):