Skip to content
Snippets Groups Projects
Commit 5346f981 authored by Philipp Arras's avatar Philipp Arras
Browse files

Less overhead during data loading

parent 8c4d1d3e
No related branches found
No related tags found
No related merge requests found
Pipeline #109337 passed
...@@ -325,7 +325,7 @@ class Observation(BaseObservation): ...@@ -325,7 +325,7 @@ class Observation(BaseObservation):
@staticmethod @staticmethod
def load(file_name, lo_hi_index=None): def load(file_name, lo_hi_index=None):
dct = dict(np.load(file_name)) dct = np.load(file_name)
antpos = [] antpos = []
for ii in range(4): for ii in range(4):
val = dct[f"antpos{ii}"] val = dct[f"antpos{ii}"]
...@@ -335,7 +335,6 @@ class Observation(BaseObservation): ...@@ -335,7 +335,6 @@ class Observation(BaseObservation):
pol = Polarization.from_list(dct["polarization"]) pol = Polarization.from_list(dct["polarization"])
direction = Direction.from_list(dct["direction"]) direction = Direction.from_list(dct["direction"])
slc = slice(None) if lo_hi_index is None else slice(*lo_hi_index) slc = slice(None) if lo_hi_index is None else slice(*lo_hi_index)
# FIXME Put barrier here that makes sure that only one full Observation is loaded at a time
return Observation( return Observation(
AntennaPositions.from_list(antpos), AntennaPositions.from_list(antpos),
dct["vis"][..., slc], dct["vis"][..., slc],
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment