maubin/ML_GravPotBCs
0
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

bugfix rank momenta

Browse source
This commit is contained in:
Mayeul Aubin 2025-07-04 10:58:42 +02:00
parent 75a3e5151e
commit 6d876485f7
1 changed files with 0 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

1
sCOCA_ML/dataset/momenta_dataset.py
View file

@ -217,7 +217,6 @@ class MomentaDataset(Dataset):
for file, varname in zip(input_paths, self.initial_conditions_variables) for file, varname in zip(input_paths, self.initial_conditions_variables)
] ]
target_array = read_field_chunk_periodic(target_path, self.N, self.N, self.N, ox, oy, oz, name=self.target_variable).array target_array = read_field_chunk_periodic(target_path, self.N, self.N, self.N, ox, oy, oz, name=self.target_variable).array
target_array = np.moveaxis(target_array, -1, 0) # Move channel dimension to the front
# Stack the input arrays # Stack the input arrays
input_tensor = np.stack(input_arrays, axis=0) input_tensor = np.stack(input_arrays, axis=0)