Skip to content

Commit

Permalink
Change action_dim to action_shape (#602)
Browse files Browse the repository at this point in the history
Noticed that in IQN and FQF there were some mismatches in the docstrings. Figured I would make a pull request to make it match.
  • Loading branch information
Squeemos authored Apr 22, 2022
1 parent 57ecebd commit e01385e
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions tianshou/utils/net/discrete.py
Original file line number Diff line number Diff line change
Expand Up @@ -160,7 +160,7 @@ class ImplicitQuantileNetwork(Critic):
:param preprocess_net: a self-defined preprocess_net which output a
flattened hidden state.
:param int action_dim: the dimension of action space.
:param int action_shape: a sequence of int for the shape of action.
:param hidden_sizes: a sequence of int for constructing the MLP after
preprocess_net. Default to empty sequence (where the MLP now contains
only a single linear layer).
Expand Down Expand Up @@ -254,7 +254,7 @@ class FullQuantileFunction(ImplicitQuantileNetwork):
:param preprocess_net: a self-defined preprocess_net which output a
flattened hidden state.
:param int action_dim: the dimension of action space.
:param int action_shape: a sequence of int for the shape of action.
:param hidden_sizes: a sequence of int for constructing the MLP after
preprocess_net. Default to empty sequence (where the MLP now contains
only a single linear layer).
Expand Down

0 comments on commit e01385e

Please sign in to comment.