From aae2a7f06438707f3133a66ee42cab9680e15deb Mon Sep 17 00:00:00 2001
From: Simon Boehm <simon@siboehm.com>
Date: Sat, 26 Jun 2021 12:14:21 +0200
Subject: [PATCH] Inline benchmark test imports

---
 tests/test_benchmark.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/tests/test_benchmark.py b/tests/test_benchmark.py
index d558d12..b2e7da6 100644
--- a/tests/test_benchmark.py
+++ b/tests/test_benchmark.py
@@ -6,8 +6,6 @@
 import pytest
 
 import lleaves
-from benchmarks.benchmark import NYC_used_columns
-from benchmarks.train_NYC_model import feature_enginering
 
 
 @pytest.mark.skipif(
@@ -15,6 +13,9 @@
     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"