Fixed default arguments for conv1d

ankane · Dec 27, 2023 · 33ef89a · 33ef89a
1 parent a18f3f5
commit 33ef89a
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,3 +1,7 @@
+## 0.14.1 (unreleased)
+
+- Fixed default arguments for `conv1d`
+
 ## 0.14.0 (2023-11-09)
 
 - Updated LibTorch to 2.1.0

diff --git a/ext/torch/ruby_arg_parser.cpp b/ext/torch/ruby_arg_parser.cpp
@@ -417,7 +417,7 @@ void FunctionParameter::set_default_str(const std::string& str) {
       default_scalar = as_integer.has_value() ? at::Scalar(as_integer.value()) :
                                                 at::Scalar(atof(str.c_str()));
     }
-  } else if (type_ == ParameterType::INT_LIST) {
+  } else if (type_ == ParameterType::INT_LIST || type_ == ParameterType::SYM_INT_LIST) {
     if (str != "None") {
       default_intlist = parse_intlist_args(str, size);
     }