Update sb3_imitation.py - fix quality

Fix quality.
edbeeching · Jul 11, 2024 · c343cb6 · c343cb6
1 parent 9cd545c
commit c343cb6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/examples/sb3_imitation.py b/examples/sb3_imitation.py
@@ -177,7 +177,7 @@ def close_env():
     policy_kwargs=policy_kwargs,
     verbose=2,
     tensorboard_log=f"logs/{args.experiment_name}",
-    device="cpu"
+    device="cpu",
     # seed=args.seed // Not currently supported as stable_baselines_wrapper.py seed() method is not yet implemented.
 )
 
@@ -191,7 +191,7 @@ def close_env():
             rng=rng,
             policy=learner.policy,
             custom_logger=logger,
-            device="cpu"
+            device="cpu",
         )
         print("Starting Imitation Learning Training using BC:")
         bc_trainer.train(n_epochs=args.bc_epochs)