Formatting

* Adding black as dev dependency
* Formatting pina code
* Formatting tests

pina/optim/__init__.py

@@ -8,4 +8,4 @@ __all__ = [
 from .optimizer_interface import Optimizer
 from .torch_optimizer import TorchOptimizer
 from .scheduler_interface import Scheduler
-from .torch_scheduler import TorchScheduler
+from .torch_scheduler import TorchScheduler

pina/optim/optimizer_interface.py

@@ -1,4 +1,4 @@
-""" Module for PINA Optimizer """
+"""Module for PINA Optimizer"""
 
 from abc import ABCMeta, abstractmethod
 
@@ -12,4 +12,4 @@ class Optimizer(metaclass=ABCMeta):  # TODO improve interface
 
     @abstractmethod
     def hook(self):
-        pass
+        pass

pina/optim/scheduler_interface.py

@@ -1,9 +1,9 @@
-""" Module for PINA Optimizer """
+"""Module for PINA Optimizer"""
 
 from abc import ABCMeta, abstractmethod
 
 
-class Scheduler(metaclass=ABCMeta): # TODO improve interface
+class Scheduler(metaclass=ABCMeta):  # TODO improve interface
 
     @property
     @abstractmethod
@@ -12,4 +12,4 @@ class Scheduler(metaclass=ABCMeta): # TODO improve interface
 
     @abstractmethod
     def hook(self):
-        pass
+        pass

pina/optim/torch_optimizer.py

@@ -1,4 +1,4 @@
-""" Module for PINA Torch Optimizer """
+"""Module for PINA Torch Optimizer"""
 
 import torch
 
@@ -16,8 +16,10 @@ class TorchOptimizer(Optimizer):
         self._optimizer_instance = None
 
     def hook(self, parameters):
-        self._optimizer_instance = self.optimizer_class(parameters,
-                                                       **self.kwargs)
+        self._optimizer_instance = self.optimizer_class(
+            parameters, **self.kwargs
+        )
+
     @property
     def instance(self):
         """

pina/optim/torch_scheduler.py

@@ -1,11 +1,13 @@
-""" Module for PINA Torch Optimizer """
+"""Module for PINA Torch Optimizer"""
 
 import torch
+
 try:
     from torch.optim.lr_scheduler import LRScheduler  # torch >= 2.0
 except ImportError:
     from torch.optim.lr_scheduler import (
-        _LRScheduler as LRScheduler, )  # torch < 2.0
+        _LRScheduler as LRScheduler,
+    )  # torch < 2.0
 
 from ..utils import check_consistency
 from .optimizer_interface import Optimizer
@@ -24,7 +26,8 @@ class TorchScheduler(Scheduler):
     def hook(self, optimizer):
         check_consistency(optimizer, Optimizer)
         self._scheduler_instance = self.scheduler_class(
-            optimizer.instance, **self.kwargs)
+            optimizer.instance, **self.kwargs
+        )
 
     @property
     def instance(self):