Add comment on flattening

Co-authored-by: Juan M. Cruz-Martinez <[email protected]>
NNPDF · Feb 22, 2024 · a7dfc0d · a7dfc0d
1 parent 44901f0
commit a7dfc0d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/n3fit/src/n3fit/model_trainer.py b/n3fit/src/n3fit/model_trainer.py
@@ -535,7 +535,8 @@ def _generate_observables(
                 # Loop over replicas
                 for replica in self.exp_info:
                     if key in ["expdata", "expdata_vl"]:
-                        replica_data.append(replica[i][key].flatten())
+                        # Save the data with shape (ndata) instead of (1, ndata)
+                        replica_data.append(replica[i][key][0])
                     else:
                         replica_data.append(replica[i][key])
                 # Stack