From e01385ea30d21b7efc3f9377d22d83e6afd39e2f Mon Sep 17 00:00:00 2001
From: Squeemos <56849503+Squeemos@users.noreply.github.com>
Date: Thu, 21 Apr 2022 17:09:57 -0700
Subject: [PATCH] Change action_dim to action_shape (#602)

Noticed that in IQN and FQF there were some mismatches in the docstrings. Figured I would make a pull request to make it match.
---
 tianshou/utils/net/discrete.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tianshou/utils/net/discrete.py b/tianshou/utils/net/discrete.py
index a81e1c6..51d19b7 100644
--- a/tianshou/utils/net/discrete.py
+++ b/tianshou/utils/net/discrete.py
@@ -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).
@@ -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).