fix integration tests (ecmwf#796)

Author: grassesi

integration_tests/small1.yaml

@@ -1,4 +1,4 @@
-streams_directory: "./config/streams/streams_test/"
+streams_directory: "./integration_tests/streams/"
 run_path: "./results"
 model_path: "./models"
 loss_fcts: [["mse", 1.0]]

integration_tests/small1_test.py

@@ -112,7 +112,7 @@ def evaluate_results(run_id):
                     "streams": {
                         "ERA5": {
                             "results_base_dir": "./results/",
-                            "channels": ["10v"],  # "all" indicator would be nice
+                            "channels": ["t_850"],  # "all" indicator would be nice
                             "evaluation": {"forecast_steps": "all", "sample": "all"},
                             "plotting": {
                                 "sample": [0, 1],

integration_tests/streams/era5_small.yml

@@ -0,0 +1,39 @@
+# (C) Copyright 2024 WeatherGenerator contributors.
+#
+# This software is licensed under the terms of the Apache Licence Version 2.0
+# which can be obtained at http://www.apache.org/licenses/LICENSE-2.0.
+#
+# In applying this licence, ECMWF does not waive the privileges and immunities
+# granted to it by virtue of its status as an intergovernmental organisation
+# nor does it submit to any jurisdiction.
+
+ERA5 :
+  type : anemoi
+  filenames : ['aifs-ea-an-oper-0001-mars-o96-1979-2022-6h-v6.zarr']
+  loss_weight : 1.
+  source_exclude : ['w_', 'skt', 'sp', 'tcw', 'cp', 'tp']
+  target_exclude : ['w_', 'skt', 'sp', 'tcw', 'cp', 'tp']
+  source : ["t_850", "z_850"]
+  target : ["t_850"]
+  diagnostic : False
+  masking_rate : 0.6
+  masking_rate_none : 0.05
+  token_size : 32
+  tokenize_spacetime : True
+  max_num_targets: -1
+  embed : 
+    net : transformer
+    num_tokens : 1
+    num_heads : 2
+    dim_embed : 16
+    num_blocks : 2
+  embed_target_coords :
+    net : linear
+    dim_embed : 16
+  target_readout :
+    type : 'obs_value'
+    num_layers : 2
+    num_heads : 2
+  pred_head :
+    ens_size : 1
+    num_layers : 1