From 1025019db56e59c688f27d9775fbb9ebc86a56ab Mon Sep 17 00:00:00 2001 From: Hamza Jazmati Date: Tue, 27 Aug 2024 09:41:56 -0400 Subject: [PATCH 1/3] Allow Custom File Systems To Override The Walk Method --- tensorboard/compat/tensorflow_stub/io/gfile.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/tensorboard/compat/tensorflow_stub/io/gfile.py b/tensorboard/compat/tensorflow_stub/io/gfile.py index 983abf2313..8ccc9b5166 100644 --- a/tensorboard/compat/tensorflow_stub/io/gfile.py +++ b/tensorboard/compat/tensorflow_stub/io/gfile.py @@ -928,6 +928,12 @@ def walk(top, topdown=True, onerror=None): """ top = compat.as_str_any(top) fs = get_filesystem(top) + + if hasattr(fs, 'walk'): + for item in fs.walk(top, onerror=onerror): + yield item + return + try: listing = listdir(top) except errors.NotFoundError as err: From f60b955832b968d9ffac7a8d1767d37abaa73c7f Mon Sep 17 00:00:00 2001 From: Hamza Jazmati Date: Tue, 27 Aug 2024 10:58:27 -0400 Subject: [PATCH 2/3] Allow Custom File Systems To Override The Walk Method --- tensorboard/compat/tensorflow_stub/io/gfile.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tensorboard/compat/tensorflow_stub/io/gfile.py b/tensorboard/compat/tensorflow_stub/io/gfile.py index 8ccc9b5166..f332a1605e 100644 --- a/tensorboard/compat/tensorflow_stub/io/gfile.py +++ b/tensorboard/compat/tensorflow_stub/io/gfile.py @@ -928,9 +928,9 @@ def walk(top, topdown=True, onerror=None): """ top = compat.as_str_any(top) fs = get_filesystem(top) - + if hasattr(fs, 'walk'): - for item in fs.walk(top, onerror=onerror): + for item in fs.walk(top, topdown=topdown, onerror=onerror): yield item return From c7a7f973eeec3f63d57e3ec7f053f80b7af46dfd Mon Sep 17 00:00:00 2001 From: Hamza Jazmati Date: Thu, 29 Aug 2024 13:29:27 -0400 Subject: [PATCH 3/3] Allow Custom File Systems To Override The Walk Method --- tensorboard/compat/tensorflow_stub/io/gfile.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tensorboard/compat/tensorflow_stub/io/gfile.py b/tensorboard/compat/tensorflow_stub/io/gfile.py index f332a1605e..aebf16861b 100644 --- a/tensorboard/compat/tensorflow_stub/io/gfile.py +++ b/tensorboard/compat/tensorflow_stub/io/gfile.py @@ -929,7 +929,7 @@ def walk(top, topdown=True, onerror=None): top = compat.as_str_any(top) fs = get_filesystem(top) - if hasattr(fs, 'walk'): + if hasattr(fs, "walk"): for item in fs.walk(top, topdown=topdown, onerror=onerror): yield item return