diff --git a/test/external/process_replay/process_replay.py b/test/external/process_replay/process_replay.py index d8e5ca2534..90a789e715 100755 --- a/test/external/process_replay/process_replay.py +++ b/test/external/process_replay/process_replay.py @@ -7,9 +7,9 @@ from tinygrad.helpers import Context, ContextVar, colored, db_connection, VERSIO PAGE_SIZE = 100 TABLE_NAME = f"process_replay_{getenv('GITHUB_RUN_ID', 'HEAD')}_{VERSION}" ASSERT_DIFF = getenv("ASSERT_PROCESS_REPLAY", int((k:="[run_process_replay]") in os.getenv("COMMIT_MESSAGE", k) or k in os.getenv("PR_TITLE", k))) -SKIP_PROCESS_REPLAY = int((k:="[skip_process_replay]") in os.getenv("COMMIT_MESSAGE", "") or k in os.getenv("PR_TITLE", "")) +REF = os.getenv("GITHUB_REF_NAME", "") +SKIP_PROCESS_REPLAY = int((k:="[skip_process_replay]") in os.getenv("COMMIT_MESSAGE", "") or k in os.getenv("PR_TITLE", "")) or REF == "master" MAX_DIFF_PCT = getenv("PROCESS_REPLAY_MAX_DIFF_PCT", 20) -assert MAX_DIFF_PCT < 100 early_stop = multiprocessing.Event() logging.basicConfig(level=logging.INFO, format='%(message)s')