Always skip saver to avoid creating files

pytorch · Mar 7, 2023 · 438f5a7 · 438f5a7
1 parent 7d9e315
commit 438f5a7
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 5 deletions.
diff --git a/docs/source/conf.py b/docs/source/conf.py
@@ -62,12 +62,9 @@
 doctest_test_doctest_blocks = ""
 
 doctest_global_setup = """
-import os
 import torch
 from torchdata.datapipes.iter import IterableWrapper, FileLister, FileOpener
 
-seperator_is_slash = os.sep == "/"
-
 io_doctest = True
 
 try:

diff --git a/torchdata/datapipes/iter/util/saver.py b/torchdata/datapipes/iter/util/saver.py
@@ -29,7 +29,7 @@ class SaverIterDataPipe(IterDataPipe[str]):
     Example:
 
     .. testcode::
-        :skipif: not seperator_is_slash
+        :skipif: io_doctest
 
         import os
 
@@ -42,7 +42,7 @@ def filepath_fn(name: str) -> str:
         print(list(saver_dp))
 
     .. testoutput::
-        :skipif: not seperator_is_slash
+        :skipif: io_doctest
 
         ['./1.txt', './2.txt', './3.txt']