remove scale_dims

ecmwf · Feb 20, 2025 · 06d3aab · 06d3aab
1 parent e7ad5d9
commit 06d3aab
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/training/tests/train/test_loss_scaling.py b/training/tests/train/test_loss_scaling.py
@@ -42,7 +42,6 @@ def fake_data(request: SubRequest) -> tuple[DictConfig, IndexCollection]:
                         "additional_scaler": request.param,
                         "general_variable": {
                             "_target_": "anemoi.training.losses.scalers.GeneralVariableLossScaler",
-                            "scale_dim": -1,  # dimension on which scaling applied
                             "weights": {
                                 "default": 1,
                                 "z": 0.1,
@@ -99,7 +98,6 @@ def fake_data(request: SubRequest) -> tuple[DictConfig, IndexCollection]:
     "_target_": "anemoi.training.losses.scalers.GraphNodeAttributeScaler",
     "nodes_name": "test_nodes",
     "nodes_attribute_name": "test_attr",
-    "apply_output_mask": False,
     "norm": "unit-sum",
 }
 
@@ -109,7 +107,6 @@ def fake_data(request: SubRequest) -> tuple[DictConfig, IndexCollection]:
     "nodes_attribute_name": "test_attr",
     "scaling_mask_attribute_name": "mask",
     "weight_frac_of_total": 0.4,
-    "apply_output_mask": False,
     "norm": "unit-sum",
 }