Keep docstring coverage policy unchanged

Author: loliverhennigh

physicsnemo/nn/functional/equivariant_ops.py

@@ -22,11 +22,15 @@
 
 
 @overload
-def smooth_log(x: Float[torch.Tensor, "..."]) -> Float[torch.Tensor, "..."]: ...
+def smooth_log(x: Float[torch.Tensor, "..."]) -> Float[torch.Tensor, "..."]:
+    """Apply smooth log elementwise to a tensor."""
+    ...
 
 
 @overload
-def smooth_log(x: TensorDict) -> TensorDict: ...
+def smooth_log(x: TensorDict) -> TensorDict:
+    """Apply smooth log elementwise to a TensorDict."""
+    ...
 
 
 def smooth_log(
@@ -59,11 +63,15 @@ def smooth_log(
 @overload
 def legendre_polynomials(
     x: Float[torch.Tensor, "..."], n: int
-) -> list[Float[torch.Tensor, "..."]]: ...
+) -> list[Float[torch.Tensor, "..."]]:
+    """Compute Legendre polynomials for a tensor input."""
+    ...
 
 
 @overload
-def legendre_polynomials(x: TensorDict, n: int) -> list[TensorDict]: ...
+def legendre_polynomials(x: TensorDict, n: int) -> list[TensorDict]:
+    """Compute Legendre polynomials for a TensorDict input."""
+    ...
 
 
 def legendre_polynomials(

test/ci_tests/check_docstring_coverage.py

@@ -148,15 +148,7 @@ def _read_baseline(baseline_path: Path) -> list[str]:
 def _run_interrogate(files: list[str], repo_root: Path) -> list[str]:
     """Run interrogate on *files* and return the MISSED items."""
     result = subprocess.run(  # noqa: S603 — command is hardcoded, not user input
-        [
-            sys.executable,
-            "-m",
-            "interrogate",
-            "-vv",
-            "--fail-under=0",
-            "--no-color",
-            "--ignore-overloaded-functions",
-        ]
+        [sys.executable, "-m", "interrogate", "-vv", "--fail-under=0", "--no-color"]
         + INTERROGATE_ARGS
         + ["--"]
         + files,