LICENSE.txt
README.rst
pyproject.toml
mlflow/__init__.py
mlflow/__main__.py
mlflow/cli.py
mlflow/client.py
mlflow/db.py
mlflow/environment_variables.py
mlflow/exceptions.py
mlflow/experiments.py
mlflow/ml_package_versions.py
mlflow/pypi_package_index.json
mlflow/runs.py
mlflow/version.py
mlflow.egg-info/PKG-INFO
mlflow.egg-info/SOURCES.txt
mlflow.egg-info/dependency_links.txt
mlflow.egg-info/entry_points.txt
mlflow.egg-info/requires.txt
mlflow.egg-info/top_level.txt
mlflow/artifacts/__init__.py
mlflow/azure/__init__.py
mlflow/azure/client.py
mlflow/catboost/__init__.py
mlflow/config/__init__.py
mlflow/data/__init__.py
mlflow/data/artifact_dataset_sources.py
mlflow/data/code_dataset_source.py
mlflow/data/dataset.py
mlflow/data/dataset_registry.py
mlflow/data/dataset_source.py
mlflow/data/dataset_source_registry.py
mlflow/data/delta_dataset_source.py
mlflow/data/digest_utils.py
mlflow/data/filesystem_dataset_source.py
mlflow/data/http_dataset_source.py
mlflow/data/huggingface_dataset.py
mlflow/data/huggingface_dataset_source.py
mlflow/data/meta_dataset.py
mlflow/data/numpy_dataset.py
mlflow/data/pandas_dataset.py
mlflow/data/pyfunc_dataset_mixin.py
mlflow/data/schema.py
mlflow/data/sources.py
mlflow/data/spark_dataset.py
mlflow/data/spark_dataset_source.py
mlflow/data/spark_delta_utils.py
mlflow/data/tensorflow_dataset.py
mlflow/data/uc_volume_dataset_source.py
mlflow/deployments/__init__.py
mlflow/deployments/base.py
mlflow/deployments/cli.py
mlflow/deployments/constants.py
mlflow/deployments/interface.py
mlflow/deployments/plugin_manager.py
mlflow/deployments/utils.py
mlflow/deployments/databricks/__init__.py
mlflow/deployments/mlflow/__init__.py
mlflow/deployments/openai/__init__.py
mlflow/deployments/server/__init__.py
mlflow/deployments/server/app.py
mlflow/deployments/server/config.py
mlflow/deployments/server/constants.py
mlflow/deployments/server/runner.py
mlflow/diviner/__init__.py
mlflow/entities/__init__.py
mlflow/entities/_mlflow_object.py
mlflow/entities/dataset.py
mlflow/entities/dataset_input.py
mlflow/entities/dataset_summary.py
mlflow/entities/experiment.py
mlflow/entities/experiment_tag.py
mlflow/entities/file_info.py
mlflow/entities/input_tag.py
mlflow/entities/lifecycle_stage.py
mlflow/entities/metric.py
mlflow/entities/multipart_upload.py
mlflow/entities/param.py
mlflow/entities/run.py
mlflow/entities/run_data.py
mlflow/entities/run_info.py
mlflow/entities/run_inputs.py
mlflow/entities/run_status.py
mlflow/entities/run_tag.py
mlflow/entities/source_type.py
mlflow/entities/view_type.py
mlflow/entities/model_registry/__init__.py
mlflow/entities/model_registry/_model_registry_entity.py
mlflow/entities/model_registry/model_version.py
mlflow/entities/model_registry/model_version_stages.py
mlflow/entities/model_registry/model_version_status.py
mlflow/entities/model_registry/model_version_tag.py
mlflow/entities/model_registry/registered_model.py
mlflow/entities/model_registry/registered_model_alias.py
mlflow/entities/model_registry/registered_model_tag.py
mlflow/fastai/__init__.py
mlflow/fastai/callback.py
mlflow/gateway/__init__.py
mlflow/gateway/app.py
mlflow/gateway/base_models.py
mlflow/gateway/cli.py
mlflow/gateway/client.py
mlflow/gateway/config.py
mlflow/gateway/constants.py
mlflow/gateway/exceptions.py
mlflow/gateway/fluent.py
mlflow/gateway/utils.py
mlflow/gateway/providers/__init__.py
mlflow/gateway/providers/ai21labs.py
mlflow/gateway/providers/anthropic.py
mlflow/gateway/providers/base.py
mlflow/gateway/providers/bedrock.py
mlflow/gateway/providers/cohere.py
mlflow/gateway/providers/huggingface.py
mlflow/gateway/providers/mistral.py
mlflow/gateway/providers/mlflow.py
mlflow/gateway/providers/mosaicml.py
mlflow/gateway/providers/openai.py
mlflow/gateway/providers/palm.py
mlflow/gateway/providers/utils.py
mlflow/gateway/schemas/__init__.py
mlflow/gateway/schemas/chat.py
mlflow/gateway/schemas/completions.py
mlflow/gateway/schemas/embeddings.py
mlflow/gluon/__init__.py
mlflow/gluon/_autolog.py
mlflow/h2o/__init__.py
mlflow/johnsnowlabs/__init__.py
mlflow/keras/__init__.py
mlflow/keras/autologging.py
mlflow/keras/callback.py
mlflow/keras/load.py
mlflow/keras/save.py
mlflow/keras/utils.py
mlflow/langchain/__init__.py
mlflow/langchain/_langchain_autolog.py
mlflow/langchain/_rag_utils.py
mlflow/langchain/api_request_parallel_processor.py
mlflow/langchain/databricks_dependencies.py
mlflow/langchain/retriever_chain.py
mlflow/langchain/runnables.py
mlflow/langchain/utils.py
mlflow/legacy_databricks_cli/__init__.py
mlflow/legacy_databricks_cli/configure/__init__.py
mlflow/legacy_databricks_cli/configure/provider.py
mlflow/lightgbm/__init__.py
mlflow/metrics/__init__.py
mlflow/metrics/base.py
mlflow/metrics/metric_definitions.py
mlflow/metrics/genai/__init__.py
mlflow/metrics/genai/base.py
mlflow/metrics/genai/genai_metric.py
mlflow/metrics/genai/metric_definitions.py
mlflow/metrics/genai/model_utils.py
mlflow/metrics/genai/prompt_template.py
mlflow/metrics/genai/utils.py
mlflow/metrics/genai/prompts/__init__.py
mlflow/metrics/genai/prompts/v1.py
mlflow/mleap/__init__.py
mlflow/models/__init__.py
mlflow/models/cli.py
mlflow/models/docker_utils.py
mlflow/models/flavor_backend.py
mlflow/models/flavor_backend_registry.py
mlflow/models/model.py
mlflow/models/python_api.py
mlflow/models/signature.py
mlflow/models/utils.py
mlflow/models/wheeled_model.py
mlflow/models/container/__init__.py
mlflow/models/container/scoring_server/__init__.py
mlflow/models/container/scoring_server/wsgi.py
mlflow/models/evaluation/__init__.py
mlflow/models/evaluation/_shap_patch.py
mlflow/models/evaluation/artifacts.py
mlflow/models/evaluation/base.py
mlflow/models/evaluation/default_evaluator.py
mlflow/models/evaluation/evaluator_registry.py
mlflow/models/evaluation/lift_curve.py
mlflow/models/evaluation/validation.py
mlflow/onnx/__init__.py
mlflow/openai/__init__.py
mlflow/openai/api_request_parallel_processor.py
mlflow/paddle/__init__.py
mlflow/paddle/_paddle_autolog.py
mlflow/pmdarima/__init__.py
mlflow/projects/__init__.py
mlflow/projects/_project_spec.py
mlflow/projects/databricks.py
mlflow/projects/docker.py
mlflow/projects/env_type.py
mlflow/projects/kubernetes.py
mlflow/projects/submitted_run.py
mlflow/projects/utils.py
mlflow/projects/backend/__init__.py
mlflow/projects/backend/abstract_backend.py
mlflow/projects/backend/loader.py
mlflow/projects/backend/local.py
mlflow/promptflow/__init__.py
mlflow/promptlab/__init__.py
mlflow/prophet/__init__.py
mlflow/protos/__init__.py
mlflow/protos/databricks_artifacts_pb2.py
mlflow/protos/databricks_filesystem_service_pb2.py
mlflow/protos/databricks_managed_catalog_messages_pb2.py
mlflow/protos/databricks_managed_catalog_service_pb2.py
mlflow/protos/databricks_pb2.py
mlflow/protos/databricks_uc_registry_messages_pb2.py
mlflow/protos/databricks_uc_registry_service_pb2.py
mlflow/protos/facet_feature_statistics_pb2.py
mlflow/protos/internal_pb2.py
mlflow/protos/mlflow_artifacts_pb2.py
mlflow/protos/model_registry_pb2.py
mlflow/protos/service_pb2.py
mlflow/protos/scalapb/__init__.py
mlflow/protos/scalapb/scalapb_pb2.py
mlflow/pyfunc/__init__.py
mlflow/pyfunc/_mlflow_pyfunc_backend_predict.py
mlflow/pyfunc/backend.py
mlflow/pyfunc/mlserver.py
mlflow/pyfunc/model.py
mlflow/pyfunc/spark_model_cache.py
mlflow/pyfunc/stdin_server.py
mlflow/pyfunc/loaders/__init__.py
mlflow/pyfunc/loaders/chat_model.py
mlflow/pyfunc/scoring_server/__init__.py
mlflow/pyfunc/scoring_server/client.py
mlflow/pyfunc/scoring_server/wsgi.py
mlflow/pyspark/__init__.py
mlflow/pyspark/ml/__init__.py
mlflow/pyspark/ml/_autolog.py
mlflow/pyspark/ml/log_model_allowlist.txt
mlflow/pytorch/__init__.py
mlflow/pytorch/_lightning_autolog.py
mlflow/pytorch/_pytorch_autolog.py
mlflow/pytorch/pickle_module.py
mlflow/recipes/__init__.py
mlflow/recipes/artifacts.py
mlflow/recipes/cli.py
mlflow/recipes/dag_help_strings.py
mlflow/recipes/recipe.py
mlflow/recipes/step.py
mlflow/recipes/cards/__init__.py
mlflow/recipes/cards/histogram_generator.py
mlflow/recipes/cards/pandas_renderer.py
mlflow/recipes/cards/templates/__init__.py
mlflow/recipes/cards/templates/base.html
mlflow/recipes/classification/__init__.py
mlflow/recipes/classification/v1/__init__.py
mlflow/recipes/classification/v1/recipe.py
mlflow/recipes/regression/__init__.py
mlflow/recipes/regression/v1/__init__.py
mlflow/recipes/regression/v1/recipe.py
mlflow/recipes/resources/recipe_dag_template.html
mlflow/recipes/steps/__init__.py
mlflow/recipes/steps/evaluate.py
mlflow/recipes/steps/predict.py
mlflow/recipes/steps/register.py
mlflow/recipes/steps/split.py
mlflow/recipes/steps/train.py
mlflow/recipes/steps/transform.py
mlflow/recipes/steps/automl/__init__.py
mlflow/recipes/steps/automl/flaml.py
mlflow/recipes/steps/ingest/__init__.py
mlflow/recipes/steps/ingest/datasets.py
mlflow/recipes/utils/__init__.py
mlflow/recipes/utils/execution.py
mlflow/recipes/utils/metrics.py
mlflow/recipes/utils/step.py
mlflow/recipes/utils/tracking.py
mlflow/recipes/utils/wrapped_recipe_model.py
mlflow/rfunc/__init__.py
mlflow/rfunc/backend.py
mlflow/sagemaker/__init__.py
mlflow/sagemaker/cli.py
mlflow/sentence_transformers/__init__.py
mlflow/server/__init__.py
mlflow/server/handlers.py
mlflow/server/prometheus_exporter.py
mlflow/server/validation.py
mlflow/server/auth/__init__.py
mlflow/server/auth/__main__.py
mlflow/server/auth/basic_auth.ini
mlflow/server/auth/cli.py
mlflow/server/auth/client.py
mlflow/server/auth/config.py
mlflow/server/auth/entities.py
mlflow/server/auth/logo.py
mlflow/server/auth/permissions.py
mlflow/server/auth/routes.py
mlflow/server/auth/sqlalchemy_store.py
mlflow/server/auth/db/__init__.py
mlflow/server/auth/db/cli.py
mlflow/server/auth/db/models.py
mlflow/server/auth/db/utils.py
mlflow/server/auth/db/migrations/__init__.py
mlflow/server/auth/db/migrations/alembic.ini
mlflow/server/auth/db/migrations/env.py
mlflow/server/auth/db/migrations/versions/8606fa83a998_initial_migration.py
mlflow/server/auth/db/migrations/versions/__init__.py
mlflow/server/graphql/__init__.py
mlflow/server/graphql/autogenerated_graphql_schema.py
mlflow/server/graphql/graphql_custom_scalars.py
mlflow/server/graphql/graphql_schema_extensions.py
mlflow/server/js/build/asset-manifest.json
mlflow/server/js/build/favicon.ico
mlflow/server/js/build/index.html
mlflow/server/js/build/manifest.json
mlflow/server/js/build/pdf.worker.js
mlflow/server/js/node_modules/node-gyp/update-gyp.py
mlflow/server/js/node_modules/node-gyp/gyp/gyp_main.py
mlflow/server/js/node_modules/node-gyp/gyp/setup.py
mlflow/server/js/node_modules/node-gyp/gyp/test_gyp.py
mlflow/server/js/node_modules/node-gyp/gyp/pylib/gyp/MSVSNew.py
mlflow/server/js/node_modules/node-gyp/gyp/pylib/gyp/MSVSProject.py
mlflow/server/js/node_modules/node-gyp/gyp/pylib/gyp/MSVSSettings.py
mlflow/server/js/node_modules/node-gyp/gyp/pylib/gyp/MSVSSettings_test.py
mlflow/server/js/node_modules/node-gyp/gyp/pylib/gyp/MSVSToolFile.py
mlflow/server/js/node_modules/node-gyp/gyp/pylib/gyp/MSVSUserFile.py
mlflow/server/js/node_modules/node-gyp/gyp/pylib/gyp/MSVSUtil.py
mlflow/server/js/node_modules/node-gyp/gyp/pylib/gyp/MSVSVersion.py
mlflow/server/js/node_modules/node-gyp/gyp/pylib/gyp/__init__.py
mlflow/server/js/node_modules/node-gyp/gyp/pylib/gyp/common.py
mlflow/server/js/node_modules/node-gyp/gyp/pylib/gyp/common_test.py
mlflow/server/js/node_modules/node-gyp/gyp/pylib/gyp/easy_xml.py
mlflow/server/js/node_modules/node-gyp/gyp/pylib/gyp/easy_xml_test.py
mlflow/server/js/node_modules/node-gyp/gyp/pylib/gyp/flock_tool.py
mlflow/server/js/node_modules/node-gyp/gyp/pylib/gyp/input.py
mlflow/server/js/node_modules/node-gyp/gyp/pylib/gyp/input_test.py
mlflow/server/js/node_modules/node-gyp/gyp/pylib/gyp/mac_tool.py
mlflow/server/js/node_modules/node-gyp/gyp/pylib/gyp/msvs_emulation.py
mlflow/server/js/node_modules/node-gyp/gyp/pylib/gyp/ninja_syntax.py
mlflow/server/js/node_modules/node-gyp/gyp/pylib/gyp/simple_copy.py
mlflow/server/js/node_modules/node-gyp/gyp/pylib/gyp/win_tool.py
mlflow/server/js/node_modules/node-gyp/gyp/pylib/gyp/xcode_emulation.py
mlflow/server/js/node_modules/node-gyp/gyp/pylib/gyp/xcode_ninja.py
mlflow/server/js/node_modules/node-gyp/gyp/pylib/gyp/xcodeproj_file.py
mlflow/server/js/node_modules/node-gyp/gyp/pylib/gyp/xml_fix.py
mlflow/server/js/node_modules/node-gyp/gyp/pylib/gyp/generator/__init__.py
mlflow/server/js/node_modules/node-gyp/gyp/pylib/gyp/generator/analyzer.py
mlflow/server/js/node_modules/node-gyp/gyp/pylib/gyp/generator/android.py
mlflow/server/js/node_modules/node-gyp/gyp/pylib/gyp/generator/cmake.py
mlflow/server/js/node_modules/node-gyp/gyp/pylib/gyp/generator/compile_commands_json.py
mlflow/server/js/node_modules/node-gyp/gyp/pylib/gyp/generator/dump_dependency_json.py
mlflow/server/js/node_modules/node-gyp/gyp/pylib/gyp/generator/eclipse.py
mlflow/server/js/node_modules/node-gyp/gyp/pylib/gyp/generator/gypd.py
mlflow/server/js/node_modules/node-gyp/gyp/pylib/gyp/generator/gypsh.py
mlflow/server/js/node_modules/node-gyp/gyp/pylib/gyp/generator/make.py
mlflow/server/js/node_modules/node-gyp/gyp/pylib/gyp/generator/msvs.py
mlflow/server/js/node_modules/node-gyp/gyp/pylib/gyp/generator/msvs_test.py
mlflow/server/js/node_modules/node-gyp/gyp/pylib/gyp/generator/ninja.py
mlflow/server/js/node_modules/node-gyp/gyp/pylib/gyp/generator/ninja_test.py
mlflow/server/js/node_modules/node-gyp/gyp/pylib/gyp/generator/xcode.py
mlflow/server/js/node_modules/node-gyp/gyp/pylib/gyp/generator/xcode_test.py
mlflow/server/js/node_modules/node-gyp/gyp/tools/graphviz.py
mlflow/server/js/node_modules/node-gyp/gyp/tools/pretty_gyp.py
mlflow/server/js/node_modules/node-gyp/gyp/tools/pretty_sln.py
mlflow/server/js/node_modules/node-gyp/gyp/tools/pretty_vcproj.py
mlflow/server/js/node_modules/node-gyp/test/fixtures/test-charmap.py
mlflow/server/js/node_modules/railroad-diagrams/railroad_diagrams.py
mlflow/shap/__init__.py
mlflow/sklearn/__init__.py
mlflow/sklearn/utils.py
mlflow/spacy/__init__.py
mlflow/spark/__init__.py
mlflow/spark/autologging.py
mlflow/statsmodels/__init__.py
mlflow/store/__init__.py
mlflow/store/_unity_catalog/__init__.py
mlflow/store/_unity_catalog/registry/__init__.py
mlflow/store/_unity_catalog/registry/rest_store.py
mlflow/store/artifact/__init__.py
mlflow/store/artifact/artifact_repo.py
mlflow/store/artifact/artifact_repository_registry.py
mlflow/store/artifact/azure_blob_artifact_repo.py
mlflow/store/artifact/azure_data_lake_artifact_repo.py
mlflow/store/artifact/cli.py
mlflow/store/artifact/cloud_artifact_repo.py
mlflow/store/artifact/databricks_artifact_repo.py
mlflow/store/artifact/databricks_models_artifact_repo.py
mlflow/store/artifact/dbfs_artifact_repo.py
mlflow/store/artifact/ftp_artifact_repo.py
mlflow/store/artifact/gcs_artifact_repo.py
mlflow/store/artifact/hdfs_artifact_repo.py
mlflow/store/artifact/http_artifact_repo.py
mlflow/store/artifact/local_artifact_repo.py
mlflow/store/artifact/mlflow_artifacts_repo.py
mlflow/store/artifact/models_artifact_repo.py
mlflow/store/artifact/optimized_s3_artifact_repo.py
mlflow/store/artifact/presigned_url_artifact_repo.py
mlflow/store/artifact/r2_artifact_repo.py
mlflow/store/artifact/runs_artifact_repo.py
mlflow/store/artifact/s3_artifact_repo.py
mlflow/store/artifact/sftp_artifact_repo.py
mlflow/store/artifact/unity_catalog_models_artifact_repo.py
mlflow/store/artifact/utils/__init__.py
mlflow/store/artifact/utils/models.py
mlflow/store/db/__init__.py
mlflow/store/db/base_sql_model.py
mlflow/store/db/db_types.py
mlflow/store/db/utils.py
mlflow/store/db_migrations/__init__.py
mlflow/store/db_migrations/alembic.ini
mlflow/store/db_migrations/env.py
mlflow/store/db_migrations/versions/0a8213491aaa_drop_duplicate_killed_constraint.py
mlflow/store/db_migrations/versions/0c779009ac13_add_deleted_time_field_to_runs_table.py
mlflow/store/db_migrations/versions/181f10493468_allow_nulls_for_metric_values.py
mlflow/store/db_migrations/versions/27a6a02d2cf1_add_model_version_tags_table.py
mlflow/store/db_migrations/versions/2b4d017a5e9b_add_model_registry_tables_to_db.py
mlflow/store/db_migrations/versions/2d6e25af4d3e_increase_max_param_val_length.py
mlflow/store/db_migrations/versions/3500859a5d39_add_model_aliases_table.py
mlflow/store/db_migrations/versions/39d1c3be5f05_add_is_nan_constraint_for_metrics_tables_if_necessary.py
mlflow/store/db_migrations/versions/451aebb31d03_add_metric_step.py
mlflow/store/db_migrations/versions/728d730b5ebd_add_registered_model_tags_table.py
mlflow/store/db_migrations/versions/7ac759974ad8_update_run_tags_with_larger_limit.py
mlflow/store/db_migrations/versions/7f2a7d5fae7d_add_datasets_inputs_input_tags_tables.py
mlflow/store/db_migrations/versions/84291f40a231_add_run_link_to_model_version.py
mlflow/store/db_migrations/versions/89d4b8295536_create_latest_metrics_table.py
mlflow/store/db_migrations/versions/90e64c465722_migrate_user_column_to_tags.py
mlflow/store/db_migrations/versions/97727af70f4d_creation_time_last_update_time_experiments.py
mlflow/store/db_migrations/versions/__init__.py
mlflow/store/db_migrations/versions/a8c4a736bde6_allow_nulls_for_run_id.py
mlflow/store/db_migrations/versions/acf3f17fdcc7_add_storage_location_field_to_model_.py
mlflow/store/db_migrations/versions/bd07f7e963c5_create_index_on_run_uuid.py
mlflow/store/db_migrations/versions/c48cb773bb87_reset_default_value_for_is_nan_in_metrics_table_for_mysql.py
mlflow/store/db_migrations/versions/cc1f77228345_change_param_value_length_to_500.py
mlflow/store/db_migrations/versions/cfd24bdc0731_update_run_status_constraint_with_killed.py
mlflow/store/db_migrations/versions/df50e92ffc5e_add_experiment_tags_table.py
mlflow/store/entities/__init__.py
mlflow/store/entities/paged_list.py
mlflow/store/model_registry/__init__.py
mlflow/store/model_registry/abstract_store.py
mlflow/store/model_registry/base_rest_store.py
mlflow/store/model_registry/databricks_workspace_model_registry_rest_store.py
mlflow/store/model_registry/file_store.py
mlflow/store/model_registry/rest_store.py
mlflow/store/model_registry/sqlalchemy_store.py
mlflow/store/model_registry/dbmodels/__init__.py
mlflow/store/model_registry/dbmodels/models.py
mlflow/store/tracking/__init__.py
mlflow/store/tracking/abstract_store.py
mlflow/store/tracking/file_store.py
mlflow/store/tracking/rest_store.py
mlflow/store/tracking/sqlalchemy_store.py
mlflow/store/tracking/dbmodels/__init__.py
mlflow/store/tracking/dbmodels/initial_models.py
mlflow/store/tracking/dbmodels/models.py
mlflow/system_metrics/__init__.py
mlflow/system_metrics/system_metrics_monitor.py
mlflow/system_metrics/metrics/__init__.py
mlflow/system_metrics/metrics/base_metrics_monitor.py
mlflow/system_metrics/metrics/cpu_monitor.py
mlflow/system_metrics/metrics/disk_monitor.py
mlflow/system_metrics/metrics/gpu_monitor.py
mlflow/system_metrics/metrics/network_monitor.py
mlflow/tensorflow/__init__.py
mlflow/tensorflow/autologging.py
mlflow/tensorflow/callback.py
mlflow/tracking/__init__.py
mlflow/tracking/artifact_utils.py
mlflow/tracking/client.py
mlflow/tracking/fluent.py
mlflow/tracking/metric_value_conversion_utils.py
mlflow/tracking/multimedia.py
mlflow/tracking/registry.py
mlflow/tracking/_model_registry/__init__.py
mlflow/tracking/_model_registry/client.py
mlflow/tracking/_model_registry/fluent.py
mlflow/tracking/_model_registry/registry.py
mlflow/tracking/_model_registry/utils.py
mlflow/tracking/_tracking_service/__init__.py
mlflow/tracking/_tracking_service/client.py
mlflow/tracking/_tracking_service/registry.py
mlflow/tracking/_tracking_service/utils.py
mlflow/tracking/context/__init__.py
mlflow/tracking/context/abstract_context.py
mlflow/tracking/context/databricks_cluster_context.py
mlflow/tracking/context/databricks_command_context.py
mlflow/tracking/context/databricks_job_context.py
mlflow/tracking/context/databricks_notebook_context.py
mlflow/tracking/context/databricks_repo_context.py
mlflow/tracking/context/default_context.py
mlflow/tracking/context/git_context.py
mlflow/tracking/context/registry.py
mlflow/tracking/context/system_environment_context.py
mlflow/tracking/default_experiment/__init__.py
mlflow/tracking/default_experiment/abstract_context.py
mlflow/tracking/default_experiment/databricks_notebook_experiment_provider.py
mlflow/tracking/default_experiment/registry.py
mlflow/tracking/request_auth/__init__.py
mlflow/tracking/request_auth/abstract_request_auth_provider.py
mlflow/tracking/request_auth/registry.py
mlflow/tracking/request_header/__init__.py
mlflow/tracking/request_header/abstract_request_header_provider.py
mlflow/tracking/request_header/databricks_request_header_provider.py
mlflow/tracking/request_header/default_request_header_provider.py
mlflow/tracking/request_header/registry.py
mlflow/transformers/__init__.py
mlflow/transformers/flavor_config.py
mlflow/transformers/hub_utils.py
mlflow/transformers/llm_inference_utils.py
mlflow/transformers/model_io.py
mlflow/transformers/peft.py
mlflow/transformers/signature.py
mlflow/transformers/torch_utils.py
mlflow/types/__init__.py
mlflow/types/llm.py
mlflow/types/schema.py
mlflow/types/utils.py
mlflow/utils/__init__.py
mlflow/utils/_capture_modules.py
mlflow/utils/_capture_transformers_modules.py
mlflow/utils/_spark_utils.py
mlflow/utils/_unity_catalog_utils.py
mlflow/utils/annotations.py
mlflow/utils/arguments_utils.py
mlflow/utils/checkpoint_utils.py
mlflow/utils/class_utils.py
mlflow/utils/cli_args.py
mlflow/utils/conda.py
mlflow/utils/credentials.py
mlflow/utils/data_utils.py
mlflow/utils/databricks_utils.py
mlflow/utils/docstring_utils.py
mlflow/utils/doctor.py
mlflow/utils/download_cloud_file_chunk.py
mlflow/utils/env_manager.py
mlflow/utils/environment.py
mlflow/utils/exception_utils.py
mlflow/utils/file_utils.py
mlflow/utils/git_utils.py
mlflow/utils/gorilla.py
mlflow/utils/insecure_hash.py
mlflow/utils/lazy_load.py
mlflow/utils/logging_utils.py
mlflow/utils/mime_type_utils.py
mlflow/utils/mlflow_tags.py
mlflow/utils/model_utils.py
mlflow/utils/name_utils.py
mlflow/utils/nfs_on_spark.py
mlflow/utils/openai_utils.py
mlflow/utils/os.py
mlflow/utils/process.py
mlflow/utils/promptlab_utils.py
mlflow/utils/proto_json_utils.py
mlflow/utils/request_utils.py
mlflow/utils/requirements_utils.py
mlflow/utils/rest_utils.py
mlflow/utils/search_utils.py
mlflow/utils/server_cli_utils.py
mlflow/utils/string_utils.py
mlflow/utils/time.py
mlflow/utils/timeout.py
mlflow/utils/uri.py
mlflow/utils/validation.py
mlflow/utils/virtualenv.py
mlflow/utils/async_logging/__init__.py
mlflow/utils/async_logging/async_logging_queue.py
mlflow/utils/async_logging/run_batch.py
mlflow/utils/async_logging/run_operations.py
mlflow/utils/autologging_utils/__init__.py
mlflow/utils/autologging_utils/client.py
mlflow/utils/autologging_utils/events.py
mlflow/utils/autologging_utils/logging_and_warnings.py
mlflow/utils/autologging_utils/metrics_queue.py
mlflow/utils/autologging_utils/safety.py
mlflow/utils/autologging_utils/versioning.py
mlflow/utils/import_hooks/__init__.py
mlflow/xgboost/__init__.py
mlflow/xgboost/_autolog.py
tests/test_cli.py
tests/test_environment_variables.py
tests/test_exceptions.py
tests/test_flavors.py
tests/test_import.py
tests/test_runs.py
tests/test_skinny_client_autolog_without_scipy.py
tests/test_skinny_client_omits_data_science_libs.py
tests/test_skinny_client_omits_sql_libs.py
tests/test_version.py