Add default tensorboard docker image
@@ -11,7 +11,8 @@ ROLE_LABELS_API = config.get_string('POLYAXON_ROLE_LABELS_API')
K8S_SERVICE_ACCOUNT_NAME = config.get_string('POLYAXON_K8S_SERVICE_ACCOUNT_NAME')
K8S_RBAC_ENABLED = config.get_boolean('POLYAXON_K8S_RBAC_ENABLED')
K8S_INGRESS_ENABLED = config.get_boolean('POLYAXON_K8S_INGRESS_ENABLED')
-K8S_INGRESS_ANNOTATIONS = config.get_string('POLYAXON_K8S_INGRESS_ANNOTATIONS', is_optional=True)
+K8S_INGRESS_ANNOTATIONS = config.get_string(
+ 'POLYAXON_K8S_INGRESS_ANNOTATIONS', is_optional=True)
TENSORBOARD_PORT_RANGE = [5500, 6500]
# Types
@@ -19,14 +20,16 @@ TYPE_LABELS_CORE = config.get_string('POLYAXON_TYPE_LABELS_CORE')
TYPE_LABELS_EXPERIMENT = config.get_string('POLYAXON_TYPE_LABELS_EXPERIMENT')
# Selectors
-NODE_SELECTORS_EXPERIMENTS = config.get_string('POLYAXON_NODE_SELECTORS_EXPERIMENTS',
- is_optional=True)
+NODE_SELECTORS_EXPERIMENTS = config.get_string(
+ 'POLYAXON_NODE_SELECTORS_EXPERIMENTS', is_optional=True)
JOB_CONTAINER_NAME = config.get_string('POLYAXON_JOB_CONTAINER_NAME')
JOB_SIDECAR_CONTAINER_NAME = config.get_string('POLYAXON_JOB_SIDECAR_CONTAINER_NAME')
-JOB_DOCKER_NAME = config.get_string('POLYAXON_JOB_DOCKER_NAME',
- is_optional=True) or 'polyaxon/polyaxon-lib'
-JOB_SIDECAR_DOCKER_IMAGE = config.get_string('POLYAXON_JOB_SIDECAR_DOCKER_IMAGE',
- is_optional=True) or 'polyaxon/polyaxon-sidecar'
-JOB_SIDECAR_LOG_SLEEP_INTERVAL = config.get_int('POLYAXON_JOB_SIDECAR_LOG_SLEEP_INTERVAL',
- is_optional=True)
+JOB_DOCKER_NAME = config.get_string(
+ 'POLYAXON_JOB_DOCKER_NAME', is_optional=True) or 'polyaxon/polyaxon-lib'
+JOB_SIDECAR_DOCKER_IMAGE = config.get_string(
+ 'POLYAXON_JOB_SIDECAR_DOCKER_IMAGE', is_optional=True) or 'polyaxon/polyaxon-sidecar'
+TENSORBOARD_DOCKER_IMAGE = config.get_string(
+ 'POLYAXON_TENSORBOARD_DOCKER_IMAGE', is_optional=True) or 'tensorflow/tensorflow:1.4.1-py3'
+JOB_SIDECAR_LOG_SLEEP_INTERVAL = config.get_int(
+ 'POLYAXON_JOB_SIDECAR_LOG_SLEEP_INTERVAL', is_optional=True)
