fix attdata

TomoATT · Jun 25, 2024 · 0399bb3 · 0399bb3
1 parent 979f435
commit 0399bb3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pytomoatt/data.py b/pytomoatt/data.py
@@ -73,9 +73,9 @@ def read(cls, fname:str, fname_params:str,
             attdata.fgrid = np.loadtxt(fname_grid)
         if isinstance(dataset_name, str) and attdata.format == 'hdf5':
             attdata._add_field(dataset_name)
-            attdata.__dict__[key], attdata.grid_glob_r, \
+            attdata.__dict__[dataset_name], attdata.grid_glob_r, \
             attdata.grid_glob_t, attdata.grid_glob_p = \
-            attdata._data_retrieval(group_name=group_name, dataset_name=key)
+            attdata._data_retrieval(group_name=group_name, dataset_name=dataset_name)
         elif isinstance(dataset_name, str) and attdata.format != 'hdf5':
             attdata._add_field('data')
             attdata.data, attdata.grid_glob_r, attdata.grid_glob_t, attdata.grid_glob_p = \