Type import fix

automl · Dec 8, 2021 · 84b01d3 · 84b01d3
1 parent 097b682
commit 84b01d3
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/test/test_pipeline/components/regression/test_base.py b/test/test_pipeline/components/regression/test_base.py
@@ -1,3 +1,5 @@
+from typing import Type
+
 import unittest
 
 import pytest
@@ -340,7 +342,7 @@ def test_fit_and_predict_with_1d_targets_as_1d(
 @pytest.mark.parametrize("X", [np.array([[1, 2, 3]] * 20)])
 @pytest.mark.parametrize("y", [np.array([[1]] * 20)])
 def test_fit_and_predict_with_1d_targets_as_2d(
-    regressor: Type[ RegressorChoice ],
+    regressor: Type[RegressorChoice],
     X: np.ndarray,
     y: np.ndarray
 ):