lib/torch.rb in torch-rb-0.12.1 vs lib/torch.rb in torch-rb-0.12.2
- old
+ new
@@ -185,9 +185,16 @@
# utils
require "torch/utils/data"
require "torch/utils/data/data_loader"
require "torch/utils/data/dataset"
+require "torch/utils/data/iterable_dataset"
+require "torch/utils/data/data_pipes/iter_data_pipe"
+require "torch/utils/data/data_pipes/filter_iter_data_pipe"
+require "torch/utils/data/data_pipes/iter/file_lister"
+require "torch/utils/data/data_pipes/iter/file_opener"
+require "torch/utils/data/data_pipes/iter/iterable_wrapper"
+require "torch/utils/data/data_pipes/iter/stream_wrapper"
require "torch/utils/data/subset"
require "torch/utils/data/tensor_dataset"
# hub
require "torch/hub"