Skip to content

Commit e3e2c6e

Browse files
committed
Remove PYTORCH_TEST_WITH_SLOW.
1 parent 8e56a37 commit e3e2c6e

File tree

3 files changed

+1
-9
lines changed

3 files changed

+1
-9
lines changed

.circleci/unittest/linux/scripts/run_test.sh

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,5 @@ set -e
55
eval "$(./conda/bin/conda shell.bash hook)"
66
conda activate ./env
77

8-
export PYTORCH_TEST_WITH_SLOW='1'
98
python -m torch.utils.collect_env
109
pytest --junitxml=test-results/junit.xml -v --durations 20

.circleci/unittest/windows/scripts/run_test.sh

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,5 @@ conda activate ./env
88
this_dir="$( cd "$( dirname "${BASH_SOURCE[0]}" )" >/dev/null 2>&1 && pwd )"
99
source "$this_dir/set_cuda_envs.sh"
1010

11-
export PYTORCH_TEST_WITH_SLOW='1'
1211
python -m torch.utils.collect_env
1312
pytest --junitxml=test-results/junit.xml -v --durations 20

test/test_models.py

Lines changed: 1 addition & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -129,16 +129,10 @@ def get_export_import_copy(m):
129129
imported = torch.jit.load(path)
130130
return imported
131131

132-
TEST_WITH_SLOW = os.getenv("PYTORCH_TEST_WITH_SLOW", "0") == "1"
133-
if not TEST_WITH_SLOW or skip:
132+
if skip:
134133
# TorchScript is not enabled, skip these tests
135134
msg = (
136135
f"The check_jit_scriptable test for {nn_module.__class__.__name__} was skipped. "
137-
"This test checks if the module's results in TorchScript "
138-
"match eager and that it can be exported. To run these "
139-
"tests make sure you set the environment variable "
140-
"PYTORCH_TEST_WITH_SLOW=1 and that the test is not "
141-
"manually skipped."
142136
)
143137
warnings.warn(msg, RuntimeWarning)
144138
return None

0 commit comments

Comments
 (0)