Apply suggestions from code review

Lightning-AI · Mar 26, 2021 · bf78331 · bf78331
1 parent 38819cb
commit bf78331
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tests/helpers/testers.py b/tests/helpers/testers.py
@@ -187,11 +187,11 @@ def _assert_half_support(
         target: torch tensor with targets
         device: determine device, either "cpu" or "cuda"
     """
-    p = preds[0].half().to(device) if preds[0].is_floating_point() else preds[0].to(device)
-    t = target[0].half().to(device) if target[0].is_floating_point() else target[0].to(device)
+    y_hat = preds[0].half().to(device) if preds[0].is_floating_point() else preds[0].to(device)
+    y = target[0].half().to(device) if target[0].is_floating_point() else target[0].to(device)
     metric_module = metric_module.to(device)
-    assert metric_module(p, t)
-    assert metric_functional(p, t)
+    assert metric_module(y_hat, y)
+    assert metric_functional(y_hat, y)
 
 
 class MetricTester: