Fix cont test function

SIAnalytics · Jun 13, 2022 · af90270 · af90270
1 parent 5c7dacc
commit af90270
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mmtune/mm/tasks/cont_test_func.py b/mmtune/mm/tasks/cont_test_func.py
@@ -34,7 +34,7 @@ def _step(s: float) -> float:
     Returns:
         float: output.
     """
-    return float(np.exp(int(np.log(np.nan_to_num(s)))))
+    return float(np.exp(int(np.nan_to_num(np.log(s)))))
 
 
 @TASKS.register_module()