Skip to content

Commit dfef96a

Browse files
authored
Merge pull request #192 from usnistgov/190_dimension_hint_for_count
`SST1RSoXSDB` fix dimension hinting for count plans
2 parents 7000f63 + d94d753 commit dfef96a

File tree

2 files changed

+2
-4
lines changed

2 files changed

+2
-4
lines changed

src/PyHyperScattering/SST1RSoXSDB.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -625,7 +625,7 @@ def loadRun(
625625
elif "spiralsearch" in md["start"]["plan_name"] and dims is None:
626626
dims = ["sam_x", "sam_y"]
627627
elif "count" in md["start"]["plan_name"] and dims is None:
628-
dims = ["epoch"]
628+
dims = ["time"]
629629
else:
630630
axes_to_include = []
631631
rsd_cutoff = 0.005

tests/test_SST1DBLoader.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -90,7 +90,6 @@ def test_SST1DB_load_energy_scan_20241209(sstdb):
9090
run = sstdb.loadRun(91175).unstack('system')
9191
assert type(run) == xr.DataArray
9292
assert 'energy' in run.indexes
93-
assert 'polarization' in run.indexes
9493

9594
@must_have_tiled
9695
def test_SST1DB_load_energy_scan_20250213(sstdb):
@@ -107,10 +106,9 @@ def test_SST1DB_load_spiral_scan_20250221(sstdb):
107106

108107
@must_have_tiled
109108
def test_SST1DB_load_count_scan_20250222(sstdb):
110-
run = sstdb.loadRun(92849,dims=['time','polarization']).unstack('system')
109+
run = sstdb.loadRun(92849).unstack('system')
111110
assert type(run) == xr.DataArray
112111
assert 'time' in run.indexes
113-
assert 'polarization' in run.indexes
114112

115113
@must_have_tiled
116114
def test_SST1DB_load_energy_scan_20250223(sstdb):

0 commit comments

Comments
 (0)