# Copyright 2018 Iguazio
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
"""
Configuration system.
Configuration can be in either a configuration file specified by
MLRUN_CONFIG_FILE environment variable or by environment variables.
Environment variables are in the format "MLRUN_httpdb__port=8080". This will be
mapped to config.httpdb.port. Values should be in JSON format.
"""
import base64
import binascii
import copy
import json
import os
import typing
import urllib.parse
from collections.abc import Mapping
from distutils.util import strtobool
from os.path import expanduser
from threading import Lock
import dotenv
import semver
import yaml
import mlrun.errors
env_prefix = "MLRUN_"
env_file_key = f"{env_prefix}CONFIG_FILE"
_load_lock = Lock()
_none_type = type(None)
default_config = {
"namespace": "", # default kubernetes namespace
"dbpath": "", # db/api url
# url to nuclio dashboard api (can be with user & token, e.g. https://username:password@dashboard-url.com)
"nuclio_dashboard_url": "",
"nuclio_version": "",
"default_nuclio_runtime": "python:3.7",
"nest_asyncio_enabled": "", # enable import of nest_asyncio for corner cases with old jupyter, set "1"
"ui_url": "", # remote/external mlrun UI url (for hyperlinks) (This is deprecated in favor of the ui block)
"remote_host": "",
"api_base_version": "v1",
"version": "", # will be set to current version
"images_tag": "", # tag to use with mlrun images e.g. mlrun/mlrun (defaults to version)
"images_registry": "", # registry to use with mlrun images e.g. quay.io/ (defaults to empty, for dockerhub)
# comma separated list of images that are in the specified images_registry, and therefore will be enriched with this
# registry when used. default to mlrun/* which means any image which is of the mlrun repository (mlrun/mlrun,
# mlrun/ml-base, etc...)
"images_to_enrich_registry": "^mlrun/*",
"kfp_url": "",
"kfp_ttl": "14400", # KFP ttl in sec, after that completed PODs will be deleted
"kfp_image": "", # image to use for KFP runner (defaults to mlrun/mlrun)
"dask_kfp_image": "", # image to use for dask KFP runner (defaults to mlrun/ml-base)
"igz_version": "", # the version of the iguazio system the API is running on
"iguazio_api_url": "", # the url to iguazio api
"spark_app_image": "", # image to use for spark operator app runtime
"spark_app_image_tag": "", # image tag to use for spark operator app runtime
"spark_history_server_path": "", # spark logs directory for spark history server
"spark_operator_version": "spark-2", # the version of the spark operator in use
"builder_alpine_image": "alpine:3.13.1", # builder alpine image (as kaniko's initContainer)
"package_path": "mlrun", # mlrun pip package
"default_base_image": "mlrun/mlrun", # default base image when doing .deploy()
"default_project": "default", # default project name
"default_archive": "", # default remote archive URL (for build tar.gz)
"mpijob_crd_version": "", # mpijob crd version (e.g: "v1alpha1". must be in: mlrun.runtime.MPIJobCRDVersions)
"hub_url": "https://raw.githubusercontent.com/mlrun/functions/{tag}/{name}/function.yaml",
"ipython_widget": True,
"log_level": "INFO",
# log formatter (options: human | json)
"log_formatter": "human",
"submit_timeout": "180", # timeout when submitting a new k8s resource
# runtimes cleanup interval in seconds
"runtimes_cleanup_interval": "300",
# runs monitoring interval in seconds
"runs_monitoring_interval": "30",
# the grace period (in seconds) that will be given to runtime resources (after they're in terminal state)
# before deleting them
"runtime_resources_deletion_grace_period": "14400",
"scrape_metrics": True,
# sets the background color that is used in printed tables in jupyter
"background_color": "#4EC64B",
"artifact_path": "", # default artifacts path/url
# Add {{workflow.uid}} to artifact_path unless user specified a path manually
"enrich_artifact_path_with_workflow_id": True,
# FIXME: Adding these defaults here so we won't need to patch the "installing component" (provazio-controller) to
# configure this values on field systems, for newer system this will be configured correctly
"v3io_api": "http://v3io-webapi:8081",
"v3io_framesd": "http://framesd:8080",
"datastore": {"async_source_mode": "disabled"},
# default node selector to be applied to all functions - json string base64 encoded format
"default_function_node_selector": "e30=",
# default priority class to be applied to functions running on k8s cluster
"default_function_priority_class_name": "",
# valid options for priority classes - separated by a comma
"valid_function_priority_class_names": "",
# default path prefix for demo data and models
"default_samples_path": "https://s3.wasabisys.com/iguazio/",
# default path for tensorboard logs
"default_tensorboard_logs_path": "/User/.tensorboard/{{project}}",
# ";" separated list of notebook cell tag names to ignore e.g. "ignore-this;ignore-that"
"ignored_notebook_tags": "",
"function_defaults": {
"image_by_kind": {
"job": "mlrun/mlrun",
"serving": "mlrun/mlrun",
"nuclio": "mlrun/mlrun",
"remote": "mlrun/mlrun",
"dask": "mlrun/ml-base",
"mpijob": "mlrun/ml-models",
},
# see enrich_function_preemption_spec for more info,
# and mlrun.api.schemas.functionPreemptionModes for available options
"preemption_mode": "prevent",
},
"httpdb": {
"port": 8080,
"dirpath": expanduser("~/.mlrun/db"),
"dsn": "sqlite:///db/mlrun.db?check_same_thread=false",
"old_dsn": "",
"debug": False,
"user": "",
"password": "",
"token": "",
"logs_path": "./db/logs",
"data_volume": "",
"real_path": "",
"db_type": "sqldb",
"max_workers": 64,
# See mlrun.api.schemas.APIStates for options
"state": "online",
"db": {
"commit_retry_timeout": 30,
"commit_retry_interval": 3,
"conflict_retry_timeout": 15,
"conflict_retry_interval": None,
# Whether to perform data migrations on initialization. enabled or disabled
"data_migrations_mode": "enabled",
# Whether or not to perform database migration from sqlite to mysql on initialization
"database_migration_mode": "enabled",
"backup": {
# Whether or not to use db backups on initialization
"mode": "enabled",
"file_format": "db_backup_%Y%m%d%H%M.db",
"use_rotation": True,
"rotation_limit": 3,
},
# None will set this to be equal to the httpdb.max_workers
"connections_pool_size": None,
"connections_pool_max_overflow": None,
},
"jobs": {
# whether to allow to run local runtimes in the API - configurable to allow the scheduler testing to work
"allow_local_run": False,
},
"authentication": {
"mode": "none", # one of none, basic, bearer, iguazio
"basic": {"username": "", "password": ""},
"bearer": {"token": ""},
"iguazio": {
"session_verification_endpoint": "data_sessions/verifications/app_service",
},
},
"nuclio": {
# One of ClusterIP | NodePort
"default_service_type": "NodePort",
# The following modes apply when user did not configure an ingress
#
# name | description
# ---------------------------------------------------------------------
# never | never enrich with an ingress
# always | always enrich with an ingress, regardless the service type
# onClusterIP | enrich with an ingress only when `mlrun.config.httpdb.nuclio.default_service_type`
# is set to ClusterIP
# ---------------------------------------------------------------------
# Note: adding a mode requires special handling on
# - mlrun.runtimes.constants.NuclioIngressAddTemplatedIngressModes
# - mlrun.runtimes.function.enrich_function_with_ingress
"add_templated_ingress_host_mode": "never",
},
"authorization": {
"mode": "none", # one of none, opa
"opa": {
"address": "",
"request_timeout": 10,
"permission_query_path": "",
"permission_filter_path": "",
"log_level": 0,
},
},
"scheduling": {
# the minimum interval that will be allowed between two scheduled jobs - e.g. a job wouldn't be
# allowed to be scheduled to run more then 2 times in X. Can't be less then 1 minute, "0" to disable
"min_allowed_interval": "10 minutes",
"default_concurrency_limit": 1,
# Firing our jobs include things like creating pods which might not be instant, therefore in the case of
# multiple schedules scheduled to the same time, there might be delays, the default of the scheduler for
# misfire_grace_time is 1 second, we do not want jobs not being scheduled because of the delays so setting
# it to None. the default for coalesce it True just adding it here to be explicit
"scheduler_config": '{"job_defaults": {"misfire_grace_time": null, "coalesce": true}}',
},
"projects": {
"leader": "mlrun",
"followers": "",
# This is used as the interval for the sync loop both when mlrun is leader and follower
"periodic_sync_interval": "1 minute",
"counters_cache_ttl": "2 minutes",
# access key to be used when the leader is iguazio and polling is done from it
"iguazio_access_key": "",
"iguazio_list_projects_default_page_size": 200,
"project_owners_cache_ttl": "30 seconds",
},
# The API needs to know what is its k8s svc url so it could enrich it in the jobs it creates
"api_url": "",
"builder": {
# setting the docker registry to be used for built images, can include the repository as well, e.g.
# index.docker.io/<username>, if not included repository will default to mlrun
"docker_registry": "",
"docker_registry_secret": "",
# whether to allow the docker registry we're pulling from to be insecure. "enabled", "disabled" or "auto"
# which will resolve by the existence of secret
"insecure_pull_registry_mode": "auto",
# whether to allow the docker registry we're pushing to, to be insecure. "enabled", "disabled" or "auto"
# which will resolve by the existence of secret
"insecure_push_registry_mode": "auto",
# the requirement specifier used by the builder when installing mlrun in images when it runs
# pip install <requirement_specifier>, e.g. mlrun==0.5.4, mlrun~=0.5,
# git+https://github.com/mlrun/mlrun@development. by default uses the version
"mlrun_version_specifier": "",
"kaniko_image": "gcr.io/kaniko-project/executor:v1.6.0", # kaniko builder image
"kaniko_init_container_image": "alpine:3.13.1",
# additional docker build args in json encoded base64 format
"build_args": "",
"pip_ca_secret_name": "",
"pip_ca_secret_key": "",
"pip_ca_path": "/etc/ssl/certs/mlrun/pip-ca-certificates.crt",
# template for the prefix that the function target image will be enforced to have (as long as it's targeted
# to be in the configured registry). Supported template values are: {project} {name}
"function_target_image_name_prefix_template": "func-{project}-{name}",
},
"v3io_api": "",
"v3io_framesd": "",
},
"model_endpoint_monitoring": {
"serving_stream_args": {"shard_count": 1, "retention_period_hours": 24},
"drift_thresholds": {"default": {"possible_drift": 0.5, "drift_detected": 0.7}},
"store_prefixes": {
"default": "v3io:///users/pipelines/{project}/model-endpoints/{kind}",
"user_space": "v3io:///projects/{project}/model-endpoints/{kind}",
},
"batch_processing_function_branch": "master",
"parquet_batching_max_events": 10000,
},
"secret_stores": {
"vault": {
# URLs to access Vault. For example, in a local env (Minikube on Mac) these would be:
# http://docker.for.mac.localhost:8200
"url": "",
"remote_url": "",
"role": "",
"token_path": "~/.mlrun/vault",
"project_service_account_name": "mlrun-vault-{project}",
"token_ttl": 180000,
# This config is for debug/testing purposes only!
"user_token": "",
},
"azure_vault": {
"url": "https://{name}.vault.azure.net",
"default_secret_name": None,
"secret_path": "~/.mlrun/azure_vault",
},
"kubernetes": {
# When this is True (the default), all project secrets will be automatically added to each job,
# unless user asks for a specific list of secrets.
"auto_add_project_secrets": True,
"project_secret_name": "mlrun-project-secrets-{project}",
"auth_secret_name": "mlrun-auth-secrets.{hashed_access_key}",
"env_variable_prefix": "MLRUN_K8S_SECRET__",
},
},
"feature_store": {
"data_prefixes": {
"default": "v3io:///projects/{project}/FeatureStore/{name}/{kind}",
"nosql": "v3io:///projects/{project}/FeatureStore/{name}/{kind}",
},
"default_targets": "parquet,nosql",
"default_job_image": "mlrun/mlrun",
"flush_interval": 300,
},
"ui": {
"projects_prefix": "projects", # The UI link prefix for projects
"url": "", # remote/external mlrun UI url (for hyperlinks)
},
"marketplace": {
"k8s_secrets_project_name": "-marketplace-secrets",
"catalog_filename": "catalog.json",
"default_source": {
# Set to false to avoid creating a global source (for example in a dark site)
"create": True,
"name": "mlrun_global_hub",
"description": "MLRun global function hub",
"url": "https://raw.githubusercontent.com/mlrun/marketplace",
"channel": "master",
},
},
"storage": {
# What type of auto-mount to use for functions. Can be one of: none, auto, v3io_credentials, v3io_fuse, pvc.
# Default is auto - which is v3io_credentials when running on Iguazio. If not Iguazio: pvc if the
# MLRUN_PVC_MOUNT env is configured or auto_mount_params contain "pvc_name". Otherwise will do nothing (none).
"auto_mount_type": "auto",
# Extra parameters to pass to the mount call (will be passed as kwargs). Parameters can be either:
# 1. A string of comma-separated parameters, using this format: "param1=value1,param2=value2"
# 2. A base-64 encoded json dictionary containing the list of parameters
"auto_mount_params": "",
},
"default_function_pod_resources": {
"requests": {"cpu": None, "memory": None, "gpu": None},
"limits": {"cpu": None, "memory": None, "gpu": None},
},
# preemptible node selector and tolerations to be added when running on spot nodes
"preemptible_nodes": {
# encoded empty dict
"node_selector": "e30=",
# encoded empty list
"tolerations": "W10=",
},
}
[docs]class Config:
_missing = object()
def __init__(self, cfg=None):
cfg = {} if cfg is None else cfg
# Can't use self._cfg = cfg → infinite recursion
object.__setattr__(self, "_cfg", cfg)
def __getattr__(self, attr):
val = self._cfg.get(attr, self._missing)
if val is self._missing:
raise AttributeError(attr)
if isinstance(val, Mapping):
return self.__class__(val)
return val
def __setattr__(self, attr, value):
# in order for the dbpath setter to work
if attr == "dbpath":
super().__setattr__(attr, value)
else:
self._cfg[attr] = value
def __dir__(self):
return list(self._cfg) + dir(self.__class__)
def __repr__(self):
name = self.__class__.__name__
return f"{name}({self._cfg!r})"
[docs] def update(self, cfg):
for key, value in cfg.items():
if hasattr(self, key):
if isinstance(value, dict):
getattr(self, key).update(value)
else:
setattr(self, key, value)
[docs] def dump_yaml(self, stream=None):
return yaml.dump(self._cfg, stream, default_flow_style=False)
[docs] @classmethod
def from_dict(cls, dict_):
return cls(copy.deepcopy(dict_))
[docs] @staticmethod
def get_build_args():
build_args = {}
if config.httpdb.builder.build_args:
build_args_json = base64.b64decode(
config.httpdb.builder.build_args
).decode()
build_args = json.loads(build_args_json)
return build_args
[docs] @staticmethod
def get_hub_url():
if not config.hub_url.endswith("function.yaml"):
if config.hub_url.startswith("http"):
return f"{config.hub_url}/{{tag}}/{{name}}/function.yaml"
elif config.hub_url.startswith("v3io"):
return f"{config.hub_url}/{{name}}/function.yaml"
return config.hub_url
[docs] @staticmethod
def decode_base64_config_and_load_to_object(
attribute_path: str, expected_type=dict
):
"""
decodes and loads the config attribute to expected type
:param attribute_path: the path in the default_config e.g preemptible_nodes.node_selector
:param expected_type: the object type valid values are : `dict`, `list` etc...
:return: the expected type instance
"""
attributes = attribute_path.split(".")
raw_attribute_value = config
for part in attributes:
try:
raw_attribute_value = raw_attribute_value.__getattr__(part)
except AttributeError:
raise mlrun.errors.MLRunNotFoundError(
"Attribute does not exist in config"
)
# There is a bug in the installer component in iguazio system that causes the configured value to be base64 of
# null (without conditioning it we will end up returning None instead of empty dict)
if raw_attribute_value and raw_attribute_value != "bnVsbA==":
try:
decoded_attribute_value = base64.b64decode(raw_attribute_value).decode()
except Exception:
raise mlrun.errors.MLRunInvalidArgumentTypeError(
f"Unable to decode {attribute_path}"
)
parsed_attribute_value = json.loads(decoded_attribute_value)
if type(parsed_attribute_value) != expected_type:
raise mlrun.errors.MLRunInvalidArgumentTypeError(
f"Expected type {expected_type}, got {type(parsed_attribute_value)}"
)
return parsed_attribute_value
return expected_type()
[docs] def get_default_function_node_selector(self) -> dict:
return self.decode_base64_config_and_load_to_object(
"default_function_node_selector", dict
)
[docs] def get_preemptible_node_selector(self) -> dict:
return self.decode_base64_config_and_load_to_object(
"preemptible_nodes.node_selector", dict
)
[docs] def get_preemptible_tolerations(self) -> list:
return self.decode_base64_config_and_load_to_object(
"preemptible_nodes.tolerations", list
)
[docs] @staticmethod
def get_valid_function_priority_class_names():
valid_function_priority_class_names = []
if not config.valid_function_priority_class_names:
return valid_function_priority_class_names
# Manually ensure we have only unique values because we want to keep the order and using a set would lose it
for priority_class_name in config.valid_function_priority_class_names.split(
","
):
if priority_class_name not in valid_function_priority_class_names:
valid_function_priority_class_names.append(priority_class_name)
return valid_function_priority_class_names
[docs] @staticmethod
def get_parsed_igz_version() -> typing.Optional[semver.VersionInfo]:
if not config.igz_version:
return None
try:
parsed_version = semver.VersionInfo.parse(config.igz_version)
return parsed_version
except ValueError:
# iguazio version is semver compatible only from 3.2, before that it will be something
# like 3.0_b177_20210806003728
semver_compatible_igz_version = config.igz_version.split("_")[0]
return semver.VersionInfo.parse(f"{semver_compatible_igz_version}.0")
[docs] def resolve_kfp_url(self, namespace=None):
if config.kfp_url:
return config.kfp_url
igz_version = self.get_parsed_igz_version()
# TODO: When Iguazio 3.4 will deprecate we can remove this line
if igz_version and igz_version <= semver.VersionInfo.parse("3.6.0-b1"):
if namespace is None:
if not config.namespace:
raise mlrun.errors.MLRunNotFoundError(
"For KubeFlow Pipelines to function, a namespace must be configured"
)
namespace = config.namespace
# When instead of host we provided namespace we tackled this issue
# https://github.com/canonical/bundle-kubeflow/issues/412
# TODO: When we'll move to kfp 1.4.0 (server side) it should be resolved
return f"http://ml-pipeline.{namespace}.svc.cluster.local:8888"
return None
[docs] @staticmethod
def get_storage_auto_mount_params():
auto_mount_params = {}
if config.storage.auto_mount_params:
try:
auto_mount_params = base64.b64decode(
config.storage.auto_mount_params, validate=True
).decode()
auto_mount_params = json.loads(auto_mount_params)
except binascii.Error:
# Importing here to avoid circular dependencies
from .utils import list2dict
# String wasn't base64 encoded. Parse it using a 'p1=v1,p2=v2' format.
mount_params = config.storage.auto_mount_params.split(",")
auto_mount_params = list2dict(mount_params)
if not isinstance(auto_mount_params, dict):
raise TypeError(
f"data in storage.auto_mount_params does not resolve to a dictionary: {auto_mount_params}"
)
return auto_mount_params
[docs] def get_default_function_pod_resources(self):
resources = {}
resource_requirements = ["requests", "limits"]
for requirement in resource_requirements:
resources[
requirement
] = self.get_default_function_pod_requirement_resources(requirement)
return resources
[docs] @staticmethod
def get_default_function_pod_requirement_resources(
requirement: str, with_gpu: bool = True
):
"""
:param requirement: kubernetes requirement resource one of the following : requests, limits
:param with_gpu: whether to return requirement resources with nvidia.com/gpu field (e.g you cannot specify GPU
requests without specifying GPU limits) https://kubernetes.io/docs/tasks/manage-gpus/scheduling-gpus/
:return: a dict containing the defaults resources (cpu, memory, nvidia.com/gpu)
"""
resources: dict = copy.deepcopy(config.default_function_pod_resources.to_dict())
gpu_type = "nvidia.com/gpu"
gpu = "gpu"
resource_requirement = resources.get(requirement, {})
resource_requirement.setdefault(gpu)
if with_gpu:
resource_requirement[gpu_type] = resource_requirement.pop(gpu)
else:
resource_requirement.pop(gpu)
return resource_requirement
[docs] def to_dict(self):
return copy.copy(self._cfg)
[docs] @staticmethod
def reload():
_populate()
@property
def version(self):
# importing here to avoid circular dependency
from mlrun.utils.version import Version
return Version().get()["version"]
@property
def kfp_image(self):
"""
When this configuration is not set we want to set it to mlrun/mlrun, but we need to use the enrich_image method.
The problem is that the mlrun.utils.helpers module is importing the config (this) module, so we must import the
module inside this function (and not on initialization), and then calculate this property value here.
"""
if not self._kfp_image:
# importing here to avoid circular dependency
import mlrun.utils.helpers
return mlrun.utils.helpers.enrich_image_url("mlrun/mlrun")
return self._kfp_image
@kfp_image.setter
def kfp_image(self, value):
self._kfp_image = value
@property
def dask_kfp_image(self):
"""
See kfp_image property docstring for why we're defining this property
"""
if not self._dask_kfp_image:
# importing here to avoid circular dependency
import mlrun.utils.helpers
return mlrun.utils.helpers.enrich_image_url("mlrun/ml-base")
return self._dask_kfp_image
@dask_kfp_image.setter
def dask_kfp_image(self, value):
self._dask_kfp_image = value
[docs] @staticmethod
def resolve_ui_url():
# ui_url is deprecated in favor of the ui.url (we created the ui block)
# since the config class is used in a "recursive" way, we can't use property like we used in other places
# since the property will need to be url, which exists in other structs as well
return config.ui.url or config.ui_url
@property
def dbpath(self):
return self._dbpath
@dbpath.setter
def dbpath(self, value):
self._dbpath = value
if value:
# importing here to avoid circular dependency
import mlrun.db
# when dbpath is set we want to connect to it which will sync configuration from it to the client
mlrun.db.get_run_db(value, force_reconnect=True)
@property
def iguazio_api_url(self):
"""
we want to be able to run with old versions of the service who runs the API (which doesn't configure this
value) so we're doing best effort to try and resolve it from other configurations
TODO: Remove this hack when 0.6.x is old enough
"""
if not self._iguazio_api_url:
if self.httpdb.builder.docker_registry and self.igz_version:
return self._extract_iguazio_api_from_docker_registry_url()
return self._iguazio_api_url
def _extract_iguazio_api_from_docker_registry_url(self):
docker_registry_url = self.httpdb.builder.docker_registry
# add schema otherwise parsing go wrong
if "://" not in docker_registry_url:
docker_registry_url = f"http://{docker_registry_url}"
parsed_registry_url = urllib.parse.urlparse(docker_registry_url)
registry_hostname = parsed_registry_url.hostname
# replace the first domain section (app service name) with dashboard
first_dot_index = registry_hostname.find(".")
if first_dot_index < 0:
# if not found it's not the format we know - can't resolve the api url from the registry url
return ""
return f"https://dashboard{registry_hostname[first_dot_index:]}"
@iguazio_api_url.setter
def iguazio_api_url(self, value):
self._iguazio_api_url = value
# Global configuration
config = Config.from_dict(default_config)
def _populate():
"""Populate configuration from config file (if exists in environment) and
from environment variables.
populate will run only once, after first call it does nothing.
"""
global _loaded
with _load_lock:
_do_populate()
def _do_populate(env=None):
global config
if "MLRUN_ENV_FILE" in os.environ:
dotenv.load_dotenv(os.environ["MLRUN_ENV_FILE"], override=True)
if not config:
config = Config.from_dict(default_config)
else:
config.update(default_config)
config_path = os.environ.get(env_file_key)
if config_path:
with open(config_path) as fp:
data = yaml.safe_load(fp)
if not isinstance(data, dict):
raise TypeError(f"configuration in {config_path} not a dict")
config.update(data)
data = read_env(env)
if data:
config.update(data)
# HACK to enable config property to both have dynamic default and to use the value from dict/env like other
# configurations - we just need a key in the dict that is different than the property name, so simply adding prefix
# underscore
config._cfg["_kfp_image"] = config._cfg["kfp_image"]
del config._cfg["kfp_image"]
config._cfg["_dask_kfp_image"] = config._cfg["dask_kfp_image"]
del config._cfg["dask_kfp_image"]
config._cfg["_iguazio_api_url"] = config._cfg["iguazio_api_url"]
del config._cfg["iguazio_api_url"]
_validate_config(config)
def _validate_config(config):
import mlrun.k8s_utils
try:
limits_gpu = config.default_function_pod_resources.limits.gpu
requests_gpu = config.default_function_pod_resources.requests.gpu
mlrun.k8s_utils.verify_gpu_requests_and_limits(
requests_gpu=requests_gpu,
limits_gpu=limits_gpu,
)
except AttributeError:
pass
def _convert_resources_to_str(config: dict = None):
resources_types = ["cpu", "memory", "gpu"]
resource_requirements = ["requests", "limits"]
if not config.get("default_function_pod_resources"):
return
for requirement in resource_requirements:
resource_requirement = config.get("default_function_pod_resources").get(
requirement
)
if not resource_requirement:
continue
for resource_type in resources_types:
value = resource_requirement.setdefault(resource_type, None)
if value is None:
continue
resource_requirement[resource_type] = str(value)
def _convert_str(value, typ):
if typ in (str, _none_type):
return value
if typ is bool:
return strtobool(value)
# e.g. int('8080') → 8080
return typ(value)
[docs]def read_env(env=None, prefix=env_prefix):
"""Read configuration from environment"""
env = os.environ if env is None else env
config = {}
for key, value in env.items():
if not key.startswith(env_prefix) or key == env_file_key:
continue
try:
value = json.loads(value) # values can be JSON encoded
except ValueError:
pass # Leave as string
key = key[len(env_prefix) :] # Trim MLRUN_
path = key.lower().split("__") # 'A__B' → ['a', 'b']
cfg = config
while len(path) > 1:
name, *path = path
cfg = cfg.setdefault(name, {})
cfg[path[0]] = value
env_dbpath = env.get("MLRUN_DBPATH", "")
is_remote_mlrun = (
env_dbpath.startswith("https://mlrun-api.") and "tenant." in env_dbpath
)
# It's already a standard to set this env var to configure the v3io api, so we're supporting it (instead
# of MLRUN_V3IO_API), in remote usage this can be auto detected from the DBPATH
v3io_api = env.get("V3IO_API")
if v3io_api:
config["v3io_api"] = v3io_api
elif is_remote_mlrun:
config["v3io_api"] = env_dbpath.replace("https://mlrun-api.", "https://webapi.")
# It's already a standard to set this env var to configure the v3io framesd, so we're supporting it (instead
# of MLRUN_V3IO_FRAMESD), in remote usage this can be auto detected from the DBPATH
v3io_framesd = env.get("V3IO_FRAMESD")
if v3io_framesd:
config["v3io_framesd"] = v3io_framesd
elif is_remote_mlrun:
config["v3io_framesd"] = env_dbpath.replace(
"https://mlrun-api.", "https://framesd."
)
uisvc = env.get("MLRUN_UI_SERVICE_HOST")
igz_domain = env.get("IGZ_NAMESPACE_DOMAIN")
# workaround to try and detect IGZ domain
if not igz_domain and "MLRUN_HTTPDB__BUILDER__DOCKER_REGISTRY" in env:
registry = env["MLRUN_HTTPDB__BUILDER__DOCKER_REGISTRY"]
if registry.startswith("docker-registry.default-tenant"):
igz_domain = registry[len("docker-registry.") :]
if ":" in igz_domain:
igz_domain = igz_domain[: igz_domain.rfind(":")]
env["IGZ_NAMESPACE_DOMAIN"] = igz_domain
# workaround wrongly sqldb dsn in 2.8
if (
config.get("httpdb", {}).get("dsn")
== "sqlite:///mlrun.sqlite3?check_same_thread=false"
):
config["httpdb"]["dsn"] = "sqlite:////mlrun/db/mlrun.db?check_same_thread=false"
# "disabled" is the helm chart default value, we don't want that value to be set cause when this value is set we
# use it in calls to the Nuclio package, and when the Nuclio package receives a value it simply uses it, and
# obviously "disabled" is not the right address.. when the Nuclio package doesn't receive a value it doing "best
# effort" to try and determine the URL, we want this "best effort" so overriding the "disabled" value
if config.get("nuclio_dashboard_url") == "disabled":
config["nuclio_dashboard_url"] = ""
if uisvc and not config.get("ui_url"):
if igz_domain:
config["ui_url"] = f"https://mlrun-ui.{igz_domain}"
if config.get("log_level"):
import mlrun.utils.logger
# logger created (because of imports mess) before the config is loaded (in tests), therefore we're changing its
# level manually
mlrun.utils.logger.set_logger_level(config["log_level"])
# The default function pod resource values are of type str; however, when reading from environment variable numbers,
# it converts them to type int if contains only number, so we want to convert them to str.
_convert_resources_to_str(config)
return config
_populate()
