torch-rb 0.14.0 → 0.14.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 844ea56ce58c08b3c2fc3b93d39755cedfc1ded9ac60a8e88ebdd252979032c2
-  data.tar.gz: fe0fb4f66a541412e0af073c149e101dc8d0d3d5855b96fc2ee42d864a18866d
+  metadata.gz: 998c7844d24482a76c6bb1a7a60c508bc4f2946688d7a4a3801a9c832bf54379
+  data.tar.gz: 2cdd9b8832159282976a5c8c44098fa1c8501b54cc4e499e908aaf3d9016061e
 SHA512:
-  metadata.gz: 4828aad90774777a00cc552ea0c524a4b3a810cc501727f91e5efc14242027b73da226c80bbc67bd48bfa8e826c2103997b5f1787b93fb588493f1290f1b0d12
-  data.tar.gz: 4721a8c989af6decd7599c802fa2a89214332f924928cb8a138e70744d5cb7198355a1839d584580864e3b299c53d7d54f4984d349885b9f1b916734a12bc219
+  metadata.gz: d5f536c8d0ce11a2a885c74e4a99211bff50716f80e1e9da69040f7899ca02a6a6e663e085188e89f1e85238893dafa1552422dbb844144f08c3a56a44084bc4
+  data.tar.gz: 455a356eb60204a02d6fab0041b5fd1ad9ac26e8eefcb383be42dd5a0d4c026586d5c46749e919c0f471708e897265f48c37b5683e86b86d225f957acd6a7049

data/CHANGELOG.md

@@ -1,3 +1,7 @@
+## 0.14.1 (2023-12-26)
+
+- Fixed default arguments for `conv1d`
+
 ## 0.14.0 (2023-11-09)
 
 - Updated LibTorch to 2.1.0

data/ext/torch/ruby_arg_parser.cpp

@@ -401,7 +401,7 @@ void FunctionParameter::set_default_str(const std::string& str) {
     if (str != "None") {
       throw std::runtime_error("default value for Tensor must be none, got: " + str);
     }
-  } else if (type_ == ParameterType::INT64) {
+  } else if (type_ == ParameterType::INT64 || type_ == ParameterType::SYM_INT) {
     default_int = atol(str.c_str());
   } else if (type_ == ParameterType::BOOL) {
     default_bool = (str == "True" || str == "true");
@@ -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);
     }
@@ -452,6 +452,31 @@ void FunctionParameter::set_default_str(const std::string& str) {
       default_string = parse_string_literal(str);
     }
   }
+  // These types weren't handled here before. Adding a default error
+  // led to a lot of test failures so adding this skip for now.
+  // We should correctly handle these though because it might be causing
+  // silent failures.
+  else if (type_ == ParameterType::TENSOR_LIST) {
+    // throw std::runtime_error("Invalid Tensor List");
+  } else if (type_ == ParameterType::GENERATOR) {
+    // throw std::runtime_error("ParameterType::GENERATOR");
+  } else if (type_ == ParameterType::PYOBJECT) {
+    // throw std::runtime_error("ParameterType::PYOBJECT");
+  } else if (type_ == ParameterType::MEMORY_FORMAT) {
+    // throw std::runtime_error("ParameterType::MEMORY_FORMAT");
+  } else if (type_ == ParameterType::DIMNAME) {
+    // throw std::runtime_error("ParameterType::DIMNAME");
+  } else if (type_ == ParameterType::DIMNAME_LIST) {
+    // throw std::runtime_error("ParameterType::DIMNAME_LIST");
+  } else if (type_ == ParameterType::SCALAR_LIST) {
+    // throw std::runtime_error("ParameterType::SCALAR_LIST");
+  } else if (type_ == ParameterType::STORAGE) {
+    // throw std::runtime_error("ParameterType::STORAGE");
+  } else if (type_ == ParameterType::QSCHEME) {
+    // throw std::runtime_error("ParameterType::QSCHEME");
+  } else {
+    throw std::runtime_error("unknown parameter type");
+  }
 }
 
 FunctionSignature::FunctionSignature(const std::string& fmt, int index)

data/lib/torch/version.rb

@@ -1,3 +1,3 @@
 module Torch
-  VERSION = "0.14.0"
+  VERSION = "0.14.1"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: torch-rb
 version: !ruby/object:Gem::Version
-  version: 0.14.0
+  version: 0.14.1
 platform: ruby
 authors:
 - Andrew Kane
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2023-11-09 00:00:00.000000000 Z
+date: 2023-12-27 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rice
@@ -237,7 +237,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.4.10
+rubygems_version: 3.5.3
 signing_key:
 specification_version: 4
 summary: Deep learning for Ruby, powered by LibTorch