Inline benchmark test imports

siboehm · Jun 26, 2021 · aae2a7f · aae2a7f
1 parent 6df42c5
commit aae2a7f
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tests/test_benchmark.py b/tests/test_benchmark.py
@@ -6,15 +6,16 @@
 import pytest
 
 import lleaves
-from benchmarks.benchmark import NYC_used_columns
-from benchmarks.train_NYC_model import feature_enginering
 
 
 @pytest.mark.skipif(
     os.environ.get("CI", "false") == "true",
     reason="We don't want to download the datasets on CI",
 )
 def test_benchmark_datsets_correct_output():
+    from benchmarks.benchmark import NYC_used_columns
+    from benchmarks.train_NYC_model import feature_enginering
+
     model_file_NYC = "tests/models/NYC_taxi/model.txt"
     model_file_airline = "tests/models/airline/model.txt"