From abe8e4d40e8403c232d9f8c325b32ea2ecb2fbe9 Mon Sep 17 00:00:00 2001 From: "Thomas J. Fan" Date: Thu, 28 Jan 2021 16:53:20 -0500 Subject: [PATCH] CI Skips test in loading_other_datasets.rst based on env flag --- doc/conftest.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/doc/conftest.py b/doc/conftest.py index 4496bb74152ac..5468184bf5509 100644 --- a/doc/conftest.py +++ b/doc/conftest.py @@ -1,6 +1,7 @@ import os from os.path import exists from os.path import join +from os import environ import warnings from sklearn.utils import IS_PYPY @@ -48,6 +49,12 @@ def setup_loading_other_datasets(): raise SkipTest("Skipping loading_other_datasets.rst, " "pandas not installed") + # checks SKLEARN_SKIP_NETWORK_TESTS to see if test should run + run_network_tests = environ.get("SKLEARN_SKIP_NETWORK_TESTS", '1') == "0" + if not run_network_tests: + raise SkipTest("Skipping loading_other_datasets.rst, tests can be " + "enabled by settting SKLEARN_SKIP_NETWORK_TESTS=0") + def setup_compose(): try: