frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

[ci] remove IN_CI env var

Browse Source 
The conventional env var to set is CI. Both circle and GHA set it, so
IN_CI is unnecessary

Pull Request resolved: https://github.com/pytorch/pytorch/pull/79229

Approved by: https://github.com/janeyx99
This commit is contained in:
Michael Suo
2022-06-10 23:52:15 -07:00
committed by PyTorch MergeBot
parent f51d5233f2
commit c978b609f7
58 changed files with 61 additions and 98 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
test/test_import_stats.py
View File

@ -5,7 +5,7 @@ import sys
import unittest
import pathlib
from torch.testing._internal.common_utils import TestCase, run_tests, IS_LINUX, IS_IN_CI
from torch.testing._internal.common_utils import TestCase, run_tests, IS_LINUX, IS_CI
REPO_ROOT = pathlib.Path(__file__).resolve().parent.parent
@ -33,7 +33,7 @@ class TestImportTime(TestCase):
)
@unittest.skipIf(not IS_LINUX, "Memory test is only implemented for Linux")
@unittest.skipIf(not IS_IN_CI, "Memory test only runs in CI")
@unittest.skipIf(not IS_CI, "Memory test only runs in CI")
@unittest.skipIf(rds_write is None, "Cannot import rds_write from tools.stats.scribe")
def test_peak_memory(self):
def profile(module, name):
@ -58,7 +58,7 @@ class TestImportTime(TestCase):
if __name__ == "__main__":
if register_rds_schema and IS_IN_CI:
if register_rds_schema and IS_CI:
try:
register_rds_schema(
"import_stats",