Black formatting in gradient_based.py

Happy-Algorithms-League · May 7, 2020 · 140da25 · 140da25
1 parent 5dd8a47
commit 140da25
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/gp/local_search/gradient_based.py b/gp/local_search/gradient_based.py
@@ -15,9 +15,14 @@
 from ..individual import Individual  # noqa: F401
 
 
-def gradient_based(individuals: List[Individual], objective: Callable,
-                   lr: float, gradient_steps: int, optimizer: torch.optim.Optimizer =
-                   None, clip_value: Optional[float] = None):
+def gradient_based(
+    individuals: List[Individual],
+    objective: Callable,
+    lr: float,
+    gradient_steps: int,
+    optimizer: torch.optim.Optimizer = None,
+    clip_value: Optional[float] = None,
+):
     """Perform a local search for numeric leaf values for the list of
     individuals based on gradient information obtained via automatic
     differentiation.