backoff>=2.2.1
databricks-sdk<=0.44.1,>=0.29.0
datasets<=3.3.2,>=2.16.1
gql[websockets]>=3.4.0
mlflow<=2.20.2,>=2.14.2
mosaicml-cli>=0.6
requests<3,>=2.26.0
boto3<2,>=1.36
typing_extensions>=4.7.1
pydantic>=2
databricks-agents==0.16.0
tenacity>=9.0.0
rich>=12.6.0
openai<2,>=1.56
rouge-score>=0.1.2
evaluate>=0.4.3
catalogue<3,>=2.0.10
IPython<9,>=8
freezegun<2,>=1.5

[all]
twine>=4.0.2
build>=0.10.0
pyright==1.1.256
parameterized==0.9
packaging<25,>=21
pytest>=7.4.0
yapf>=0.40.0
isort>=5.9.3
databricks-connect==16.0.0
ipywidgets<9,>=8
pre-commit>=2.17.0
pylint>=3.0.0

[dev]
isort>=5.9.3
packaging<25,>=21
parameterized==0.9
pre-commit>=2.17.0
pylint>=3.0.0
pyright==1.1.256
pytest>=7.4.0
yapf>=0.40.0

[notebook]
ipywidgets<9,>=8

[release]
build>=0.10.0
twine>=4.0.2

[spark]
databricks-connect==16.0.0
