Skip to content

Commit c65d5d1

Browse files
authored
add regions to integration tests (#1648)
1 parent 7a4d122 commit c65d5d1

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

integration_tests/small1_test.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,8 +15,8 @@
1515

1616
import omegaconf
1717
import pytest
18-
1918
from weathergen.evaluate.run_evaluation import evaluate_from_config
19+
2020
from weathergen.run_train import inference_from_args, train_with_args
2121
from weathergen.utils.metrics import get_train_metrics_path
2222

@@ -105,6 +105,7 @@ def evaluate_results(run_id):
105105
"dpi_val": 300,
106106
},
107107
"evaluation": {
108+
"regions": ["global"],
108109
"metrics": ["rmse", "l1", "mse"],
109110
"verbose": True,
110111
"summary_plots": True,

integration_tests/small_multi_stream_test.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,8 +23,8 @@
2323

2424
import omegaconf
2525
import pytest
26-
2726
from weathergen.evaluate.run_evaluation import evaluate_from_config
27+
2828
from weathergen.run_train import inference_from_args, train_with_args
2929
from weathergen.utils.metrics import get_train_metrics_path
3030

@@ -105,6 +105,7 @@ def evaluate_multi_stream_results(run_id):
105105
"dpi_val": 300,
106106
},
107107
"evaluation": {
108+
"regions": ["global"],
108109
"metrics": ["rmse", "l1", "mse"],
109110
"verbose": True,
110111
"summary_plots": True,

0 commit comments

Comments
 (0)