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