amend

facebookresearch · Jun 10, 2024 · b7dc251 · b7dc251
1 parent 808b58a
commit b7dc251
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/benchmarl/experiment/experiment.py b/benchmarl/experiment/experiment.py
@@ -466,7 +466,9 @@ def _setup_collector(self):
                 frames_per_batch=self.config.collected_frames_per_batch(self.on_policy),
                 total_frames=self.config.get_max_n_frames(self.on_policy),
                 init_random_frames=(
-                    self.config.off_policy_init_random_frames if not self.on_policy else 0
+                    self.config.off_policy_init_random_frames
+                    if not self.on_policy
+                    else 0
                 ),
             )
         else: