sagemaker-core>=2.10.1
graphene<4,>=3
typing_extensions>=4.9.0
tblib>=1.7.0
PyYAML<7.0,>=6.0
paramiko>=2.11.0
jinja2<4.0,>=3.0
sagemaker-mlflow<1.0.0,>=0.0.1
mlflow<4.0.0,>=3.0.0
nest_asyncio>=1.5.0

[notebook]
ipywidgets>=8.0.0
rich>=13.0.0
matplotlib>=3.5.0

[test]
pytest
pytest-cov
pytest-xdist
mock
pydantic
pandas
scipy
omegaconf
graphene
IPython
