torch-rb 0.25.0 → 0.27.0

data/ext/torch/utils.h

@@ -8,7 +8,7 @@
 #include <rice/stl.hpp>
 
 static_assert(
-  TORCH_VERSION_MAJOR == 2 && TORCH_VERSION_MINOR == 13,
+  TORCH_VERSION_MAJOR == 2 && TORCH_VERSION_MINOR == 15,
   "Incompatible LibTorch version"
 );
 

data/lib/torch/nn/module.rb

@@ -337,7 +337,7 @@ module Torch
           if vars.any?
             vars.map(&:inspect)
           else
-            options.map { |k, v| [k, v.inspect] }.to_h
+            options.to_h { |k, v| [k, v.inspect] }
           end
         str % vars
       end
@@ -346,7 +346,7 @@ module Torch
       # remove tensors for performance
       # so we can skip call to inspect
       def dict
-        instance_variables.reject { |k| instance_variable_get(k).is_a?(Tensor) }.map { |k| [k[1..-1].to_sym, instance_variable_get(k)] }.to_h
+        instance_variables.reject { |k| instance_variable_get(k).is_a?(Tensor) }.to_h { |k| [k[1..-1].to_sym, instance_variable_get(k)] }
       end
 
       def load_from_state_dict(state_dict, prefix, local_metadata, strict, missing_keys, unexpected_keys, error_msgs)

data/lib/torch/optim/lr_scheduler/multi_step_lr.rb

@@ -3,7 +3,7 @@ module Torch
     module LRScheduler
       class MultiStepLR < LRScheduler
         def initialize(optimizer, milestones, gamma: 0.1, last_epoch: -1)
-          @milestones = milestones.map.with_index.map { |v, i| [v, i + 1] }.to_h
+          @milestones = milestones.map.with_index.to_h { |v, i| [v, i + 1] }
           @gamma = gamma
           super(optimizer, last_epoch)
         end

data/lib/torch/optim/optimizer.rb

@@ -35,13 +35,13 @@ module Torch
         raise NotImplementedYet
 
         pack_group = lambda do |group|
-          packed = group.select { |k, _| k != :params }.map { |k, v| [k.to_s, v] }.to_h
+          packed = group.select { |k, _| k != :params }.to_h { |k, v| [k.to_s, v] }
           packed["params"] = group[:params].map { |p| p.object_id }
           packed
         end
 
         param_groups = @param_groups.map { |g| pack_group.call(g) }
-        packed_state = @state.map { |k, v| [k.is_a?(Tensor) ? k.object_id : k, v] }.to_h
+        packed_state = @state.to_h { |k, v| [k.is_a?(Tensor) ? k.object_id : k, v] }
 
         {
           "state" => packed_state,

data/lib/torch/version.rb

@@ -1,3 +1,3 @@
 module Torch
-  VERSION = "0.25.0"
+  VERSION = "0.27.0"
 end

data/lib/torch.rb

@@ -272,7 +272,7 @@ module Torch
     qint32: 14,
     bfloat16: 15
   }
-  ENUM_TO_DTYPE = DTYPE_TO_ENUM.map(&:reverse).to_h
+  ENUM_TO_DTYPE = DTYPE_TO_ENUM.invert
 
   TENSOR_TYPE_CLASSES = []
 
@@ -408,7 +408,7 @@ module Torch
 
     def tensor(data, **options)
       if options[:dtype].nil? && defined?(Numo::NArray) && data.is_a?(Numo::NArray)
-        numo_to_dtype = _dtype_to_numo.map(&:reverse).to_h
+        numo_to_dtype = _dtype_to_numo.invert
         options[:dtype] = numo_to_dtype[data.class]
       end
 

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: torch-rb
 version: !ruby/object:Gem::Version
-  version: 0.25.0
+  version: 0.27.0
 platform: ruby
 authors:
 - Andrew Kane
@@ -241,7 +241,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 4.0.14
+rubygems_version: 4.0.20
 specification_version: 4
 summary: Deep learning for Ruby, powered by LibTorch
 test_files: []