Skip to content

Commit b48d350

Browse files
kyudspeterxcli
authored andcommitted
[Data] Support SaveMode for more write operations (ray-project#57946)
## Description See ray-project#57924 ## Related issues Fixes ray-project#57924 ## Additional information - [x] Verify by manual testing -> confirmed with example using `write_numpy` Signed-off-by: kyuds <kyuseung1016@gmail.com> Signed-off-by: peterxcli <peterxcli@gmail.com>
1 parent 02093c7 commit b48d350

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

python/ray/data/dataset.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4372,6 +4372,7 @@ def write_tfrecords(
43724372
open_stream_args=arrow_open_stream_args,
43734373
filename_provider=filename_provider,
43744374
dataset_uuid=self._uuid,
4375+
mode=mode,
43754376
)
43764377
self.write_datasink(
43774378
datasink,
@@ -4469,6 +4470,7 @@ def write_webdataset(
44694470
open_stream_args=arrow_open_stream_args,
44704471
filename_provider=filename_provider,
44714472
dataset_uuid=self._uuid,
4473+
mode=mode,
44724474
)
44734475
self.write_datasink(
44744476
datasink,
@@ -4569,6 +4571,7 @@ def write_numpy(
45694571
open_stream_args=arrow_open_stream_args,
45704572
filename_provider=filename_provider,
45714573
dataset_uuid=self._uuid,
4574+
mode=mode,
45724575
)
45734576
self.write_datasink(
45744577
datasink,

0 commit comments

Comments
 (0)