diff --git a/tests/distributions/test_base.py b/tests/distributions/test_base.py index 6c824c6..911d01a 100644 --- a/tests/distributions/test_base.py +++ b/tests/distributions/test_base.py @@ -3,6 +3,7 @@ import pytest import numpy as np + from xgboost_distribution.distributions import ( AVAILABLE_DISTRIBUTIONS, Normal, diff --git a/tests/distributions/test_exponential.py b/tests/distributions/test_exponential.py index 784c711..f8e26c5 100644 --- a/tests/distributions/test_exponential.py +++ b/tests/distributions/test_exponential.py @@ -2,6 +2,7 @@ import numpy as np import pandas as pd + from xgboost_distribution.distributions import Exponential diff --git a/tests/distributions/test_laplace.py b/tests/distributions/test_laplace.py index be9c027..7fd687b 100644 --- a/tests/distributions/test_laplace.py +++ b/tests/distributions/test_laplace.py @@ -1,6 +1,7 @@ import pytest import numpy as np + from xgboost_distribution.distributions import Laplace diff --git a/tests/distributions/test_log_normal.py b/tests/distributions/test_log_normal.py index a516586..c903fdf 100644 --- a/tests/distributions/test_log_normal.py +++ b/tests/distributions/test_log_normal.py @@ -2,6 +2,7 @@ import numpy as np import pandas as pd + from xgboost_distribution.distributions import LogNormal diff --git a/tests/distributions/test_negative_binomial.py b/tests/distributions/test_negative_binomial.py index 05ec9e2..6fcd0d7 100644 --- a/tests/distributions/test_negative_binomial.py +++ b/tests/distributions/test_negative_binomial.py @@ -3,6 +3,7 @@ import numpy as np import pandas as pd from scipy.special import logit + from xgboost_distribution.distributions import NegativeBinomial diff --git a/tests/distributions/test_normal.py b/tests/distributions/test_normal.py index f5044e1..149c9b5 100644 --- a/tests/distributions/test_normal.py +++ b/tests/distributions/test_normal.py @@ -1,6 +1,7 @@ import pytest import numpy as np + from xgboost_distribution.distributions import Normal diff --git a/tests/distributions/test_poisson.py b/tests/distributions/test_poisson.py index 12712d6..c93998e 100644 --- a/tests/distributions/test_poisson.py +++ b/tests/distributions/test_poisson.py @@ -2,6 +2,7 @@ import numpy as np import pandas as pd + from xgboost_distribution.distributions import Poisson diff --git a/tests/distributions/test_utils.py b/tests/distributions/test_utils.py index 88bffc6..ca0301a 100644 --- a/tests/distributions/test_utils.py +++ b/tests/distributions/test_utils.py @@ -2,6 +2,7 @@ import numpy as np import pandas as pd + from xgboost_distribution.distributions.utils import ( check_all_ge_zero, check_all_gt_zero, diff --git a/tests/test_data.py b/tests/test_data.py index 155a501..646722f 100644 --- a/tests/test_data.py +++ b/tests/test_data.py @@ -3,6 +3,7 @@ import pytest import numpy as np + from xgboost_distribution.model import XGBDistribution diff --git a/tests/test_model.py b/tests/test_model.py index c37b3da..ff33ef0 100644 --- a/tests/test_model.py +++ b/tests/test_model.py @@ -8,6 +8,7 @@ import joblib import numpy as np from sklearn.exceptions import NotFittedError + from xgboost_distribution.distributions import AVAILABLE_DISTRIBUTIONS from xgboost_distribution.model import XGBDistribution