pydantic<3.0.0,>=2.0.0
pyyaml>=6.0

[dev]
ruff
pytest>=7.0.0
pytest-cov>=4.0.0
mypy>=1.8.0
types-protobuf
types-grpcio
types-PyYAML
interrogate>=1.7.0

[full]
cuvis-ai-schemas[lightning,numpy,proto,torch]

[lightning]
pytorch-lightning>=2.0.0

[numpy]
numpy>=1.21.0

[proto]
grpcio>=1.56.0
protobuf>=4.25.0
grpcio-tools>=1.56.0

[torch]
torch>=2.0.0
torchvision
