envs.py 529 Bytes
Newer Older
1
import os
helloyongyang's avatar
helloyongyang committed
2
from functools import lru_cache
3
4


helloyongyang's avatar
helloyongyang committed
5
6
7
8
@lru_cache(maxsize=None)
def CHECK_ENABLE_PROFILING_DEBUG():
    ENABLE_PROFILING_DEBUG = os.getenv("ENABLE_PROFILING_DEBUG", "false").lower() == "true"
    return ENABLE_PROFILING_DEBUG
9

helloyongyang's avatar
helloyongyang committed
10
11
12
13
14

@lru_cache(maxsize=None)
def CHECK_ENABLE_GRAPH_MODE():
    ENABLE_GRAPH_MODE = os.getenv("ENABLE_GRAPH_MODE", "false").lower() == "true"
    return ENABLE_GRAPH_MODE
15
16
17
18
19
20


@lru_cache(maxsize=None)
def GET_RUNNING_FLAG():
    RUNNING_FLAG = os.getenv("RUNNING_FLAG", "infer")
    return RUNNING_FLAG