catch diagonal embedding for no errors

Signed-off-by: nstarman <nstarman@users.noreply.github.com>
GalOrrery · Aug 8, 2023 · a6e0c34 · a6e0c34
1 parent 5bba9c6
commit a6e0c34
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/src/stream_ml/pytorch/builtin/_isochrone/core.py b/src/stream_ml/pytorch/builtin/_isochrone/core.py
@@ -296,9 +296,8 @@ def ln_likelihood(
         )
 
         # Covariance: star (N, [I], F, F)
-        stds = data[self.phot_err_names].array
         std_data = (
-            xp.diag_embed(stds)[:, None, :, :]
+            xp.diag_embed(data[self.phot_err_names].array)[:, None, :, :]
             if self.phot_err_names is not None
             else self.xp.zeros(1)
         )