Skip to content

[Data] Remove obsolete _DatasetWrapper #59300

@bveeramani

Description

@bveeramani

The StreamingSplitDataIterator wraps its underlying dataset with a _DatasetWrapper class to avoid #52549. Since that issue has been now fixed, we should clean up the code and remove the unnecessary wrapper:

class _DatasetWrapper:
# A temporary workaround for https://github.com/ray-project/ray/issues/52549
def __init__(self, dataset: "Dataset") -> None:
self._dataset = dataset

Metadata

Metadata

Assignees

No one assigned

    Labels

    dataRay Data-related issuesgood-first-issueGreat starter issue for someone just starting to contribute to Ray

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions