🔨 x, y, z is ensure coords

arafune · Mar 10, 2024 · a282b1f · a282b1f
1 parent 803f70a
commit a282b1f
Showing 1 changed file with 5 additions and 14 deletions.
diff --git a/src/arpes/xarray_extensions.py b/src/arpes/xarray_extensions.py
@@ -1314,20 +1314,11 @@ def spectrometer_settings(self) -> dict[str, Any]:
 
     @property
     def sample_pos(self) -> tuple[float, float, float]:
-        x, y, z = None, None, None
-        with contextlib.suppress(KeyError):
-            x = self._obj.attrs["x"]
-
-        with contextlib.suppress(KeyError):
-            y = self._obj.attrs["y"]
-
-        with contextlib.suppress(KeyError):
-            z = self._obj.attrs["z"]
-
-        def do_float(w: float | None) -> float:
-            return float(w) if w is not None else np.nan
-
-        return (do_float(x), do_float(y), do_float(z))
+        return (
+            float(self._obj.attrs["x"]),
+            float(self._obj.attrs["y"]),
+            float(self._obj.attrs["z"]),
+        )
 
     @property
     def sample_angles(