Correct tests

golmschenk · Oct 10, 2024 · d464747 · d464747
1 parent 0ffb59f
commit d464747
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/unit_tests/test_lightning_module.py b/tests/unit_tests/test_lightning_module.py
@@ -27,7 +27,7 @@ def create_fake_qusi_lightning_module() -> QusiLightningModule:
 
 
 def create_fake_metric_group() -> MetricGroup:
-    fake_metric_group = MetricGroup(loss_metric=Mock(return_value=torch.tensor([1])),
+    fake_metric_group = MetricGroup(loss_metric=Mock(return_value=torch.tensor(1)),
                                     state_based_logging_metrics=ModuleList([MockStateBasedMetric()]),
                                     functional_logging_metrics=ModuleList([MockFunctionalMetric()]))
     return fake_metric_group