Remove unused statement

dftd3 · Apr 21, 2024 · 93a1217 · 93a1217
1 parent 83b8dee
commit 93a1217
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 6 deletions.
diff --git a/src/tad_dftd3/model/c6.py b/src/tad_dftd3/model/c6.py
@@ -193,10 +193,7 @@ def _atomic_c6_full(
     # c6 = torch.sum(torch.sum(torch.mul(gw, rc6), dim=-1), dim=-1)
 
     rc6 = reference.c6[numbers.unsqueeze(-1), numbers.unsqueeze(-2)]
-    return torch.einsum(
-        "...ijab,...ia,...jb->...ij",
-        *(rc6, weights, weights),
-    )
+    return _einsum(rc6, weights, weights)
 
 
 def _atomic_c6_chunked(

diff --git a/test/test_model/test_c6.py b/test/test_model/test_c6.py
@@ -109,8 +109,6 @@ def test_chunked(dtype: torch.dtype, size: int, chunk_size: int) -> None:
 
 ###############################################################################
 
-Callable[[Tensor, Tensor, reference.Reference], Tensor]
-
 
 class C6Func(Protocol):
     """