fix for torch backend

LLNL · May 3, 2024 · d717e79 · d717e79
1 parent 1375167
commit d717e79
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/MuyGPyS/gp/deformation/isotropy.py b/MuyGPyS/gp/deformation/isotropy.py
@@ -84,7 +84,7 @@ def __call__(
         if length_scale is None:
             length_scale = self.length_scale(**kwargs)
         # This is brittle and I hate it. I'm not sure where to put this logic.
-        if isinstance(length_scale, mm.ndarray):
+        if isinstance(length_scale, mm.ndarray) and len(length_scale.shape) > 0:
             shape = [None] * dists.ndim
             shape[0] = slice(None)
             length_scale = length_scale[tuple(shape)]