diff --git a/.gitignore b/.gitignore
index 5f25717..77d23d6 100644
--- a/.gitignore
+++ b/.gitignore
@@ -7,7 +7,7 @@ wheels/
 *.egg-info
 
 # Virtual environments
-.venv
+/.venv
 
 # Models
 /models
@@ -17,3 +17,6 @@ wheels/
 
 # Worktrees
 /worktree
+
+# IDE
+/.vscode
diff --git a/qmp/hamiltonian/hamiltonian.py b/qmp/hamiltonian/hamiltonian.py
index f6f4958..d56b9a3 100644
--- a/qmp/hamiltonian/hamiltonian.py
+++ b/qmp/hamiltonian/hamiltonian.py
@@ -3,7 +3,6 @@
 """
 
 import os
-import typing
 import platformdirs
 import torch
 import torch.utils.cpp_extension
@@ -153,9 +152,6 @@ def apply_within(
             A tensor of shape [batch_size_j] representing the output amplitudes on the given configurations.
         """
         self._prepare_data(configs_i.device)
-        _apply_within: typing.Callable[
-            [torch.Tensor, torch.Tensor, torch.Tensor, torch.Tensor, torch.Tensor, torch.Tensor], torch.Tensor
-        ]
         _apply_within = getattr(
             self._load_module(configs_i.device.type, configs_i.size(1), self.particle_cut),
             "apply_within",
@@ -195,9 +191,6 @@ def find_relative(
         if configs_exclude is None:
             configs_exclude = configs_i
         self._prepare_data(configs_i.device)
-        _find_relative: typing.Callable[
-            [torch.Tensor, torch.Tensor, int, torch.Tensor, torch.Tensor, torch.Tensor, torch.Tensor], torch.Tensor
-        ]
         _find_relative = getattr(
             self._load_module(configs_i.device.type, configs_i.size(1), self.particle_cut),
             "find_relative",
@@ -258,7 +251,6 @@ def diagonal_term(self, configs: torch.Tensor) -> torch.Tensor:
             A complex64 tensor of shape [batch_size] representing the diagonal term of the Hamiltonian for the given configurations.
         """
         self._prepare_data(configs.device)
-        _diagonal_term: typing.Callable[[torch.Tensor, torch.Tensor, torch.Tensor, torch.Tensor], torch.Tensor]
         _diagonal_term = getattr(
             self._load_module(configs.device.type, configs.size(1), self.particle_cut),
             "diagonal_term",