Open MinuraPunchihewa opened 1 month ago
Hey @astrojuanlu,
This is my implementation of the SafetensorsDataset
. I would appreciate a review.
Thank you for the PR, @MinuraPunchihewa! It looks great to me. I tested it manually, and everything works perfectly!
Thank you, @DimedS. I appreciate it. I just updated the release notes as well.
Description
This PR adds the
SafetensorsDataset
to support interactions with tensors stored in files in the Safetensors format.Fixes https://github.com/kedro-org/kedro-plugins/issues/221
Development notes
I have used the PickleDataset as a base for the implementation of this dataset (as it goes about the same manner to access files).
These changes have been tested,
Checklist
RELEASE.md
file