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

Before Change

        return batch
    elif dataclasses.is_dataclass(batch):
        return type(batch)(
            **{
                name: pin_memory(value)
                for name, value in dataclasses.asdict(batch).items()
            }
        )
    elif dataclasses.is_dataclass(batch):
        return type(batch)(
            **{

After Change

        retval = dataclasses.replace(
            batch,
            **{
                field.name: pin_memory(getattr(batch, field.name))
                for field in dataclasses.fields(batch)
            }
        )

Instances