Keep backwards compatibility

ankane · Jul 30, 2023 · d44b444 · d44b444
1 parent e15876f
commit d44b444
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/ext/torch/torch.cpp b/ext/torch/torch.cpp
@@ -62,13 +62,12 @@ void init_torch(Rice::Module& m) {
       })
     .define_singleton_function(
       "_load",
-      [](const std::string filename) {
+      [](const std::string &filename) {
         // https://github.com/pytorch/pytorch/issues/20356#issuecomment-567663701
         std::ifstream input(filename, std::ios::binary);
         std::vector<char> bytes(
             (std::istreambuf_iterator<char>(input)),
             (std::istreambuf_iterator<char>()));
-
         input.close();
         return torch::pickle_load(bytes);
       })

diff --git a/lib/torch.rb b/lib/torch.rb
@@ -388,6 +388,9 @@ def save(obj, f)
     end
 
     def load(f)
+      # keep backwards compatibility
+      File.open(f, "rb") { |f2| f2.read(1) }
+
       to_ruby(_load(f))
     end