add timeLimit arg to __init__ (coin-or#576)

pulp/apis/mosek_api.py

@@ -28,6 +28,8 @@
 from .. import constants
 import sys
 
+from typing import Optional
+
 
 class MOSEK(LpSolver):
     """Mosek lp and mip solver (via Mosek Optimizer API)."""
@@ -54,7 +56,8 @@ def __init__(
             self,
             mip=True,
             msg=True,
-            options={},
+            timeLimit: Optional[float] = None,
+            options: Optional[dict] = None,
             task_file_name="",
             sol_type=mosek.soltype.bas,
         ):
@@ -66,6 +69,8 @@ def __init__(
 
             @param msg: Enable Mosek log output.
 
+            @param float timeLimit: maximum time for solver (in seconds)
+
             @param options: Accepts a dictionary of Mosek solver parameters. Ignore to
                             use default parameter values. Eg: options = {mosek.dparam.mio_max_time:30}
                             sets the maximum time spent by the Mixed Integer optimizer to 30 seconds.
@@ -85,9 +90,19 @@ def __init__(
             """
             self.mip = mip
             self.msg = msg
+            self.timeLimit = timeLimit
             self.task_file_name = task_file_name
             self.solution_type = sol_type
+            if options is None:
+                options = {}
             self.options = options
+            if self.timeLimit is not None:
+                timeLimit_keys = {"MSK_DPAR_MIO_MAX_TIME", mosek.dparam.mio_max_time}
+                if not timeLimit_keys.isdisjoint(self.options.keys()):
+                    raise ValueError(
+                        "timeLimit parameter has been provided trough `timeLimit` and `options`."
+                    )
+                self.options["MSK_DPAR_MIO_MAX_TIME"] = self.timeLimit
 
         def available(self):
             """True if Mosek is available."""