def _default_root_dir():
return get_env(_TRACKING_DIR_ENV_VAR) or os.path.abspath("mlruns")
def _get_experiment_id():
return int(
env.get_env(_EXPERIMENT_ID_ENV_VAR)
or Experiment.DEFAULT_EXPERIMENT_ID)
def is_tracking_uri_set():
"""Returns True if the tracking URI has been set, False otherwise."""
if _tracking_uri or env.get_env(_TRACKING_URI_ENV_VAR):
return True
return False
def _default_table_prefix():
return get_env(_DYNAMODB_TABLE_PREFIX_VAR) or "mlflow"
def _default_endpoint_url():
return get_env(_DYNAMODB_ENDPOINT_URL_VAR)
def _default_root_dir():
return get_env(_TRACKING_DIR_ENV_VAR) or os.path.abspath(
DEFAULT_LOCAL_FILE_AND_ARTIFACT_PATH)
def _get_experiment_id_from_env():
experiment_name = env.get_env(_EXPERIMENT_NAME_ENV_VAR)
if experiment_name is not None:
exp = MlflowClient().get_experiment_by_name(experiment_name)
return exp.experiment_id if exp else None
return env.get_env(_EXPERIMENT_ID_ENV_VAR)
def _get_experiment_id():
return int(_active_experiment_id or env.get_env(_EXPERIMENT_ID_ENV_VAR)
or (env.get_env(_AUTODETECT_EXPERIMENT)
and is_in_databricks_notebook() and get_notebook_id())
or Experiment.DEFAULT_EXPERIMENT_ID)
