Fix import pollution alert

dftd3 · Dec 21, 2023 · 3440455 · 3440455
1 parent 5f9bef3
commit 3440455
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 0 deletions.
diff --git a/src/tad_dftd3/typing/builtin.py b/src/tad_dftd3/typing/builtin.py
@@ -20,3 +20,5 @@
 handles some version checking.
 """
 from tad_mctc.typing import Any, Callable, NoReturn, TypedDict
+
+__all__ = ["Any", "Callable", "NoReturn", "TypedDict"]
diff --git a/src/tad_dftd3/typing/d3.py b/src/tad_dftd3/typing/d3.py
@@ -20,4 +20,7 @@
 """
 from tad_mctc.typing import Callable, Tensor
 
+__all__ = ["WeightingFunction"]
+
+
 WeightingFunction = Callable[[Tensor], Tensor]
diff --git a/src/tad_dftd3/typing/pytorch.py b/src/tad_dftd3/typing/pytorch.py
@@ -28,3 +28,14 @@
     get_default_device,
     get_default_dtype,
 )
+
+__all__ = [
+    "DD",
+    "CountingFunction",
+    "DampingFunction",
+    "Molecule",
+    "Tensor",
+    "TensorOrTensors",
+    "get_default_device",
+    "get_default_dtype",
+]