diff --git a/quantile_forest/tests/test_quantile_forest.py b/quantile_forest/tests/test_quantile_forest.py index 2b04af5..ce4bde0 100755 --- a/quantile_forest/tests/test_quantile_forest.py +++ b/quantile_forest/tests/test_quantile_forest.py @@ -4,7 +4,7 @@ import math import warnings -from typing import Any +from typing import Any, Dict import numpy as np import pytest @@ -41,7 +41,7 @@ X_california = california.data[perm] y_california = california.target[perm] -FOREST_REGRESSORS: dict[str, Any] = { +FOREST_REGRESSORS: Dict[str, Any] = { "ExtraTreesQuantileRegressor": ExtraTreesQuantileRegressor, "RandomForestQuantileRegressor": RandomForestQuantileRegressor, }