442 lines
9.9 KiB
Python
442 lines
9.9 KiB
Python
# This file was auto-generated by update_ml_package_versions.py.
|
|
# Please do not edit it manually.
|
|
|
|
_ML_PACKAGE_VERSIONS = {
|
|
"sklearn": {
|
|
"package_info": {
|
|
"pip_release": "scikit-learn"
|
|
},
|
|
"models": {
|
|
"minimum": "0.24.1",
|
|
"maximum": "1.6.1"
|
|
},
|
|
"autologging": {
|
|
"minimum": "0.24.1",
|
|
"maximum": "1.6.1"
|
|
}
|
|
},
|
|
"pytorch": {
|
|
"package_info": {
|
|
"pip_release": "torch",
|
|
"module_name": "torch"
|
|
},
|
|
"models": {
|
|
"minimum": "1.9.0",
|
|
"maximum": "2.6.0"
|
|
},
|
|
"autologging": {
|
|
"minimum": "1.9.0",
|
|
"maximum": "2.6.0"
|
|
}
|
|
},
|
|
"pytorch-lightning": {
|
|
"package_info": {
|
|
"pip_release": "pytorch-lightning",
|
|
"module_name": "lightning"
|
|
},
|
|
"autologging": {
|
|
"minimum": "1.9.0",
|
|
"maximum": "2.5.1"
|
|
}
|
|
},
|
|
"keras": {
|
|
"package_info": {
|
|
"pip_release": "keras"
|
|
},
|
|
"models": {
|
|
"minimum": "3.0.2",
|
|
"maximum": "3.9.2"
|
|
},
|
|
"autologging": {
|
|
"minimum": "3.0.2",
|
|
"maximum": "3.9.2"
|
|
}
|
|
},
|
|
"tensorflow": {
|
|
"package_info": {
|
|
"pip_release": "tensorflow"
|
|
},
|
|
"models": {
|
|
"minimum": "2.7.4",
|
|
"maximum": "2.19.0"
|
|
},
|
|
"autologging": {
|
|
"minimum": "2.7.4",
|
|
"maximum": "2.19.0"
|
|
}
|
|
},
|
|
"xgboost": {
|
|
"package_info": {
|
|
"pip_release": "xgboost"
|
|
},
|
|
"models": {
|
|
"minimum": "1.4.2",
|
|
"maximum": "3.0.0"
|
|
},
|
|
"autologging": {
|
|
"minimum": "1.4.2",
|
|
"maximum": "3.0.0"
|
|
}
|
|
},
|
|
"lightgbm": {
|
|
"package_info": {
|
|
"pip_release": "lightgbm"
|
|
},
|
|
"models": {
|
|
"minimum": "3.3.5",
|
|
"maximum": "4.6.0"
|
|
},
|
|
"autologging": {
|
|
"minimum": "3.3.5",
|
|
"maximum": "4.6.0"
|
|
}
|
|
},
|
|
"catboost": {
|
|
"package_info": {
|
|
"pip_release": "catboost"
|
|
},
|
|
"models": {
|
|
"minimum": "1.0.0",
|
|
"maximum": "1.2.8"
|
|
}
|
|
},
|
|
"fastai": {
|
|
"package_info": {
|
|
"pip_release": "fastai"
|
|
},
|
|
"models": {
|
|
"minimum": "2.4.1",
|
|
"maximum": "2.7.19"
|
|
},
|
|
"autologging": {
|
|
"minimum": "2.4.1",
|
|
"maximum": "2.7.19"
|
|
}
|
|
},
|
|
"onnx": {
|
|
"package_info": {
|
|
"pip_release": "onnx"
|
|
},
|
|
"models": {
|
|
"minimum": "1.13.1",
|
|
"maximum": "1.17.0"
|
|
}
|
|
},
|
|
"spacy": {
|
|
"package_info": {
|
|
"pip_release": "spacy"
|
|
},
|
|
"models": {
|
|
"minimum": "3.4.4",
|
|
"maximum": "3.8.4"
|
|
}
|
|
},
|
|
"statsmodels": {
|
|
"package_info": {
|
|
"pip_release": "statsmodels"
|
|
},
|
|
"models": {
|
|
"minimum": "0.12.2",
|
|
"maximum": "0.14.4"
|
|
},
|
|
"autologging": {
|
|
"minimum": "0.12.2",
|
|
"maximum": "0.14.4"
|
|
}
|
|
},
|
|
"spark": {
|
|
"package_info": {
|
|
"pip_release": "pyspark",
|
|
"module_name": "pyspark"
|
|
},
|
|
"models": {
|
|
"minimum": "3.1.2",
|
|
"maximum": "3.5.5"
|
|
},
|
|
"autologging": {
|
|
"minimum": "3.1.2",
|
|
"maximum": "3.5.5"
|
|
}
|
|
},
|
|
"mleap": {
|
|
"package_info": {
|
|
"pip_release": "mleap"
|
|
},
|
|
"models": {
|
|
"minimum": "0.18.0",
|
|
"maximum": "0.23.3"
|
|
}
|
|
},
|
|
"prophet": {
|
|
"package_info": {
|
|
"pip_release": "prophet"
|
|
},
|
|
"models": {
|
|
"minimum": "1.1.1",
|
|
"maximum": "1.1.6"
|
|
}
|
|
},
|
|
"pmdarima": {
|
|
"package_info": {
|
|
"pip_release": "pmdarima"
|
|
},
|
|
"models": {
|
|
"minimum": "1.8.1",
|
|
"maximum": "2.0.4"
|
|
}
|
|
},
|
|
"diviner": {
|
|
"package_info": {
|
|
"pip_release": "diviner"
|
|
},
|
|
"models": {
|
|
"minimum": "0.1.0",
|
|
"maximum": "0.1.1"
|
|
}
|
|
},
|
|
"h2o": {
|
|
"package_info": {
|
|
"pip_release": "h2o"
|
|
},
|
|
"models": {
|
|
"minimum": "3.40.0.1",
|
|
"maximum": "3.46.0.7"
|
|
}
|
|
},
|
|
"shap": {
|
|
"package_info": {
|
|
"pip_release": "shap"
|
|
},
|
|
"models": {
|
|
"minimum": "0.42.1",
|
|
"maximum": "0.47.1"
|
|
}
|
|
},
|
|
"paddle": {
|
|
"package_info": {
|
|
"pip_release": "paddlepaddle"
|
|
},
|
|
"models": {
|
|
"minimum": "2.5.2",
|
|
"maximum": "3.0.0"
|
|
},
|
|
"autologging": {
|
|
"minimum": "2.5.2",
|
|
"maximum": "3.0.0"
|
|
}
|
|
},
|
|
"transformers": {
|
|
"package_info": {
|
|
"pip_release": "transformers"
|
|
},
|
|
"models": {
|
|
"minimum": "4.35.2",
|
|
"maximum": "4.51.2"
|
|
},
|
|
"autologging": {
|
|
"minimum": "4.35.2",
|
|
"maximum": "4.51.2"
|
|
}
|
|
},
|
|
"openai": {
|
|
"package_info": {
|
|
"pip_release": "openai"
|
|
},
|
|
"models": {
|
|
"minimum": "1.0.1",
|
|
"maximum": "1.73.0"
|
|
},
|
|
"autologging": {
|
|
"minimum": "1.17.0",
|
|
"maximum": "1.73.0"
|
|
}
|
|
},
|
|
"dspy": {
|
|
"package_info": {
|
|
"pip_release": "dspy"
|
|
},
|
|
"models": {
|
|
"minimum": "2.5.17",
|
|
"maximum": "2.6.17"
|
|
},
|
|
"autologging": {
|
|
"minimum": "2.5.17",
|
|
"maximum": "2.6.17"
|
|
}
|
|
},
|
|
"langchain": {
|
|
"package_info": {
|
|
"pip_release": "langchain"
|
|
},
|
|
"models": {
|
|
"minimum": "0.0.354",
|
|
"maximum": "0.3.23"
|
|
},
|
|
"autologging": {
|
|
"minimum": "0.1.0",
|
|
"maximum": "0.3.23"
|
|
}
|
|
},
|
|
"langgraph": {
|
|
"package_info": {
|
|
"pip_release": "langgraph"
|
|
},
|
|
"models": {
|
|
"minimum": "0.2.0",
|
|
"maximum": "0.3.29"
|
|
},
|
|
"autologging": {
|
|
"minimum": "0.2.0",
|
|
"maximum": "0.3.29"
|
|
}
|
|
},
|
|
"llama_index": {
|
|
"package_info": {
|
|
"pip_release": "llama-index",
|
|
"module_name": "llama_index.core"
|
|
},
|
|
"models": {
|
|
"minimum": "0.10.44",
|
|
"maximum": "0.12.30"
|
|
},
|
|
"autologging": {
|
|
"minimum": "0.10.44",
|
|
"maximum": "0.12.30"
|
|
}
|
|
},
|
|
"autogen": {
|
|
"package_info": {
|
|
"pip_release": "autogen-agentchat"
|
|
},
|
|
"autologging": {
|
|
"minimum": "0.2.36",
|
|
"maximum": "0.2.40"
|
|
}
|
|
},
|
|
"gemini": {
|
|
"package_info": {
|
|
"pip_release": "google-genai",
|
|
"module_name": "google.genai"
|
|
},
|
|
"autologging": {
|
|
"minimum": "1.0.0",
|
|
"maximum": "1.10.0"
|
|
}
|
|
},
|
|
"anthropic": {
|
|
"package_info": {
|
|
"pip_release": "anthropic"
|
|
},
|
|
"autologging": {
|
|
"minimum": "0.30.0",
|
|
"maximum": "0.49.0"
|
|
}
|
|
},
|
|
"crewai": {
|
|
"package_info": {
|
|
"pip_release": "crewai",
|
|
"module_name": "crewai"
|
|
},
|
|
"autologging": {
|
|
"minimum": "0.80.0",
|
|
"maximum": "0.114.0"
|
|
}
|
|
},
|
|
"mistral": {
|
|
"package_info": {
|
|
"pip_release": "mistralai",
|
|
"module_name": "mistralai"
|
|
},
|
|
"autologging": {
|
|
"minimum": "1.0.0",
|
|
"maximum": "1.6.0"
|
|
}
|
|
},
|
|
"sentence_transformers": {
|
|
"package_info": {
|
|
"pip_release": "sentence-transformers"
|
|
},
|
|
"models": {
|
|
"minimum": "2.2.2",
|
|
"maximum": "4.0.2"
|
|
}
|
|
},
|
|
"johnsnowlabs": {
|
|
"package_info": {
|
|
"pip_release": "johnsnowlabs"
|
|
},
|
|
"models": {
|
|
"minimum": "4.4.6",
|
|
"maximum": "5.5.5"
|
|
}
|
|
},
|
|
"promptflow": {
|
|
"package_info": {
|
|
"pip_release": "promptflow"
|
|
},
|
|
"models": {
|
|
"minimum": "1.3.0",
|
|
"maximum": "1.17.2"
|
|
}
|
|
},
|
|
"litellm": {
|
|
"package_info": {
|
|
"pip_release": "litellm"
|
|
},
|
|
"autologging": {
|
|
"minimum": "1.52.9",
|
|
"maximum": "1.66.0"
|
|
}
|
|
},
|
|
"groq": {
|
|
"package_info": {
|
|
"pip_release": "groq"
|
|
},
|
|
"autologging": {
|
|
"minimum": "0.13.0",
|
|
"maximum": "0.22.0"
|
|
}
|
|
},
|
|
"bedrock": {
|
|
"package_info": {
|
|
"pip_release": "boto3",
|
|
"module_name": "boto3"
|
|
},
|
|
"autologging": {
|
|
"minimum": "1.33.0",
|
|
"maximum": "1.37.33"
|
|
}
|
|
}
|
|
}
|
|
|
|
# A mapping of flavor name to the module name to be imported for autologging.
|
|
# This is used for checking version compatibility in autologging.
|
|
# DO NOT EDIT MANUALLY
|
|
FLAVOR_TO_MODULE_NAME = {
|
|
"sklearn": "sklearn",
|
|
"pytorch": "torch",
|
|
"pytorch-lightning": "lightning",
|
|
"keras": "keras",
|
|
"tensorflow": "tensorflow",
|
|
"xgboost": "xgboost",
|
|
"lightgbm": "lightgbm",
|
|
"fastai": "fastai",
|
|
"statsmodels": "statsmodels",
|
|
"spark": "pyspark",
|
|
"paddle": "paddle",
|
|
"transformers": "transformers",
|
|
"openai": "openai",
|
|
"dspy": "dspy",
|
|
"langchain": "langchain",
|
|
"llama_index": "llama_index.core",
|
|
"autogen": "autogen",
|
|
"gemini": "google.genai",
|
|
"anthropic": "anthropic",
|
|
"crewai": "crewai",
|
|
"mistral": "mistralai",
|
|
"litellm": "litellm",
|
|
"groq": "groq",
|
|
"bedrock": "boto3",
|
|
"pyspark.ml": "pyspark"
|
|
}
|