From 93a1217e72d47b8b8b4989ddccc67771bb374cff Mon Sep 17 00:00:00 2001
From: marvinfriede <51965259+marvinfriede@users.noreply.github.com>
Date: Mon, 22 Apr 2024 00:30:54 +0200
Subject: [PATCH] Remove unused statement

---
 src/tad_dftd3/model/c6.py  | 5 +----
 test/test_model/test_c6.py | 2 --
 2 files changed, 1 insertion(+), 6 deletions(-)

diff --git a/src/tad_dftd3/model/c6.py b/src/tad_dftd3/model/c6.py
index 699b048..e55c3de 100644
--- 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
index 13c1b80..f49ad22 100644
--- 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):
     """