Optimize imports

miplearn/__init__.py

@@ -7,18 +7,18 @@ from .classifiers import (
     Classifier,
     Regressor,
 )
+from .classifiers.adaptive import AdaptiveClassifier
 from .classifiers.sklearn import (
     ScikitLearnRegressor,
     ScikitLearnClassifier,
 )
-from .classifiers.adaptive import AdaptiveClassifier
 from .classifiers.threshold import MinPrecisionThreshold
 from .components.component import Component
 from .components.dynamic_lazy import DynamicLazyConstraintsComponent
 from .components.dynamic_user_cuts import UserCutsComponent
-from .components.static_lazy import StaticLazyConstraintsComponent
 from .components.objective import ObjectiveValueComponent
 from .components.primal import PrimalSolutionComponent
+from .components.static_lazy import StaticLazyConstraintsComponent
 from .components.steps.convert_tight import ConvertTightIneqsIntoEqsStep
 from .components.steps.drop_redundant import DropRedundantInequalitiesStep
 from .components.steps.relax_integrality import RelaxIntegralityStep

miplearn/benchmark.py

@@ -4,7 +4,7 @@
 
 import logging
 import os
-from typing import Dict, Union, List
+from typing import Dict, List
 
 import pandas as pd
 

miplearn/classifiers/adaptive.py

@@ -3,8 +3,7 @@
 #  Released under the modified BSD license. See COPYING.md for more details.
 
 import logging
-from copy import deepcopy
-from typing import Dict, Callable, Optional
+from typing import Dict, Optional
 
 import numpy as np
 from sklearn.linear_model import LogisticRegression
@@ -14,8 +13,8 @@ from sklearn.pipeline import make_pipeline
 from sklearn.preprocessing import StandardScaler
 
 from miplearn.classifiers import Classifier
-from miplearn.classifiers.sklearn import ScikitLearnClassifier
 from miplearn.classifiers.counting import CountingClassifier
+from miplearn.classifiers.sklearn import ScikitLearnClassifier
 
 logger = logging.getLogger(__name__)
 

miplearn/classifiers/cv.py

@@ -3,7 +3,7 @@
 #  Released under the modified BSD license. See COPYING.md for more details.
 
 import logging
-from typing import Optional, Callable, List
+from typing import Optional, List
 
 import numpy as np
 from sklearn.dummy import DummyClassifier

miplearn/components/component.py

@@ -6,9 +6,9 @@ from typing import Any, List, TYPE_CHECKING, Tuple, Dict, Hashable
 
 import numpy as np
 
+from miplearn.features import TrainingSample, Features
 from miplearn.instance import Instance
 from miplearn.types import LearningSolveStats
-from miplearn.features import TrainingSample, Features
 
 if TYPE_CHECKING:
     from miplearn.solvers.learning import LearningSolver

miplearn/components/objective.py

@@ -11,9 +11,9 @@ from sklearn.linear_model import LinearRegression
 from miplearn.classifiers import Regressor
 from miplearn.classifiers.sklearn import ScikitLearnRegressor
 from miplearn.components.component import Component
+from miplearn.features import TrainingSample, Features
 from miplearn.instance import Instance
 from miplearn.types import LearningSolveStats
-from miplearn.features import TrainingSample, Features
 
 if TYPE_CHECKING:
     from miplearn.solvers.learning import LearningSolver

miplearn/components/primal.py

@@ -20,12 +20,12 @@ from miplearn.classifiers.adaptive import AdaptiveClassifier
 from miplearn.classifiers.threshold import MinPrecisionThreshold, Threshold
 from miplearn.components import classifier_evaluation_dict
 from miplearn.components.component import Component
+from miplearn.features import TrainingSample, Features
 from miplearn.instance import Instance
 from miplearn.types import (
     Solution,
     LearningSolveStats,
 )
-from miplearn.features import TrainingSample, Features
 
 logger = logging.getLogger(__name__)
 

miplearn/instance.py

@@ -9,8 +9,8 @@ import pickle
 from abc import ABC, abstractmethod
 from typing import Any, List, Optional, Hashable, IO, cast
 
-from miplearn.types import VarIndex
 from miplearn.features import TrainingSample, Features
+from miplearn.types import VarIndex
 
 logger = logging.getLogger(__name__)
 

miplearn/log.py

@@ -5,8 +5,8 @@
 import logging
 import sys
 import time
-import warnings
 import traceback
+import warnings
 
 _formatwarning = warnings.formatwarning
 

miplearn/solvers/internal.py

@@ -4,7 +4,7 @@
 
 import logging
 from abc import ABC, abstractmethod
-from typing import Any, Dict, List, Optional, Tuple
+from typing import Any, Dict, List, Optional
 
 from miplearn.instance import Instance
 from miplearn.types import (

miplearn/types.py

@@ -7,6 +7,7 @@ from typing import Optional, Dict, Callable, Any, Union, Tuple, TYPE_CHECKING
 from mypy_extensions import TypedDict
 
 if TYPE_CHECKING:
+    # noinspection PyUnresolvedReferences
     from miplearn.solvers.learning import InternalSolver
 
 VarIndex = Union[str, int, Tuple[Union[str, int]]]