Method: Torch::Utils::Data::DataPipes::Iter::FileOpener#initialize

Defined in:
lib/torch/utils/data/data_pipes/iter/file_opener.rb

#initialize(datapipe, mode: "r", encoding: nil, length: -1)) ⇒ FileOpener

Returns a new instance of FileOpener.



7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
# File 'lib/torch/utils/data/data_pipes/iter/file_opener.rb', line 7

def initialize(datapipe, mode: "r", encoding: nil, length: -1)
  super()
  @datapipe = datapipe
  @mode = mode
  @encoding = encoding

  if !["b", "t", "rb", "rt", "r"].include?(@mode)
    raise ArgumentError, "Invalid mode #{mode}"
  end

  if mode.include?("b") && !encoding.nil?
    raise ArgumentError, "binary mode doesn't take an encoding argument"
  end

  @length = length
end