From 6d876485f75eeb86e0ce81c20e512b2c1cf11ca4 Mon Sep 17 00:00:00 2001 From: Mayeul Aubin Date: Fri, 4 Jul 2025 10:58:42 +0200 Subject: [PATCH] bugfix rank momenta --- sCOCA_ML/dataset/momenta_dataset.py | 1 - 1 file changed, 1 deletion(-) diff --git a/sCOCA_ML/dataset/momenta_dataset.py b/sCOCA_ML/dataset/momenta_dataset.py index a36b650..328269c 100644 --- a/sCOCA_ML/dataset/momenta_dataset.py +++ b/sCOCA_ML/dataset/momenta_dataset.py @@ -217,7 +217,6 @@ class MomentaDataset(Dataset): 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 = np.moveaxis(target_array, -1, 0) # Move channel dimension to the front # Stack the input arrays input_tensor = np.stack(input_arrays, axis=0)