feat(datasets): Added PyTorchDataset

kedro-org / kedro-plugins

First-party plugins maintained by the Kedro team.

Apache License 2.0

91 stars 84 forks source link

feat(datasets): Added PyTorchDataset #735

Closed bpmeek closed 1 month ago

bpmeek commented 3 months ago

Description

Closes 714

Development notes

Added PyTorchDataset to use pytorch's recommended save and load methods. Added tests to kedro-datasets/tests/kedro_datasets_experimental/pytorch/test_pytorch_dataset.py

Checklist

[X] Updated the documentation to reflect the code changes
[X] Added a description of this change in the relevant RELEASE.md file
[X] Added tests to cover my changes

ankatiyar commented 3 months ago

Hey @bpmeek, could you resolve the DCO failures following the instructions here - https://github.com/kedro-org/kedro-plugins/pull/735/checks?check_run_id=26337551853

bpmeek commented 3 months ago

I'm not sure what to do about the unit-test failures, none of them are related to my changes.

running pytest kedro-datasets/tests/kedro_datasets_experimental/pytorch/test_pytorch_dataset.py passes successfully.

ankatiyar commented 3 months ago

@bpmeek Yeah the failures don't seem to be related, I'll take a look at the CI failures! Thanks for fixing the DCO check!