fix test values

Xilinx · Nov 19, 2024 · 170ef90 · 170ef90
1 parent c399e9b
commit 170ef90
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/brevitas/core/test_standalone_scaling.py b/tests/brevitas/core/test_standalone_scaling.py
@@ -22,7 +22,7 @@ def test_scaling_state_dict():
 
 @torch.no_grad()
 def test_scaling_min_val_runtime():
-    scaling_min_val = torch.tensor(1e-6)
+    scaling_min_val = 1e-6
     scaling_op = ParameterFromRuntimeStatsScaling(
         collect_stats_steps=1,
         scaling_stats_impl=AbsMax(),
@@ -38,7 +38,7 @@ def test_scaling_min_val_runtime():
 @torch.no_grad()
 def test_scaling_min_val_param():
     inp = torch.zeros(1, 5)
-    scaling_min_val = torch.tensor(1e-6)
+    scaling_min_val = 1e-6
     scaling_op = ParameterFromStatsFromParameterScaling(
         scaling_stats_impl=AbsMax(),
         scaling_min_val=scaling_min_val,