Merge pull request #197 from MannLabs/fix-lr-scheduler-docstring

fix return type in docstring

peptdeep/model/model_interface.py

@@ -49,13 +49,13 @@ def step(self, epoch: int, loss: float):
         """
         raise NotImplementedError
 
-    def get_last_lr(self) -> float:
+    def get_last_lr(self) -> List[float]:
         """
         Get the last learning rate.
 
         Returns
         -------
-        float
+        List[float]
             The last learning rate.
         """
         raise NotImplementedError
@@ -91,13 +91,13 @@ def step(self, epoch: int = None, loss=None):
         """
         return self.lambda_lr.step()
 
-    def get_last_lr(self) -> float:
+    def get_last_lr(self) -> List[float]:
         """
         Get the last learning rate.
 
         Returns
         -------
-        float
+        List[float]
             The last learning rate.
         """
         return self.lambda_lr.get_last_lr()