attrs<26,>=24
boto3<2.0,>=1.42.2
cloudpickle>=2.2.1
docker
fastapi
google-pasta
importlib-metadata<7.0,>=1.4.0
jsonschema
numpy<3.0,>=1.26.4
omegaconf<3,>=2.2
packaging<25,>=23.0
pandas>=2.3.0
pathos
platformdirs
protobuf<6.32,>=3.12
psutil
PyYAML>=6.0.1
requests
sagemaker-core<2.0.0,>=1.0.71
schema
smdebug_rulesconfig==1.0.1
tblib<4,>=1.7.0
tqdm
urllib3<3.0.0,>=1.26.8
uvicorn
graphene<4,>=3

[all]
pyspark==3.3.2
sagemaker-feature-store-pyspark-3.3
accelerate<=0.27.0,>=0.24.1
sagemaker_schema_inference_artifacts>=0.0.5
uvicorn>=0.30.1
fastapi>=0.111.0
nest-asyncio
urllib3<3.0.0,>=1.26.8
docker<8.0.0,>=5.0.2
PyYAML<7,>=6.0.1
scipy==1.13.0
sagemaker-mlflow>=0.1.0

[feature-processor]
pyspark==3.3.2
sagemaker-feature-store-pyspark-3.3

[huggingface]
accelerate<=0.27.0,>=0.24.1
sagemaker_schema_inference_artifacts>=0.0.5
uvicorn>=0.30.1
fastapi>=0.111.0
nest-asyncio

[local]
urllib3<3.0.0,>=1.26.8
docker<8.0.0,>=5.0.2
PyYAML<7,>=6.0.1

[sagemaker-mlflow]
sagemaker-mlflow>=0.1.0

[scipy]
scipy==1.13.0

[test]
tox==3.24.5
numpy<3.0,>=2.0.0
build[virtualenv]==1.2.1
flake8==7.1.2
pytest==6.2.5
pytest-cov==3.0.0
pytest-rerunfailures==10.2
pytest-timeout==2.1.0
pytest-xdist==2.4.0
coverage<6.2,>=5.2
mock==4.0.3
contextlib2==21.6.0
awslogs==0.14.0
black==24.3.0
stopit==1.1.2
apache-airflow==2.10.4
apache-airflow-providers-amazon==7.2.1
Flask-Limiter==3.11
attrs<26,>=24
fabric==3.2.2
requests==2.32.2
sagemaker-experiments==0.1.35
Jinja2==3.1.6
pyvis==0.2.1
pandas>=2.3.0
scikit-learn==1.6.1
cloudpickle==2.2.1
jsonpickle<4.0.0
PyYAML>=6.0.1
xgboost<=1.7.6,>=1.6.2
pillow<=11,>=10.0.1
opentelemetry-proto==1.27.0
opentelemetry_exporter_otlp==1.27.0
protobuf==4.25.8
tensorboard<=2.18.0,>=2.16.2
transformers==4.48.0
sentencepiece==0.1.99
tritonclient[http]<2.37.0
onnx==1.17.0
nbformat<6,>=5.9
accelerate<=0.27.0,>=0.24.1
schema==0.7.5
tensorflow<=2.19.0,>=2.16.2
mlflow<3,>=2.14.2
huggingface_hub==0.26.2
uvicorn>=0.30.1
fastapi==0.115.4
nest-asyncio
sagemaker-mlflow>=0.1.0
deepdiff>=8.0.0
orderly-set<5.4.0
lexicon
networkx==3.2.1
mypy-boto3-appflow==1.35.39
mypy-boto3-rds==1.35.72
mypy-boto3-redshift-data==1.35.51
mypy-boto3-s3==1.35.76
mypy-extensions==1.0.0
mypy==1.9.0
pyspark==3.3.2
sagemaker-feature-store-pyspark-3.3
sagemaker_schema_inference_artifacts>=0.0.5
fastapi>=0.111.0
urllib3<3.0.0,>=1.26.8
docker<8.0.0,>=5.0.2
PyYAML<7,>=6.0.1
scipy==1.13.0

[test:python_version < "3.10"]
google-re2<1.1.20250805
