torch-rb 0.23.0 → 0.24.0

data/lib/torch/hub.rb

@@ -6,37 +6,17 @@ module Torch
       end
 
       def download_url_to_file(url, dst)
-        uri = URI(url)
-        tmp = nil
-        location = nil
+        require "open-uri"
 
-        puts "Downloading #{url}..."
-        Net::HTTP.start(uri.host, uri.port, use_ssl: uri.scheme == "https") do |http|
-          request = Net::HTTP::Get.new(uri)
-
-          http.request(request) do |response|
-            case response
-            when Net::HTTPRedirection
-              location = response["location"]
-            when Net::HTTPSuccess
-              tmp = "#{Dir.tmpdir}/#{Time.now.to_f}" # TODO better name
-              File.open(tmp, "wb") do |f|
-                response.read_body do |chunk|
-                  f.write(chunk)
-                end
-              end
-            else
-              raise Error, "Bad response"
-            end
-          end
-        end
+        uri = URI.parse(url)
+        raise "Invalid URL" unless uri.is_a?(URI::HTTP) # includes https
 
-        if location
-          download_url_to_file(location, dst)
-        else
-          FileUtils.mv(tmp, dst)
-          nil
+        puts "Downloading #{url}..."
+        uri.open(max_redirects: 10) do |download|
+          # TODO move file when possible
+          IO.copy_stream(download, dst.to_str)
         end
+        nil
       end
 
       def load_state_dict_from_url(url, model_dir: nil)

data/lib/torch/nn/module.rb

@@ -197,7 +197,7 @@ module Torch
         named_buffers.values
       end
 
-      # TODO set recurse: true in 0.18.0
+      # TODO set recurse: true in future
       def named_buffers(prefix: "", recurse: false)
         buffers = {}
         if recurse

data/lib/torch/nn/rnn_base.rb

@@ -161,7 +161,7 @@ module Torch
       private
 
       def _flat_weights
-        @all_weights.flatten.map { |v| instance_variable_get("@#{v}") }.compact
+        @all_weights.flatten.filter_map { |v| instance_variable_get("@#{v}") }
       end
 
       def _get_flat_weights

data/lib/torch/version.rb

@@ -1,3 +1,3 @@
 module Torch
-  VERSION = "0.23.0"
+  VERSION = "0.24.0"
 end

data/lib/torch.rb

@@ -279,7 +279,7 @@ module Torch
   def self._make_tensor_class(dtype, cuda = false)
     cls = Class.new
     device = cuda ? "cuda" : "cpu"
-    cls.define_singleton_method("new") do |*args|
+    cls.define_singleton_method(:new) do |*args|
       if args.size == 1 && args.first.is_a?(Tensor)
         args.first.send(dtype).to(device)
       elsif args.size == 1 && args.first.is_a?(ByteStorage) && dtype == :uint8
@@ -347,7 +347,7 @@ module Torch
       # from_blob does not own the data, so we need to keep
       # a reference to it for duration of tensor
       # can remove when passing pointer directly
-      tensor.instance_variable_set("@_numo_data", data)
+      tensor.instance_variable_set(:@_numo_data, data)
       tensor
     end
 
@@ -426,11 +426,11 @@ module Torch
       end
 
       if options[:dtype].nil?
-        if data.all? { |v| v.is_a?(Integer) }
+        if data.all?(Integer)
           options[:dtype] = :int64
         elsif data.all? { |v| v == true || v == false }
           options[:dtype] = :bool
-        elsif data.any? { |v| v.is_a?(Complex) }
+        elsif data.any?(Complex)
           options[:dtype] = :complex64
         end
       end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: torch-rb
 version: !ruby/object:Gem::Version
-  version: 0.23.0
+  version: 0.24.0
 platform: ruby
 authors:
 - Andrew Kane
@@ -234,14 +234,14 @@ required_ruby_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
-      version: '3.2'
+      version: '3.3'
 required_rubygems_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 4.0.3
+rubygems_version: 4.0.6
 specification_version: 4
 summary: Deep learning for Ruby, powered by LibTorch
 test_files: []