torch>=1.9.0
numpy>=1.19.0
tqdm>=4.60.0

[all]
notmiwae-pytorch[dev,notebooks]

[dev]
pytest>=6.0
pytest-cov>=2.0
black>=22.0
isort>=5.0
flake8>=4.0

[notebooks]
jupyter>=1.0
matplotlib>=3.3
pandas>=1.2
scikit-learn>=0.24
torchvision>=0.10
