Optimize imports

tests/classifiers/test_cv.py

@@ -6,8 +6,8 @@ import numpy as np
 from numpy.linalg import norm
 from sklearn.svm import SVC
 
-from miplearn.classifiers.sklearn import ScikitLearnClassifier
 from miplearn.classifiers.cv import CrossValidatedClassifier
+from miplearn.classifiers.sklearn import ScikitLearnClassifier
 from tests.classifiers import _build_circle_training_data
 
 E = 0.1

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

tests/solvers/test_learning_solver.py

@@ -3,10 +3,10 @@
 #  Released under the modified BSD license. See COPYING.md for more details.
 
 import logging
-import dill
-import pickle
-import tempfile
 import os
+import tempfile
+
+import dill
 
 from miplearn.instance import PickleGzInstance, write_pickle_gz, read_pickle_gz
 from miplearn.solvers.gurobi import GurobiSolver

tests/test_benchmark.py

@@ -4,10 +4,10 @@
 
 import os.path
 
-from miplearn.benchmark import BenchmarkRunner
-from miplearn.problems.stab import MaxWeightStableSetGenerator
 from scipy.stats import randint
 
+from miplearn.benchmark import BenchmarkRunner
+from miplearn.problems.stab import MaxWeightStableSetGenerator
 from miplearn.solvers.learning import LearningSolver