Optimize imports

tests/components/steps/test_drop_redundant.py

@@ -9,11 +9,11 @@ import numpy as np
 from miplearn.classifiers import Classifier
 from miplearn.components.steps.drop_redundant import DropRedundantInequalitiesStep
 from miplearn.components.steps.relax_integrality import RelaxIntegralityStep
+from miplearn.features import TrainingSample, Features
 from miplearn.instance import Instance
 from miplearn.solvers.gurobi import GurobiSolver
 from miplearn.solvers.internal import InternalSolver
 from miplearn.solvers.learning import LearningSolver
-from miplearn.features import TrainingSample, Features
 from tests.fixtures.infeasible import get_infeasible_instance
 from tests.fixtures.redundant import get_instance_with_redundancy
 

tests/components/test_objective.py

@@ -4,6 +4,7 @@
 from typing import Hashable, Dict
 from unittest.mock import Mock
 
+import numpy as np
 import pytest
 from numpy.testing import assert_array_equal
 
@@ -15,8 +16,6 @@ from miplearn.solvers.learning import LearningSolver
 from miplearn.solvers.pyomo.gurobi import GurobiPyomoSolver
 from tests.fixtures.knapsack import get_knapsack_instance
 
-import numpy as np
-
 
 @pytest.fixture
 def instance(features: Features) -> Instance:

tests/components/test_primal.py

@@ -1,7 +1,6 @@
 #  MIPLearn: Extensible Framework for Learning-Enhanced Mixed-Integer Optimization
 #  Copyright (C) 2020-2021, UChicago Argonne, LLC. All rights reserved.
 #  Released under the modified BSD license. See COPYING.md for more details.
-from typing import Dict
 from unittest.mock import Mock
 
 import numpy as np
@@ -12,9 +11,9 @@ from miplearn.classifiers import Classifier
 from miplearn.classifiers.threshold import Threshold
 from miplearn.components import classifier_evaluation_dict
 from miplearn.components.primal import PrimalSolutionComponent
+from miplearn.features import TrainingSample, VariableFeatures, Features
 from miplearn.instance import Instance
 from miplearn.problems.tsp import TravelingSalesmanGenerator
-from miplearn.features import TrainingSample, VariableFeatures, Features
 from miplearn.solvers.learning import LearningSolver
 
 

tests/components/test_static_lazy.py

@@ -11,18 +11,18 @@ from numpy.testing import assert_array_equal
 from miplearn.classifiers import Classifier
 from miplearn.classifiers.threshold import Threshold, MinProbabilityThreshold
 from miplearn.components.static_lazy import StaticLazyConstraintsComponent
-from miplearn.instance import Instance
-from miplearn.solvers.internal import InternalSolver
-from miplearn.solvers.learning import LearningSolver
-from miplearn.types import (
-    LearningSolveStats,
-)
 from miplearn.features import (
     TrainingSample,
     InstanceFeatures,
     ConstraintFeatures,
     Features,
 )
+from miplearn.instance import Instance
+from miplearn.solvers.internal import InternalSolver
+from miplearn.solvers.learning import LearningSolver
+from miplearn.types import (
+    LearningSolveStats,
+)
 
 
 @pytest.fixture