[Docs] De-flake doctests (#37743)

Signed-off-by: Balaji Veeramani <[email protected]>
ray-project · Jul 26, 2023 · 431d280 · 431d280
1 parent 6048b6b
commit 431d280
Show file tree

Hide file tree

Showing 4 changed files with 24 additions and 7 deletions.
diff --git a/bazel/python.bzl b/bazel/python.bzl
@@ -23,6 +23,7 @@ def doctest(files, gpu = False, name="doctest", deps=[], srcs=[], data=[], args=
             "--doctest-modules",
             "--doctest-glob='*.md'",
             "-c=$(location //bazel:conftest.py)",
+            "--disable-warnings",
             "-v"
         ] + args + ["$(location :%s)" % file for file in files],
         data = ["//bazel:conftest.py"] + files + data,

diff --git a/doc/source/data/key-concepts.rst b/doc/source/data/key-concepts.rst
@@ -161,10 +161,11 @@ or remote filesystems.
 
     transformed_ds.write_parquet("/tmp/iris")
 
-    print(sorted(os.listdir("/tmp/iris")))
+    print(os.listdir("/tmp/iris"))
 
 .. testoutput::
-
+    :options: +MOCK
+
     ['..._000000.parquet', '..._000001.parquet']
 
 

diff --git a/python/ray/data/read_api.py b/python/ray/data/read_api.py
@@ -1635,6 +1635,15 @@ def read_sql(
         For examples of reading from larger databases like MySQL and PostgreSQL, see
         :ref:`Reading from SQL Databases <reading_sql>`.
 
+        .. testcode::
+            :hide:
+
+            import os
+            try:
+                os.remove("example.db")
+            except OSError:
+                pass
+            
         .. testcode::
 
             import sqlite3

diff --git a/python/ray/train/tensorflow/tensorflow_checkpoint.py b/python/ray/train/tensorflow/tensorflow_checkpoint.py
@@ -66,14 +66,20 @@ def from_model(
             A :py:class:`TensorflowCheckpoint` containing the specified model.
 
         Examples:
+        
+            .. testcode::
 
-        .. testcode::
+                from ray.train.tensorflow import TensorflowCheckpoint
+                import tensorflow as tf
 
-            from ray.train.tensorflow import TensorflowCheckpoint
-            import tensorflow as tf
+                model = tf.keras.applications.resnet.ResNet101()
+                checkpoint = TensorflowCheckpoint.from_model(model)
+
+            .. testoutput::
+                :options: +MOCK
+                :hide:
 
-            model = tf.keras.applications.resnet.ResNet101()
-            checkpoint = TensorflowCheckpoint.from_model(model)
+                ...  # Model may or may not be downloaded
 
         """
         checkpoint = cls.from_dict(