torchgeo/pyproject.toml

[build-system]
requires = [
    # setuptools 61+ required for pyproject.toml support
    "setuptools>=61,<69",
]
build-backend = "setuptools.build_meta"

# https://packaging.python.org/en/latest/specifications/declaring-project-metadata/
[project]
name = "torchgeo"
description = "TorchGeo: datasets, samplers, transforms, and pre-trained models for geospatial data"
readme = "README.md"
requires-python = ">=3.9,<4"
license = {file = "LICENSE"}
authors = [
    {name = "Adam J. Stewart", email = "ajstewart426@gmail.com"},
]
maintainers = [
    {name = "Adam J. Stewart", email = "ajstewart426@gmail.com"},
    {name = "Caleb Robinson", email = "Caleb.Robinson@microsoft.com"},
    {name = "Isaac Corley", email = "isaac.corley@my.utsa.edu"},
    {name = "Nils Lehmann", email = "n.lehmann@tum.de"},
    {name = "Ashwin Nair", email = "ashnair0007@gmail.com"},
]
keywords = ["pytorch", "deep learning", "machine learning", "remote sensing", "satellite imagery", "earth observation", "geospatial"]
classifiers = [
    "Development Status :: 3 - Alpha",
    "Intended Audience :: Science/Research",
    "License :: OSI Approved :: MIT License",
    "Operating System :: OS Independent",
    "Programming Language :: Python :: 3",
    "Programming Language :: Python :: 3.9",
    "Programming Language :: Python :: 3.10",
    "Programming Language :: Python :: 3.11",
    "Topic :: Scientific/Engineering :: Artificial Intelligence",
    "Topic :: Scientific/Engineering :: GIS",
]
dependencies = [
    # einops 0.3+ required for einops.repeat
    "einops>=0.3,<0.7",
    # fiona 1.8.19+ required to fix erroneous warning
    # https://github.com/Toblerity/Fiona/issues/986
    "fiona>=1.8.19,<2",
    # kornia 0.6.9+ required for kornia.augmentation.RandomBrightness
    "kornia>=0.6.9,<0.7",
    # lightly 1.4.4+ required for MoCo v3 support
    "lightly>=1.4.4",
    # lightning 1.8+ is first release
    "lightning>=1.8,<3",
    # matplotlib 3.3.3+ required for Python 3.9 wheels
    "matplotlib>=3.3.3,<4",
    # numpy 1.19.3+ required by Python 3.9 wheels
    "numpy>=1.19.3,<2",
    # pillow 8+ required for Python 3.9 wheels
    "pillow>=8,<11",
    # pyproj 3+ required for Python 3.9 wheels
    "pyproj>=3,<4",
    # rasterio 1.2+ required for Python 3.9 wheels
    "rasterio>=1.2,<2",
    # rtree 1+ required for len(index), index & index, index | index
    "rtree>=1,<2",
    # segmentation-models-pytorch 0.2+ required for smp.losses module
    "segmentation-models-pytorch>=0.2,<0.4",
    # shapely 1.7.1+ required for Python 3.9 wheels
    "shapely>=1.7.1,<3",
    # timm 0.4.12 required by segmentation-models-pytorch
    "timm>=0.4.12,<0.10",
    # torch 1.12+ required by torchvision
    "torch>=1.12,<3",
    # torchmetrics 0.10+ required for binary/multiclass/multilabel classification metrics
    "torchmetrics>=0.10,<2",
    # torchvision 0.13+ required for torchvision.models._api.WeightsEnum
    "torchvision>=0.13,<0.16",
]
dynamic = ["version"]

