action_spec to model

facebookresearch · Oct 21, 2023 · 2b7ec4b · 2b7ec4b
1 parent 811a0a3
commit 2b7ec4b
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/examples/extending/model/custom_model.py b/examples/extending/model/custom_model.py
@@ -39,6 +39,7 @@ def __init__(
         # And access some of the ones already available to your module
         _ = self.input_spec  # Like its input_spec
         _ = self.output_spec  # or output_spec
+        _ = self.action_spec  # the action spec of the env
         _ = self.agent_group  # the name of the agent group the model is for
         _ = self.n_agents  # or the number of agents this module is for