diff --git a/pyproject.toml b/pyproject.toml index 8af9c1c2..b81e01ac 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -87,6 +87,7 @@ Repository = 'https://github.com/thousandbrainsproject.tbp.monty' Issues = 'https://github.com/thousandbrainsproject/tbp.monty/issues' [tool.deptry] +exclude= ['venv', '\.venv', '\.direnv', '\.git', 'setup\.py'] experimental_namespace_package = true # recognize tbp.monty as a namespace package ignore = ['DEP002'] known_first_party = [ @@ -94,6 +95,7 @@ known_first_party = [ 'habitat_sim', # imported via conda (habitat-sim) 'magnum', # transitive dependency bundled with habitat-sim 'quaternion', # imported via conda (quaternion) + 'tests', 'tools', 'torch', # imported via conda (pytorch) 'torchvision', # imported via conda (torchvision) diff --git a/tests/unit/dependencies_test.py b/tests/unit/dependencies_test.py index 59b19def..318181c3 100644 --- a/tests/unit/dependencies_test.py +++ b/tests/unit/dependencies_test.py @@ -15,7 +15,3 @@ def test_torch_geometric(self): import torch_geometric import torch_geometric.data import torch_geometric.typing # noqa: F401 - - def test_torch_sparse(self): - import torch_sparse # noqa: F401 - from torch_sparse import SparseTensor # noqa: F401 diff --git a/tools/generate_api_docs/source/conf.py b/tools/generate_api_docs/source/conf.py index a60078e2..ebcc5542 100644 --- a/tools/generate_api_docs/source/conf.py +++ b/tools/generate_api_docs/source/conf.py @@ -143,21 +143,16 @@ autodoc_member_order = "groupwise" autodoc_inherit_docstrings = True autodoc_mock_imports = [ - "PIL", "gym", "habitat_sim", "magnum", "matplotlib", - "numba", "numpy", "pandas", "quaternion", "scipy", - "skimage", - "sklearn", "torch", "torch_geometric", - "torch_sparse", "torchvision", ] # If true, the current module name will be prepended to all description @@ -180,7 +175,6 @@ "pillow": ("https://pillow.readthedocs.io/en/stable/", None), "python": ("https://docs.python.org/3", None), "torch": ("https://pytorch.org/docs/stable/", None), - "torch_sparse": ("https://pytorch.org/docs/stable/", None), "torch_geometric": ("https://pytorch-geometric.readthedocs.io/en/latest/", None), "torchvision": ("https://pytorch.org/vision/stable/", None), }