03acf3cd9eeaed74b415aaf86178751bd4a212fa,ml/rl/readers/data_streamer.py,,pin_memory,#Any#,137

Before Change

            }
        )
    elif dataclasses.is_dataclass(batch):
        return type(batch)(
            **{
                name: pin_memory(value)
                for name, value in dataclasses.asdict(batch).items()
            }
        )
    elif isinstance(batch, collections.Mapping):
        // NB: preserving OrderedDict
        return type(batch)((k, pin_memory(sample)) for k, sample in batch.items())
    elif isinstance(batch, collections.Sequence):

After Change

    elif isinstance(batch, string_classes):
        return batch
    elif dataclasses.is_dataclass(batch):
        retval = dataclasses.replace(
            batch,
            **{
                field.name: pin_memory(getattr(batch, field.name))
                for field in dataclasses.fields(batch)
            }
        )
        return retval
    elif isinstance(batch, collections.Mapping):
        // NB: preserving OrderedDict
        return type(batch)((k, pin_memory(sample)) for k, sample in batch.items())
    elif isinstance(batch, collections.Sequence):

Instances