Skip to content

Commit 7148b1e

Browse files
thomasthomas
thomas
authored and
thomas
committed
update
1 parent 53d6a17 commit 7148b1e

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

src/lightning/data/streaming/dataset.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -78,7 +78,7 @@ def __init__(
7878
if self.distributed_env.world_size > 1:
7979
if drop_last is False:
8080
logger.warn(
81-
"You're operating within a distributed environment and have enabled the `drop_last` option. "
81+
"You're operating within a distributed environment and have disabled the `drop_last` option. "
8282
"Please note that this configuration may lead to training interruptions if your system depends "
8383
"on distributed collectives."
8484
)

tests/tests_data/streaming/test_dataset.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -902,5 +902,5 @@ def detect(cls):
902902
assert not dataset.drop_last
903903

904904
warn_value = logger_mock.warn._mock_mock_calls[0].args[0]
905-
assert warn_value == "You're operating within a distributed environment and have enabled the `drop_last`" \
905+
assert warn_value == "You're operating within a distributed environment and have disabled the `drop_last`" \
906906
" option. Please note that this configuration may lead to training interruptions if your system depends on distributed collectives." # noqa: E501

0 commit comments

Comments
 (0)