Use classmethod as a decorator (#481)

Because it solves warning coming from linters and type checkers.
coin-or · Sep 27, 2021 · 449e4b2 · 449e4b2
1 parent c3be1f0
commit 449e4b2
Showing 1 changed file with 7 additions and 10 deletions.
diff --git a/pulp/pulp.py b/pulp/pulp.py
@@ -311,8 +311,9 @@ def add_expression(self, e):
         self.expression = e
         self.addVariableToConstraints(e)
 
+    @classmethod
     def matrix(
-        self,
+        cls,
         name,
         indexs,
         lowBound=None,
@@ -340,10 +341,9 @@ def matrix(
                 for i in index
             ]
 
-    matrix = classmethod(matrix)
-
+    @classmethod
     def dicts(
-        self,
+        cls,
         name,
         indexs,
         lowBound=None,
@@ -386,9 +386,8 @@ def dicts(
                 )
         return d
 
-    dicts = classmethod(dicts)
-
-    def dict(self, name, indexs, lowBound=None, upBound=None, cat=const.LpContinuous):
+    @classmethod
+    def dict(cls, name, indexs, lowBound=None, upBound=None, cat=const.LpContinuous):
         if not isinstance(indexs, tuple):
             indexs = (indexs,)
         if "%" not in name:
@@ -420,11 +419,9 @@ def dict(self, name, indexs, lowBound=None, upBound=None, cat=const.LpContinuous
 
         d = {}
         for i in index:
-            d[i] = self(name % i, lowBound, upBound, cat)
+            d[i] = cls(name % i, lowBound, upBound, cat)
         return d
 
-    dict = classmethod(dict)
-
     def getLb(self):
         return self.lowBound