[project.optional-dependencies]
datasets = [
    # h5py 3+ required for Python 3.9 wheels
    "h5py>=3,<4",
    # laspy 2+ required for laspy.read
    "laspy>=2,<3",
    # opencv-python 4.4.0.46+ required for Python 3.9 wheels
    "opencv-python>=4.4.0.46,<5",
    # pandas 1.1.3+ required for Python 3.9 wheels
    "pandas>=1.1.3,<3",
    # pycocotools 2.0.4+ required to avoid use of deprecated setuptools fetch_build_eggs
    "pycocotools>=2.0.4,<3",
    # pyvista 0.29+ required for to avoid segfault during testing
    "pyvista>=0.29,<0.41",
    # radiant-mlhub 0.3+ required for newer tqdm support required by lightning
    "radiant-mlhub>=0.3,<0.6",
    # rarfile 4+ required for wheels
    "rarfile>=4,<5",
    # scikit-image 0.18+ required for numpy 1.17+ compatibility
    # https://github.com/scikit-image/scikit-image/issues/3655
    "scikit-image>=0.18,<0.22",
    # scipy 1.6.2+ required for scikit-image 0.18+ compatibility
    "scipy>=1.6.2,<2",
    # zipfile-deflate64 0.2+ required for extraction bugfix:
    # https://github.com/brianhelba/zipfile-deflate64/issues/19
    "zipfile-deflate64>=0.2,<0.3",
]
docs = [
    # ipywidgets 7+ required by nbsphinx
    "ipywidgets>=7,<9",
    # nbsphinx 0.8.5 fixes bug with nbformat attributes
    "nbsphinx>=0.8.5,<0.10",
    # release versions missing files, must install from master
    "pytorch-sphinx-theme",
    # sphinx 4+ required for autodoc_typehints_description_target = documented
    # sphinx 6+ is incompatible with pytorch-sphinx-theme
    # https://github.com/pytorch/pytorch_sphinx_theme/issues/175
    "sphinx>=4,<6",
]
style = [
    # black 21.8+ required for Jupyter support
    "black[jupyter]>=21.8,<24",
    # flake8 3.8+ depends on pyflakes 2.2+, which fixes a bug with mypy error code ignores:
    # https://github.com/PyCQA/pyflakes/pull/455
    "flake8>=3.8,<7",
    # isort 5.8+ required for extend_skip option
    "isort[colors]>=5.8,<6",
    # pydocstyle 6.1+ required for pyproject.toml support
    "pydocstyle[toml]>=6.1,<7",
    # pyupgrade 2.8+ required for --py39-plus flag
    "pyupgrade>=2.8,<4",
]
tests = [
    # hydra-core 1+ required for omegaconf 2 support
    "hydra-core>=1,<2",
    # mypy 0.900+ required for pyproject.toml support
    "mypy>=0.900,<2",
    # nbmake 1.3.3+ required for variable mocking
    "nbmake>=1.3.3,<2",
    # omegaconf 2+ required by lightning, 2.0.1+ required by hydra-core
    "omegaconf>=2.0.1,<3",
    # pytest 6.2+ required for pytest.MonkeyPatch
    "pytest>=6.2,<8",
    # pytest-cov 2.4+ required for pytest --cov flags
    "pytest-cov>=2.4,<5",
    # tensorboard 2.9.1+ required by lightning
    "tensorboard>=2.9.1,<3",
]
all = [
    "torchgeo[datasets,docs,style,tests]",
]

[project.urls]
Homepage = "https://github.com/microsoft/torchgeo"
Documentation = "https://torchgeo.readthedocs.io"

[tool.black]
target-version = ["py39", "py310"]
color = true
skip_magic_trailing_comma = true

[tool.coverage.report]
# Ignore warnings for overloads
# https://github.com/nedbat/coveragepy/issues/970#issuecomment-612602180
exclude_lines = [
    "pragma: no cover",
    "@overload",
]

[tool.isort]
profile = "black"
known_first_party = ["docs", "tests", "torchgeo", "train"]
skip_gitignore = true
color_output = true

[tool.mypy]
python_version = "3.9"
ignore_missing_imports = true
show_error_codes = true
exclude = "(build|data|dist|docs/src|images|logo|logs|output)/"

# Strict
warn_unused_configs = true
disallow_any_generics = true
disallow_subclassing_any = true
disallow_untyped_calls = true
disallow_untyped_defs = true
disallow_incomplete_defs = true
check_untyped_defs = true
disallow_untyped_decorators = true
no_implicit_optional = true
warn_redundant_casts = true
warn_unused_ignores = true
warn_return_any = true
no_implicit_reexport = true
strict_equality = true

[tool.pydocstyle]
convention = "google"
match_dir = "(datamodules|datasets|losses|models|samplers|torchgeo|trainers|transforms)"

