Skip to content

Commit

Permalink
fix(transform): Remove explicit copy of time axis in CollateProducts.
Browse files Browse the repository at this point in the history
Instead, the copy_from parameter will handle directly copying all
index_map entries, including the time axis.
  • Loading branch information
ljgray committed Feb 14, 2023
1 parent 0ba0957 commit 4012c28
Showing 1 changed file with 0 additions and 8 deletions.
8 changes: 0 additions & 8 deletions draco/analysis/transform.py
Original file line number Diff line number Diff line change
Expand Up @@ -203,13 +203,6 @@ def process(self, ss):
ss_prod = ss.prod
ss_conj = np.zeros(ss_prod.size, dtype=bool)

# Add the time-like axis to the kwargs
output_kwargs = (
{"ra": ss.ra[:]}
if isinstance(ss, containers.SiderealStream)
else {"time": ss.time[:]}
)

# Create output container
sp = ss.__class__(
freq=bt_freq,
Expand All @@ -220,7 +213,6 @@ def process(self, ss):
copy_from=ss,
distributed=True,
comm=ss.comm,
**output_kwargs,
)

# Check if frequencies are already ordered
Expand Down

0 comments on commit 4012c28

Please sign in to comment.