mlflow--mlflow
wehub-resource-sync
97 行
2.8 KiB
Plaintext
97 行
2.8 KiB
Plaintext
## This file describes extra ML library dependencies that you, as an end user,
|
|
## must install in order to use various MLflow Python modules.
|
|
# Required by mlflow.spacy
|
|
spacy>=3.3.0
|
|
# Required by mlflow.tensorflow
|
|
tensorflow>=2.10.0; platform_system!="Darwin" or platform_machine!="arm64"
|
|
tensorflow-macos>=2.10.0; platform_system=="Darwin" and platform_machine=="arm64"
|
|
tensorboard
|
|
# Required by mlflow.pytorch
|
|
torch>=1.11.0
|
|
torchvision>=0.12.0
|
|
lightning>=1.8.1
|
|
# Required by mlflow.xgboost
|
|
xgboost>=0.82
|
|
# Required by mlflow.lightgbm
|
|
lightgbm
|
|
# Required by mlflow.catboost
|
|
catboost
|
|
# Required by mlflow.statsmodels
|
|
statsmodels
|
|
# Required by mlflow.h2o
|
|
h2o
|
|
# Required by mlflow.onnx
|
|
onnx>=1.17.0
|
|
onnxruntime
|
|
onnxscript
|
|
tf2onnx
|
|
# Required by mlflow.spark and using Delta with MLflow Tracking datasets
|
|
pyspark<4.1.0
|
|
# Required by mlflow.paddle
|
|
paddlepaddle
|
|
# Required by mlflow.prophet
|
|
# NOTE: Prophet's whl build process will fail with dependencies not being present.
|
|
# Installation will default to setup.py in order to install correctly.
|
|
# To install in dev environment, ensure that gcc>=8 is installed to allow pystan
|
|
# to compile the model binaries. See: https://gcc.gnu.org/install/
|
|
# Avoid 0.25 due to https://github.com/dr-prodigy/python-holidays/issues/1200
|
|
holidays!=0.25
|
|
prophet
|
|
# Required by mlflow.shap
|
|
# and shap evaluation functionality
|
|
shap>=0.42.1
|
|
# Required by mlflow.pmdarima
|
|
pmdarima
|
|
# Required for using Hugging Face datasets with MLflow Tracking
|
|
# Avoid datasets < 2.19.1 due to an incompatibility issue https://github.com/huggingface/datasets/issues/6737
|
|
datasets>=2.19.1
|
|
# Required by mlflow.transformers
|
|
transformers
|
|
sentencepiece
|
|
setfit
|
|
librosa
|
|
ffmpeg
|
|
accelerate
|
|
# Required by mlflow.openai
|
|
openai
|
|
tiktoken
|
|
tenacity
|
|
# Required by mlflow.llama_index
|
|
# Floor matches the supported minimum in mlflow/ml-package-versions.yml. Without it, a
|
|
# transitive conflict in the combined ML env silently backtracks llama-index to an
|
|
# unsupported 0.10.x (pydantic v1 models), which crashes Settings serialization.
|
|
llama-index>=0.12.38
|
|
# Required for an agent example of mlflow.llama_index
|
|
llama-index-agent-openai
|
|
# Required by mlflow.langchain
|
|
langchain
|
|
langchain-openai
|
|
# Required by mlflow.sentence_transformers
|
|
sentence-transformers
|
|
# Required by mlflow.anthropic
|
|
anthropic
|
|
# Required by mlflow.ag2
|
|
ag2
|
|
# Required by mlflow.dspy
|
|
# In dspy 2.6.9, `dspy.__name__` is not 'dspy', but 'dspy.__metadata__',
|
|
# which causes auto-logging tests to fail.
|
|
dspy!=2.6.9
|
|
# Required by mlflow.litellm
|
|
litellm
|
|
# Required by mlflow.gemini
|
|
google-genai
|
|
# Required by mlflow.groq
|
|
groq
|
|
# Required by mlflow.mistral
|
|
mistralai
|
|
# Required by mlflow.autogen
|
|
autogen-agentchat
|
|
# Required by mlflow.semantic_kernel
|
|
semantic-kernel
|
|
# Required by mlflow.agno
|
|
agno
|
|
# Required by mlflow.strands
|
|
strands-agents
|
|
# Required by mlflow.haystack
|
|
haystack-ai
|