torch~=1.9
pytorch-lightning~=1.5
shapely~=1.8
matplotlib~=3.2
wandb~=0.12

[all]
mkdocs-material~=7.3
mkdocstrings~=0.16
setuptools==59.5.0
pytest~=6.2
isort~=5.10
yapf~=0.31
pylint~=2.11

[dev]
mkdocs-material~=7.3
mkdocstrings~=0.16
setuptools==59.5.0
pytest~=6.2
isort~=5.10
yapf~=0.31
pylint~=2.11

[docs]
mkdocs-material~=7.3
mkdocstrings~=0.16
setuptools==59.5.0

[lint]
isort~=5.10
yapf~=0.31
pylint~=2.11

[tests]
pytest~=6.2
