🤖 Lint code

Transpile-AI · Jun 15, 2023 · 459bbc9 · 459bbc9
1 parent 4512923
commit 459bbc9
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/ivy_gym_demos/optimization/optimize_policy.py b/ivy_gym_demos/optimization/optimize_policy.py
@@ -67,8 +67,7 @@ def main(
 
     # compile loss function
     compiled_loss_fn = ic.compile(
-        lambda initial_state, pol_vs:
-    loss_fn(env, initial_state, policy, pol_vs, steps)
+        lambda initial_state, pol_vs: loss_fn(env, initial_state, policy, pol_vs, steps)
     )
 
     # optimizer