We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
2 parents b77b25c + 8462e25 commit 8247af1Copy full SHA for 8247af1
1 file changed
tests/models/test_task.py
@@ -35,18 +35,3 @@ def test_transform_prediction_and_target() -> None:
35
transform_target=torch.log10,
36
transform_inference=torch.log10,
37
)
38
-
39
- # Test wrong combination of inputs
40
- with pytest.raises(
41
- AssertionError,
42
- match=(
43
- "Please specify both `transform_inference` and `transform_target`"
44
- ),
45
- ):
46
- EnergyReconstruction(
47
- hidden_size=gnn.nb_outputs,
48
- target_labels="energy",
49
- loss_function=LogCoshLoss(),
50
- transform_prediction_and_target=torch.log10,
51
- transform_inference=torch.log10,
52
- )
0 commit comments