mlflow.environment_variables
This module defines environment variables used in MLflow.
-
mlflow.environment_variables.
MLFLOW_DFS_TMP
= 'MLFLOW_DFS_TMP' Specifies the
dfs_tmpdir
parameter to use formlflow.spark.save_model
,mlflow.spark.log_model
andmlflow.spark.load_model
. See https://www.mlflow.org/docs/latest/python_api/mlflow.spark.html#mlflow.spark.save_model for more information. (default:/tmp/mlflow
)
-
mlflow.environment_variables.
MLFLOW_GCS_DEFAULT_TIMEOUT
= 'MLFLOW_GCS_DEFAULT_TIMEOUT' Specifies the default timeout to use when downloading/uploading a file from/to GCS (default:
None
). If None,google.cloud.storage.constants._DEFAULT_TIMEOUT
is used.
-
mlflow.environment_variables.
MLFLOW_GCS_DOWNLOAD_CHUNK_SIZE
= 'MLFLOW_GCS_DOWNLOAD_CHUNK_SIZE' Specifies the chunk size to use when downloading a file from GCS (default:
None
). If None, the chunk size is automatically determined by thegoogle-cloud-storage
package.
-
mlflow.environment_variables.
MLFLOW_GCS_UPLOAD_CHUNK_SIZE
= 'MLFLOW_GCS_UPLOAD_CHUNK_SIZE' Specifies the chunk size to use when uploading a file to GCS. (default:
None
). If None, the chunk size is automatically determined by thegoogle-cloud-storage
package.
-
mlflow.environment_variables.
MLFLOW_HTTP_REQUEST_BACKOFF_FACTOR
= 'MLFLOW_HTTP_REQUEST_BACKOFF_FACTOR' Specifies the backoff increase factor between MLflow HTTP request failures (default:
2
)
-
mlflow.environment_variables.
MLFLOW_HTTP_REQUEST_MAX_RETRIES
= 'MLFLOW_HTTP_REQUEST_MAX_RETRIES' Specifies the maximum number of retries for MLflow HTTP requests (default:
5
)
-
mlflow.environment_variables.
MLFLOW_HTTP_REQUEST_TIMEOUT
= 'MLFLOW_HTTP_REQUEST_TIMEOUT' Specifies the timeout in seconds for MLflow HTTP requests (default:
120
)
-
mlflow.environment_variables.
MLFLOW_KERBEROS_TICKET_CACHE
= 'MLFLOW_KERBEROS_TICKET_CACHE' Specifies the location of a Kerberos ticket cache to use for HDFS artifact operations. (default:
None
)
-
mlflow.environment_variables.
MLFLOW_KERBEROS_USER
= 'MLFLOW_KERBEROS_USER' Specifies a Kerberos user for HDFS artifact operations. (default:
None
)
-
mlflow.environment_variables.
MLFLOW_PYARROW_EXTRA_CONF
= 'MLFLOW_PYARROW_EXTRA_CONF' Specifies extra pyarrow configurations for HDFS artifact operations. (default:
None
)
-
mlflow.environment_variables.
MLFLOW_S3_ENDPOINT_URL
= 'MLFLOW_S3_ENDPOINT_URL' Specifies the S3 endpoint URL to use for S3 artifact operations. (default:
None
)
-
mlflow.environment_variables.
MLFLOW_S3_IGNORE_TLS
= 'MLFLOW_S3_IGNORE_TLS' Specifies whether or not to skip TLS certificate verification for S3 artifact operations. (default:
False
)
-
mlflow.environment_variables.
MLFLOW_S3_UPLOAD_EXTRA_ARGS
= 'MLFLOW_S3_UPLOAD_EXTRA_ARGS' Specifies extra arguments for S3 artifact uploads. (default:
None
)
-
mlflow.environment_variables.
MLFLOW_SQLALCHEMYSTORE_ECHO
= 'MLFLOW_SQLALCHEMYSTORE_ECHO' Specifies the
echo
parameter to use forsqlalchemy.create_engine
in the SQLAlchemy tracking store. See https://docs.sqlalchemy.org/en/14/core/engines.html#sqlalchemy.create_engine.params.echo for more information. (default:False
)
-
mlflow.environment_variables.
MLFLOW_SQLALCHEMYSTORE_MAX_OVERFLOW
= 'MLFLOW_SQLALCHEMYSTORE_MAX_OVERFLOW' Specifies the
max_overflow
parameter to use forsqlalchemy.create_engine
in the SQLAlchemy tracking store. See https://docs.sqlalchemy.org/en/14/core/engines.html#sqlalchemy.create_engine.params.max_overflow for more information. (default:None
)
-
mlflow.environment_variables.
MLFLOW_SQLALCHEMYSTORE_POOL_RECYCLE
= 'MLFLOW_SQLALCHEMYSTORE_POOL_RECYCLE' Specifies the
pool_recycle
parameter to use forsqlalchemy.create_engine
in the SQLAlchemy tracking store. See https://docs.sqlalchemy.org/en/14/core/engines.html#sqlalchemy.create_engine.params.pool_recycle for more information. (default:None
)
-
mlflow.environment_variables.
MLFLOW_SQLALCHEMYSTORE_POOL_SIZE
= 'MLFLOW_SQLALCHEMYSTORE_POOL_SIZE' Specifies the
pool_size
parameter to use forsqlalchemy.create_engine
in the SQLAlchemy tracking store. See https://docs.sqlalchemy.org/en/14/core/engines.html#sqlalchemy.create_engine.params.pool_size for more information. (default:None
)