Formatting
* Adding black as dev dependency * Formatting pina code * Formatting tests
This commit is contained in:
committed by
Nicola Demo
parent
4c4482b155
commit
42ab1a666b
@@ -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):
|
||||
"""
|
||||
|
||||
Reference in New Issue
Block a user