fix lint.

tests/test_convolution_fwt.py

@@ -313,6 +313,9 @@ def _compare_coeffs(
 
     Returns:
         A list with bools from allclose.
+
+    Raises:
+        TypeError: In case of a problem with the list structures.
     """
     test_list: List[bool] = []
     for ptwtcs, pywtcs in zip(ptwt_res, pywt_res):

tests/test_cwt.py

@@ -1,6 +1,6 @@
 """Test the continuous transformation code."""
 
-from typing import Any, Union
+from typing import Any
 
 import numpy as np
 import pytest

tests/test_jit.py

@@ -1,6 +1,6 @@
 """Ensure pytorch's torch.jit.trace feature works properly."""
 
-from typing import Any, Dict, List, NamedTuple, Optional, Tuple, Union
+from typing import Dict, List, NamedTuple, Optional, Tuple, Union
 
 import numpy as np
 import pytest