lib/torch/utils/data/data_loader.rb in torch-rb-0.5.3 vs lib/torch/utils/data/data_loader.rb in torch-rb-0.6.0

@@ -58,10 +58,10 @@
           when Integer
             Torch.tensor(batch)
           when Array
             batch.transpose.map { |v| default_convert(v) }
           else
-            raise NotImplementedYet
+            batch
           end
         end
 
         def auto_collation?
           !@batch_sampler.nil?