diff --git a/tests/common/test_pbs_error_probability_settings.py b/tests/common/test_pbs_error_probability_settings.py index 507b570af..9521fc929 100644 --- a/tests/common/test_pbs_error_probability_settings.py +++ b/tests/common/test_pbs_error_probability_settings.py @@ -4,11 +4,9 @@ import numpy import pytest -from concrete.fhe.compilation import Configuration from sklearn.exceptions import ConvergenceWarning from torch import nn -from concrete import fhe from concrete.ml.common.utils import get_model_name from concrete.ml.pytest.torch_models import FCSmall from concrete.ml.pytest.utils import sklearn_models_and_datasets diff --git a/tests/deployment/test_client_server.py b/tests/deployment/test_client_server.py index d748d39fa..e6f23ea1b 100644 --- a/tests/deployment/test_client_server.py +++ b/tests/deployment/test_client_server.py @@ -9,11 +9,9 @@ import numpy import pytest -from concrete.fhe.compilation import Configuration from sklearn.exceptions import ConvergenceWarning from torch import nn -from concrete import fhe from concrete.ml.common.utils import get_model_name from concrete.ml.deployment.fhe_client_server import FHEModelClient, FHEModelDev, FHEModelServer from concrete.ml.pytest.torch_models import FCSmall diff --git a/tests/sklearn/test_dump_onnx.py b/tests/sklearn/test_dump_onnx.py index 4967dea15..86ad97db3 100644 --- a/tests/sklearn/test_dump_onnx.py +++ b/tests/sklearn/test_dump_onnx.py @@ -9,7 +9,6 @@ import pytest from sklearn.exceptions import ConvergenceWarning -from concrete import fhe from concrete.ml.common.utils import is_model_class_in_a_list from concrete.ml.pytest.utils import get_model_name, sklearn_models_and_datasets from concrete.ml.sklearn import get_sklearn_tree_models diff --git a/tests/sklearn/test_sklearn_models.py b/tests/sklearn/test_sklearn_models.py index 76a395db1..d56ceeac9 100644 --- a/tests/sklearn/test_sklearn_models.py +++ b/tests/sklearn/test_sklearn_models.py @@ -39,7 +39,6 @@ import pandas import pytest import torch -from concrete.fhe import ParameterSelectionStrategy from sklearn.decomposition import PCA from sklearn.exceptions import ConvergenceWarning, UndefinedMetricWarning from sklearn.metrics import make_scorer, matthews_corrcoef, top_k_accuracy_score diff --git a/tests/torch/test_compile_torch.py b/tests/torch/test_compile_torch.py index d4568be01..fd3e09432 100644 --- a/tests/torch/test_compile_torch.py +++ b/tests/torch/test_compile_torch.py @@ -13,7 +13,6 @@ import pytest import torch import torch.quantization -from concrete.fhe import ParameterSelectionStrategy # pylint: disable=ungrouped-imports from torch import nn from concrete.ml.common.utils import manage_parameters_for_pbs_errors, to_tuple