using ValueType instead of RuntimeType

pytorch · Sep 9, 2023 · f08afa1 · f08afa1
1 parent e781c33
commit f08afa1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/torch/testing/_internal/common_modules.py b/torch/testing/_internal/common_modules.py
@@ -2530,13 +2530,13 @@ def module_error_inputs_torch_nn_RNN_GRU(module_info, device, dtype, requires_gr
         ErrorModuleInput(
             ModuleInput(constructor_input=FunctionInput(10, 0, 1)),
             error_on=ModuleErrorEnum.FORWARD_ERROR,
-            error_type=RuntimeError,
+            error_type=ValueError,
             error_regex="hidden_size must be greater than zero"
         ),
         ErrorModuleInput(
             ModuleInput(constructor_input=FunctionInput(10, 10, 0)),
             error_on=ModuleErrorEnum.FORWARD_ERROR,
-            error_type=RuntimeError,
+            error_type=ValueError,
             error_regex="num_layers must be greater than zero"
         ),
     ]