torchvision 0.5.0 → 0.5.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: e769d6e2716a72256fc6f0360a429a5d61bc9df8ca9fd0182330303957e8bd65
-  data.tar.gz: c1e52d021196b77db9d1ec7cd7be5a5ed41277c2a8c6a5ffd59f05782690ef43
+  metadata.gz: 658be49c788bf6f989b0bf082b83cdb7e422adebab0ba1a29a18d863677af9a5
+  data.tar.gz: 04f2edb3628a315f81cc25e1fc0a47d9de7d16c212f3d82908f18594be84f0c4
 SHA512:
-  metadata.gz: a45a3566061ed0683d7d47ed52aec0fd1921a07d1b4ca2d798409e1b08931a1dd6065b6c8df9d49fa06c09d51b4f6d42e12621832aec04ec23a5ee84e916abaf
-  data.tar.gz: 88e1669aa4ea9256ddf1c3b6bcfb6df93e313331b5de1b32c12b348c2e32f981f9333d3bad54f7577063c17361d0841e823e8d2cb67df3d8916a3527991a9cca
+  metadata.gz: c60559ac4527348565c900d7cb725f7dac58f97275842db04092c35b90b56a02d55d5a6444ee440ad96d4562cd929eb24cffd513f5c0f82b3cd7ea2e411054b2
+  data.tar.gz: 17cb226cfadbf29896ab3ec1cf3aeb96fd0ea15cd509e47f0cb996ade9aec63aeaffb7ef005ca23ed258f200f84c71303539cc34ebbda7b4a9caae75a8347443

data/CHANGELOG.md

@@ -1,3 +1,7 @@
+## 0.5.1 (2026-06-29)
+
+- Fixed download for CIFAR datasets
+
 ## 0.5.0 (2026-04-07)
 
 - Switched to `numo-narray-alt`

data/lib/torchvision/datasets/cifar10.rb

@@ -1,7 +1,7 @@
 module TorchVision
   module Datasets
     class CIFAR10 < VisionDataset
-      # https://www.cs.toronto.edu/~kriz/cifar.html
+      # https://cave.cs.toronto.edu/kriz/cifar.html
 
       def initialize(root, train: true, download: false, transform: nil, target_transform: nil)
         super(root, transform: transform, target_transform: target_transform)
@@ -82,7 +82,7 @@ module TorchVision
       end
 
       def url
-        "https://www.cs.toronto.edu/~kriz/cifar-10-binary.tar.gz"
+        "https://cave.cs.toronto.edu/kriz/cifar-10-binary.tar.gz"
       end
 
       def filename

data/lib/torchvision/datasets/cifar100.rb

@@ -1,7 +1,7 @@
 module TorchVision
   module Datasets
     class CIFAR100 < CIFAR10
-      # https://www.cs.toronto.edu/~kriz/cifar.html
+      # https://cave.cs.toronto.edu/kriz/cifar.html
 
       private
 
@@ -10,7 +10,7 @@ module TorchVision
       end
 
       def url
-        "https://www.cs.toronto.edu/~kriz/cifar-100-binary.tar.gz"
+        "https://cave.cs.toronto.edu/kriz/cifar-100-binary.tar.gz"
       end
 
       def filename

data/lib/torchvision/datasets/mnist.rb

@@ -59,7 +59,7 @@ module TorchVision
               download_file(url, download_root: raw_folder, filename: resource[:filename], sha256: resource[:sha256])
               success = true
               break
-            rescue Errno::ECONNREFUSED, Net::HTTPFatalError, Net::HTTPClientException => e
+            rescue Errno::ECONNREFUSED => e
               puts "Failed to download (trying next): #{e.message}"
             end
           end

data/lib/torchvision/datasets/vision_dataset.rb

@@ -29,34 +29,25 @@ module TorchVision
         dest = File.join(download_root, filename)
         return dest if File.exist?(dest)
 
-        temp_path = "#{Dir.tmpdir}/#{Time.now.to_f}" # TODO better name
-
-        uri = URI(url)
-
-        # Net::HTTP automatically adds Accept-Encoding for compression
-        # of response bodies and automatically decompresses gzip
-        # and deflateresponses unless a Range header was sent.
-        # https://ruby-doc.org/stdlib-2.6.4/libdoc/net/http/rdoc/Net/HTTP.html
-        Net::HTTP.start(uri.host, uri.port, use_ssl: uri.scheme == "https", open_timeout: 3) do |http|
-          request = Net::HTTP::Get.new(uri)
-
-          puts "Downloading #{url}..."
-          File.open(temp_path, "wb") do |f|
-            http.request(request) do |response|
-              response.value # raise error if not success
-              response.read_body do |chunk|
-                f.write(chunk)
-              end
+        uri = URI.parse(url)
+        raise "Invalid URL" unless uri.is_a?(URI::HTTP) # includes https
+
+        uri.open(open_timeout: 3, redirect: false) do |download|
+          digest =
+            if download.respond_to?(:path)
+              download.flush
+              Digest::SHA256.file(download.path).hexdigest
+            else
+              Digest::SHA256.hexdigest(download.string)
             end
+
+          if digest != sha256
+            raise Error, "Bad hash"
           end
-        end
 
-        unless check_integrity(temp_path, sha256)
-          raise Error, "Bad hash"
+          IO.copy_stream(download, dest.to_str)
         end
 
-        FileUtils.mv(temp_path, dest)
-
         dest
       end
 

data/lib/torchvision/transforms/functional.rb

@@ -19,7 +19,7 @@ module TorchVision
           std = Torch.tensor(std, dtype: dtype, device: tensor.device)
 
           # TODO
-          if std.to_a.any? { |v| v == 0 }
+          if std.to_a.any?(0)
             raise ArgumentError, "std evaluated to zero after conversion to #{dtype}, leading to division by zero."
           end
           if mean.ndim == 1

data/lib/torchvision/version.rb

@@ -1,3 +1,3 @@
 module TorchVision
-  VERSION = "0.5.0"
+  VERSION = "0.5.1"
 end

data/lib/torchvision.rb

@@ -6,7 +6,7 @@ require "torch"
 # stdlib
 require "digest"
 require "fileutils"
-require "net/http"
+require "open-uri"
 require "rubygems/package"
 require "tmpdir"
 

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: torchvision
 version: !ruby/object:Gem::Version
-  version: 0.5.0
+  version: 0.5.1
 platform: ruby
 authors:
 - Andrew Kane
@@ -119,7 +119,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 4.0.6
+rubygems_version: 4.0.14
 specification_version: 4
 summary: Computer vision datasets, transforms, and models for Ruby
 test_files: []