Merge pull request #368 from asogaard/fix-unit-test

Remove obsolete unit test
graphnet-team · Dec 7, 2022 · 8247af1 · 8247af1
2 parents b77b25c + 8462e25
commit 8247af1
Showing 1 changed file with 0 additions and 15 deletions.
diff --git a/tests/models/test_task.py b/tests/models/test_task.py
@@ -35,18 +35,3 @@ def test_transform_prediction_and_target() -> None:
             transform_target=torch.log10,
             transform_inference=torch.log10,
         )
-
-    # Test wrong combination of inputs
-    with pytest.raises(
-        AssertionError,
-        match=(
-            "Please specify both `transform_inference` and `transform_target`"
-        ),
-    ):
-        EnergyReconstruction(
-            hidden_size=gnn.nb_outputs,
-            target_labels="energy",
-            loss_function=LogCoshLoss(),
-            transform_prediction_and_target=torch.log10,
-            transform_inference=torch.log10,
-        )