Merge pull request #252 from allenai/rev_to_0.2.1

Updates for 0.2.1.
allenai · Feb 17, 2021 · 49297d4 · 49297d4
2 parents c3ff52c + 8c80f93
commit 49297d4
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/.VERSION b/.VERSION
@@ -1 +1 @@
-0.2.1a0
+0.2.1
diff --git a/allenact/algorithms/onpolicy_sync/engine.py b/allenact/algorithms/onpolicy_sync/engine.py
@@ -411,12 +411,12 @@ def act(self, rollouts: RolloutStorage):
                 rollouts.masks[rollouts.step : rollouts.step + 1],
             )
 
-        # Assume actions do not contain a step dimension
-        actions = (
-            actor_critic_output.distributions.sample()
-            if not self.deterministic_agents
-            else actor_critic_output.distributions.mode()
-        )
+            # Assume actions do not contain a step dimension
+            actions = (
+                actor_critic_output.distributions.sample()
+                if not self.deterministic_agents
+                else actor_critic_output.distributions.mode()
+            )
 
         return actions, actor_critic_output, memory, step_observation