diff --git a/streaming/base/stream.py b/streaming/base/stream.py index 1fd9fc9ff..15e288a53 100644 --- a/streaming/base/stream.py +++ b/streaming/base/stream.py @@ -454,8 +454,8 @@ def get_shards(self, world: World, allow_unsafe_types: bool) -> List[Reader]: wait_for_file_to_exist( filename, TICK, self.download_timeout, f'Index file {os.path.join(self.remote or "", self.split or "", basename)} ' + - f'-> {filename} took too long to download. Either increase the ' + - f'`download_timeout` value or check the other traceback.') + f'-> {filename} took too long to download or failed to download. Either increase the ' + + f'`download_timeout` value or check the local rank 0 traceback.') # Load the index. try: