diff --git a/docs/user-guide/dream/dream-instrument-view.ipynb b/docs/user-guide/dream/dream-instrument-view.ipynb index 8987b224..ad4dc3fa 100644 --- a/docs/user-guide/dream/dream-instrument-view.ipynb +++ b/docs/user-guide/dream/dream-instrument-view.ipynb @@ -183,7 +183,7 @@ }, "outputs": [], "source": [ - "mantle_view = dream.instrument_view(dg[\"mantle\"].hist(wavelength=50), dim=\"wavelength\")\n", + "mantle_view = dream.instrument_view(dg[\"mantle\"].hist(tof=50), dim=\"tof\")\n", "mantle_view" ] }, @@ -201,7 +201,7 @@ }, "outputs": [], "source": [ - "mantle_view[1].controls[\"wavelength\"][\"slider\"].value = 43" + "mantle_view[1].controls[\"tof\"][\"slider\"].value = 43" ] }, { @@ -242,7 +242,7 @@ "name": "python", "nbconvert_exporter": "python", "pygments_lexer": "ipython3", - "version": "3.10.13" + "version": "3.10.14" } }, "nbformat": 4, diff --git a/src/ess/dream/io/geant4.py b/src/ess/dream/io/geant4.py index 85070ada..50fb896b 100644 --- a/src/ess/dream/io/geant4.py +++ b/src/ess/dream/io/geant4.py @@ -83,6 +83,7 @@ def _load_raw_events(file_path: str) -> sc.DataArray: file_path, sep="\t", header_parser="bracket", data_columns=[] ) table = table.rename_dims(row="event") + table.coords.pop("lambda") return sc.DataArray( sc.ones(sizes=table.sizes, with_variances=True, unit="counts"), coords=table.coords, @@ -90,8 +91,6 @@ def _load_raw_events(file_path: str) -> sc.DataArray: def _adjust_coords(da: sc.DataArray) -> None: - da.coords["wavelength"] = da.coords.pop("lambda") - da.coords["wavelength"].unit = "angstrom" da.coords["position"] = sc.spatial.as_vectors( da.coords.pop("x_pos"), da.coords.pop("y_pos"), da.coords.pop("z_pos") ) diff --git a/tests/dream/io/geant4_test.py b/tests/dream/io/geant4_test.py index b97bcc7f..b2162a16 100644 --- a/tests/dream/io/geant4_test.py +++ b/tests/dream/io/geant4_test.py @@ -113,7 +113,6 @@ def test_load_geant4_csv_mantle_has_expected_coords(file): assert "sector" not in mantle.bins.coords assert "tof" in mantle.bins.coords - assert "wavelength" in mantle.bins.coords assert "position" in mantle.coords @@ -128,7 +127,6 @@ def test_load_geant4_csv_endcap_backward_has_expected_coords(file): assert "sector" not in endcap.bins.coords assert "tof" in endcap.bins.coords - assert "wavelength" in endcap.bins.coords assert "position" in endcap.coords @@ -143,7 +141,6 @@ def test_load_geant4_csv_endcap_forward_has_expected_coords(file): assert "sector" not in endcap.bins.coords assert "tof" in endcap.bins.coords - assert "wavelength" in endcap.bins.coords assert "position" in endcap.coords @@ -157,7 +154,6 @@ def test_load_geant4_csv_high_resolution_has_expected_coords(file): assert_index_coord(hr.coords["sector"], values=set(range(1, 5))) assert "tof" in hr.bins.coords - assert "wavelength" in hr.bins.coords assert "position" in hr.coords @@ -174,7 +170,6 @@ def test_load_geant4_csv_sans_has_expected_coords(file): assert_index_coord(sans.coords["sector"], values=set(range(1, 5))) assert "tof" in sans.bins.coords - assert "wavelength" in sans.bins.coords assert "position" in sans.coords