def __init__(self,
db=None,
container_manager=None,
var_autoforward=ENVIRONMENT_VARIABLES_AUTOFORWARD):
db = db or Database()
container_manager: ContainerManager = container_manager or DockerSwarmContainerManager(
)
# Ensure that environment variable exists, failing fast
for x in var_autoforward:
os.environ[x]
self._var_autoforward = var_autoforward
self._data_dir_path = os.environ['DATA_DIR_PATH']
self._logs_dir_path = os.environ['LOGS_DIR_PATH']
self._params_dir_path = os.environ['PARAMS_DIR_PATH']
self._host_workdir_path = os.environ['HOST_WORKDIR_PATH']
self._docker_workdir_path = os.environ['DOCKER_WORKDIR_PATH']
self._predictor_image = '{}:{}'.format(
os.environ['RAFIKI_IMAGE_PREDICTOR'], os.environ['RAFIKI_VERSION'])
self._predictor_port = os.environ['PREDICTOR_PORT']
self._rafiki_addr = os.environ['RAFIKI_ADDR']
self._app_mode = os.environ['APP_MODE']
self._db = db
self._container_manager = container_manager
def __init__(self, db=None, container_manager=None):
if db is None:
db = Database()
if container_manager is None:
container_manager = DockerSwarmContainerManager()
self._postgres_host = os.environ['POSTGRES_HOST']
self._postgres_port = os.environ['POSTGRES_PORT']
self._postgres_user = os.environ['POSTGRES_USER']
self._postgres_password = os.environ['POSTGRES_PASSWORD']
self._postgres_db = os.environ['POSTGRES_DB']
self._redis_host = os.environ['REDIS_HOST']
self._redis_port = os.environ['REDIS_PORT']
self._admin_host = os.environ['ADMIN_HOST']
self._admin_host = os.environ['ADMIN_HOST']
self._admin_port = os.environ['ADMIN_PORT']
self._advisor_host = os.environ['ADVISOR_HOST']
self._advisor_port = os.environ['ADVISOR_PORT']
self._data_workdir = os.environ['DATA_WORKDIR_PATH']
self._logs_workdir = os.environ['LOGS_WORKDIR_PATH']
self._data_docker_workdir = os.environ['DATA_DOCKER_WORKDIR_PATH']
self._logs_docker_workdir = os.environ['LOGS_DOCKER_WORKDIR_PATH']
self._predictor_image = '{}:{}'.format(
os.environ['RAFIKI_IMAGE_PREDICTOR'], os.environ['RAFIKI_VERSION'])
self._predictor_port = os.environ['PREDICTOR_PORT']
self._rafiki_addr = os.environ['RAFIKI_ADDR']
self._db = db
self._container_manager = container_manager
def __init__(self, db=Database(), container_manager=DockerSwarmContainerManager()):
self._base_worker_image = '{}:{}'.format(os.environ['RAFIKI_IMAGE_WORKER'],
os.environ['RAFIKI_VERSION'])
self._db = db
self._services_manager = ServicesManager(db, container_manager)
with self._db:
self._seed_users()
def __init__(self, db=None, container_manager=None):
if db is None:
db = Database()
if container_manager is None:
container_manager = DockerSwarmContainerManager()
self._base_worker_image = '{}:{}'.format(
os.environ['RAFIKI_IMAGE_WORKER'], os.environ['RAFIKI_VERSION'])
self._db = db
self._services_manager = ServicesManager(db, container_manager)
def __init__(self,
meta_store=None,
container_manager=None,
data_store=None,
param_store=None):
self._meta_store = meta_store or MetaStore()
container_manager = container_manager or DockerSwarmContainerManager()
self._data_store: DataStore = data_store or FileDataStore()
self._param_store: ParamStore = param_store or FileParamStore()
self._base_worker_image = '{}:{}'.format(
os.environ['RAFIKI_IMAGE_WORKER'], os.environ['RAFIKI_VERSION'])
self._services_manager = ServicesManager(self._meta_store,
container_manager)