flatten fields

ecmwf · Sep 15, 2024 · 6cea6b5 · 6cea6b5
1 parent ba5b567
commit 6cea6b5
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/src/anemoi/inference/runner.py b/src/anemoi/inference/runner.py
@@ -140,9 +140,7 @@ def run(
                     f"Expected datetime: {date.isoformat()} (for lag {lag})"
                 )
 
-        input_fields_numpy = input_fields.to_numpy(dtype=np.float32)
-
-        print(input_fields_numpy.shape)
+        input_fields_numpy = input_fields.to_numpy(dtype=np.float32, flatten=True)
 
         input_fields_numpy = input_fields_numpy.reshape(
             len(self.lagged),