[tool.pytest.ini_options]
# Skip slow tests by default
addopts = "-m 'not slow'"
# https://docs.pytest.org/en/latest/how-to/capture-warnings.html
filterwarnings = [
    # Treat all warnings as errors
    "error",

    # Warnings raised by dependencies of dependencies, out of our control
    # https://github.com/Cadene/pretrained-models.pytorch/issues/221
    "ignore:.* is deprecated and will be removed in Pillow 10:DeprecationWarning:pretrainedmodels.datasets.utils",
    # https://github.com/pytorch/vision/pull/5898
    "ignore:.* is deprecated and will be removed in Pillow 10:DeprecationWarning:torchvision.transforms.functional_pil",
    "ignore:.* is deprecated and will be removed in Pillow 10:DeprecationWarning:torchvision.transforms._functional_pil",
    # https://github.com/rwightman/pytorch-image-models/pull/1256
    "ignore:.* is deprecated and will be removed in Pillow 10:DeprecationWarning:timm.data",
    # https://github.com/pytorch/pytorch/issues/72906
    # https://github.com/pytorch/pytorch/pull/69823
    "ignore:distutils Version classes are deprecated. Use packaging.version instead:DeprecationWarning",
    "ignore:The distutils package is deprecated and slated for removal in Python 3.12:DeprecationWarning:torch.utils.tensorboard",
    # https://github.com/Lightning-AI/lightning/issues/13256
    # https://github.com/Lightning-AI/lightning/pull/13261
    "ignore:torch.distributed._sharded_tensor will be deprecated:DeprecationWarning:torch.distributed._sharded_tensor",
    # https://github.com/Lightning-AI/lightning/issues/13989
    "ignore:SelectableGroups dict interface is deprecated. Use select.:DeprecationWarning:lightning.pytorch.trainer.connectors.callback_connector",
    # https://github.com/Lightning-AI/lightning/issues/14594
    "ignore:To copy construct from a tensor, it is recommended to use:UserWarning:lightning.pytorch.core.module",
    # https://github.com/rasterio/rasterio/issues/1742
    # https://github.com/rasterio/rasterio/pull/1753
    "ignore:Using or importing the ABCs from 'collections' instead of from 'collections.abc' is deprecated:DeprecationWarning:rasterio.crs",
    # https://github.com/pytorch/pytorch/issues/60053
    # https://github.com/pytorch/pytorch/pull/60059
    "ignore:Named tensors and all their associated APIs are an experimental feature and subject to change:UserWarning:torch.nn.functional",
    # https://github.com/tensorflow/tensorboard/issues/5798
    "ignore:Call to deprecated create function:DeprecationWarning:tensorboard.compat.proto",
    # https://github.com/treebeardtech/nbmake/issues/68
    'ignore:The \(fspath. py.path.local\) argument to NotebookFile is deprecated:pytest.PytestDeprecationWarning:nbmake.pytest_plugin',
    # https://github.com/kornia/kornia/issues/777
    "ignore:Default upsampling behavior when mode=bilinear is changed to align_corners=False since 0.4.0:UserWarning:torch.nn.functional",
    # https://github.com/scikit-image/scikit-image/issues/6663
    # https://github.com/scikit-image/scikit-image/pull/6637
    "ignore:`np.bool8` is a deprecated alias for `np.bool_`.:DeprecationWarning:skimage.util.dtype",
    # https://github.com/lanpa/tensorboardX/pull/677
    "ignore:ANTIALIAS is deprecated and will be removed in Pillow 10:DeprecationWarning:tensorboardX.summary",
    # https://github.com/Lightning-AI/lightning/issues/16756
    "ignore:Deprecated call to `pkg_resources.declare_namespace:DeprecationWarning",

    # Expected warnings
    # Lightning warns us about using num_workers=0, but it's faster on macOS
    "ignore:The dataloader, .*, does not have many workers which may be a bottleneck:UserWarning",
    # Lightning warns us about using the CPU when GPU/MPS is available
    "ignore:GPU available but not used.:UserWarning",
    "ignore:MPS available but not used.:UserWarning",
    # Lightning warns us if TensorBoard is not installed
    "ignore:Starting from v1.9.0, `tensorboardX` has been removed as a dependency of the `lightning.pytorch` package:UserWarning",
    # https://github.com/kornia/kornia/pull/1611
    "ignore:`ColorJitter` is now following Torchvision implementation.:DeprecationWarning:kornia.augmentation._2d.intensity.color_jitter",
    # https://github.com/kornia/kornia/pull/1663
    "ignore:`RandomGaussianBlur` has changed its behavior and now randomly sample sigma for both axes.:DeprecationWarning:kornia.augmentation._2d.intensity.gaussian_blur",

    # Unexpected warnings, worth investigating
    # Lightning is having trouble inferring the batch size for ChesapeakeCVPRDataModule and CycloneDataModule for some reason
    "ignore:Trying to infer the `batch_size` from an ambiguous collection:UserWarning",
]
markers = [
    "slow: marks tests as slow",
]
norecursedirs = [
    ".ipynb_checkpoints",
    "data",
    "__pycache__",
]
testpaths = [
    "tests",
    "docs/tutorials",
]

# https://setuptools.pypa.io/en/latest/userguide/pyproject_config.html
[tool.setuptools.dynamic]
version = {attr = "torchgeo.__version__"}

[tool.setuptools.package-data]
torchgeo = ["py.typed"]

[tool.setuptools.packages.find]
include = ["torchgeo*"]