def strip_newline()

in torchdata/datapipes/iter/util/plain_text_reader.py [0:0]

• 9 lines of code
• 4 McCabe index (conditional complexity)

    def strip_newline(self, stream: Union[Iterator[bytes], Iterator[str]]) -> Union[Iterator[bytes], Iterator[str]]:
        if not self._strip_newline:
            yield from stream
            return

        for line in stream:
            if isinstance(line, str):
                yield line.strip("\r\n")
            else:
                yield line.strip(b"\r\n")