move utilsforecast imports to performance test

Nixtla · Oct 12, 2023 · be8580e · be8580e
1 parent 80f6d50
commit be8580e
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 13 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -45,4 +45,4 @@ wheel.py-api = "cp37"
 
 [tool.cibuildwheel]
 test-requires = "pytest"
-test-command = "pytest {project}/tests"
+test-command = "pytest {project}/tests/test_correctness.py"
diff --git a/tests/test_scalers.py b/tests/test_scalers.py
@@ -1,11 +1,5 @@
 import numpy as np
 import pytest
-from utilsforecast.target_transforms import (
-    LocalStandardScaler as UtilsStandardScaler,
-    LocalMinMaxScaler as UtilsMinMaxScaler,
-    LocalRobustScaler as UtilsRobustScaler,
-)
-
 from coreforecast.grouped_array import GroupedArray
 from coreforecast.scalers import (
     LocalMinMaxScaler,
@@ -67,12 +61,6 @@ def scaler_inverse_transform(x, stats):
     "robust-iqr": LocalRobustScaler("iqr"),
     "robust-mad": LocalRobustScaler("mad"),
 }
-scaler2utils = {
-    "standard": UtilsStandardScaler(),
-    "minmax": UtilsMinMaxScaler(),
-    "robust-iqr": UtilsRobustScaler("iqr"),
-    "robust-mad": UtilsRobustScaler("mad"),
-}
 scalers = list(scaler2fns.keys())
 
 
@@ -121,6 +109,19 @@ def test_correctness(data, indptr, scaler_name, dtype):
 @pytest.mark.parametrize("dtype", [np.float32, np.float64])
 @pytest.mark.parametrize("lib", ["core", "utils"])
 def test_performance(benchmark, data, indptr, scaler_name, dtype, lib):
+    from utilsforecast.target_transforms import (
+        LocalStandardScaler as UtilsStandardScaler,
+        LocalMinMaxScaler as UtilsMinMaxScaler,
+        LocalRobustScaler as UtilsRobustScaler,
+    )
+
+    scaler2utils = {
+        "standard": UtilsStandardScaler(),
+        "minmax": UtilsMinMaxScaler(),
+        "robust-iqr": UtilsRobustScaler("iqr"),
+        "robust-mad": UtilsRobustScaler("mad"),
+    }
+
     ga = GroupedArray(data.astype(dtype), indptr)
     if lib == "core":
         scaler = scaler2core[scaler_name]