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 for mlflow.spark.save_model, mlflow.spark.log_model and mlflow.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 the google-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 the google-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 for sqlalchemy.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 for sqlalchemy.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 for sqlalchemy.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 for sqlalchemy.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)