Fix for Python 3.8

tensorly/backend/pytorch_backend.py

@@ -216,11 +216,6 @@ def update_index(tensor, index, values):
     def lstsq(a, b, rcond=None, driver="gelsd"):
         return torch.linalg.lstsq(a, b, rcond=rcond, driver=driver)
 
-    @staticmethod
-    def eigh(tensor):
-        """Legacy only, deprecated from PyTorch 1.8.0"""
-        return torch.symeig(tensor, eigenvectors=True)
-
     @staticmethod
     def sign(tensor):
         """torch.sign does not support complex numbers."""

tensorly/utils/jointdiag.py

@@ -1,4 +1,3 @@
-from typing import Any
 from itertools import combinations
 import tensorly as tl
 
@@ -13,7 +12,7 @@ def jointdiag(
     max_iter: int = 50,
     threshold: float = 1e-10,
     verbose: bool = False,
-) -> tuple[Any, Any]:
+):
     """
     Jointly diagonalizes n matrices, organized in tensor of dimension (k,k,n).
     Returns the diagonalized matrices, along with the transformation matrix.