def __init__(self):
def get_critical_env(env):
"""
A more strict way to get critical environment variable
"""
try:
var = os.environ[env]
except KeyError as _:
logger.error(f"{env} not defined")
sys.exit(1)
else:
return var
for env in Preprocessor.env_list:
setattr(self, '_' + env.lower(), get_critical_env(env))
self._trtis_conifig_generator = TRTISConfigGenerator()
self.model_root_path = self._model_store
self.model_path = os.path.join(self.model_root_path,
self._serving_name, "1")
def __init__(self):
def get_critical_env(env):
'''
A more strict way to get critical environment variable
'''
try:
var = os.environ[env]
except KeyError:
logger.error(f'{env} not defined')
sys.exit(1)
else:
return var
for env in Preprocessor.env_list:
setattr(self, '_' + env.lower(), get_critical_env(env))
self.using_ormbfile = True if os.getenv(USING_ORMBFILE_ENV, '') else False
self._trtis_conifig_generator = TRTISConfigGenerator()
self.model_root_path = self._model_store
self.model_path = os.path.join(
self.model_root_path, self._serving_name, '1')