mypy: Disable implicit optionals

miplearn/classifiers/adaptive.py

@@ -58,7 +58,7 @@ class AdaptiveClassifier(Classifier):
 
     def __init__(
         self,
-        candidates: Dict[str, CandidateClassifierSpecs] = None,
+        candidates: Optional[Dict[str, CandidateClassifierSpecs]] = None,
     ) -> None:
         super().__init__()
         if candidates is None:

miplearn/solvers/gurobi.py

@@ -149,9 +149,9 @@ class GurobiSolver(InternalSolver):
     def solve(
         self,
         tee: bool = False,
-        iteration_cb: IterationCallback = None,
-        lazy_cb: LazyCallback = None,
-        user_cut_cb: UserCutCallback = None,
+        iteration_cb: Optional[IterationCallback] = None,
+        lazy_cb: Optional[LazyCallback] = None,
+        user_cut_cb: Optional[UserCutCallback] = None,
     ) -> MIPSolveStats:
         self._raise_if_callback()
         assert self.model is not None

miplearn/solvers/internal.py

@@ -52,9 +52,9 @@ class InternalSolver(ABC, EnforceOverrides):
     def solve(
         self,
         tee: bool = False,
-        iteration_cb: IterationCallback = None,
-        lazy_cb: LazyCallback = None,
-        user_cut_cb: UserCutCallback = None,
+        iteration_cb: Optional[IterationCallback] = None,
+        lazy_cb: Optional[LazyCallback] = None,
+        user_cut_cb: Optional[UserCutCallback] = None,
     ) -> MIPSolveStats:
         """
         Solves the currently loaded instance. After this method finishes,

miplearn/solvers/learning.py

@@ -93,9 +93,9 @@ class LearningSolver:
 
     def __init__(
         self,
-        components: List[Component] = None,
+        components: Optional[List[Component]] = None,
         mode: str = "exact",
-        solver: InternalSolver = None,
+        solver: Optional[InternalSolver] = None,
         use_lazy_cb: bool = False,
         solve_lp: bool = True,
         simulate_perfect: bool = False,

miplearn/solvers/pyomo/base.py

@@ -88,9 +88,9 @@ class BasePyomoSolver(InternalSolver):
     def solve(
         self,
         tee: bool = False,
-        iteration_cb: IterationCallback = None,
-        lazy_cb: LazyCallback = None,
-        user_cut_cb: UserCutCallback = None,
+        iteration_cb: Optional[IterationCallback] = None,
+        lazy_cb: Optional[LazyCallback] = None,
+        user_cut_cb: Optional[UserCutCallback] = None,
     ) -> MIPSolveStats:
         if lazy_cb is not None:
             raise Exception("lazy callback not currently supported")

miplearn/solvers/pyomo/gurobi.py

@@ -28,7 +28,7 @@ class GurobiPyomoSolver(BasePyomoSolver):
 
     def __init__(
         self,
-        params: SolverParams = None,
+        params: Optional[SolverParams] = None,
     ) -> None:
         if params is None:
             params = {}

miplearn/solvers/pyomo/xpress.py

@@ -3,6 +3,7 @@
 #  Released under the modified BSD license. See COPYING.md for more details.
 
 import logging
+from typing import Optional
 
 from overrides import overrides
 from pyomo import environ as pe
@@ -25,7 +26,10 @@ class XpressPyomoSolver(BasePyomoSolver):
         {"Threads": 4} to set the number of threads.
     """
 
-    def __init__(self, params: SolverParams = None) -> None:
+    def __init__(
+        self,
+        params: Optional[SolverParams] = None,
+    ) -> None:
         if params is None:
             params = {}
         params["randomseed"] = randint(low=0, high=1000).rvs()