More cleanup

Author: Aaron Meyer

tensorly/decomposition/tests/test_cp.py

@@ -18,7 +18,6 @@
     CP_NN_HALS,
 )
 from ...cp_tensor import cp_to_tensor
-from ...cp_tensor import cp_to_tensor
 from ...random import random_cp
 from ...tenalg import khatri_rao
 from ... import backend as T

tensorly/decomposition/tests/test_tr_svd.py

@@ -1,13 +1,10 @@
-import pytest
-import tensorly as tl
 
-from .._tr_svd import tensor_ring, TensorRing
+from .._tr_svd import tensor_ring
 from ...random import random_tr
 from ...testing import (
     assert_,
     assert_array_almost_equal,
     assert_raises,
-    assert_class_wrapper_correctly_passes_arguments,
 )
 
 

tensorly/random/base.py

@@ -1,4 +1,3 @@
-import numpy as np
 from ..cp_tensor import cp_to_tensor, CPTensor, cp_normalize, validate_cp_rank
 from ..tucker_tensor import tucker_to_tensor, TuckerTensor, validate_tucker_rank
 from ..tt_tensor import tt_to_tensor, TTTensor, validate_tt_rank

tensorly/solvers/penalizations.py

@@ -1,6 +1,3 @@
-import tensorly as tl
-import numpy as np
-import copy
 import warnings
 
 
@@ -24,10 +21,10 @@ def process_regularization_weights(ridge_coefficients, sparsity_coefficients, n_
     list of floats
         sparsity coefficients, processed
     """
-    if ridge_coefficients is None or isinstance(ridge_coefficients, (int, float)):
+    if ridge_coefficients is None or isinstance(ridge_coefficients, int | float):
         # Populate None or the input float in a list for all modes
         ridge_coefficients = [ridge_coefficients] * n_modes
-    if sparsity_coefficients is None or isinstance(sparsity_coefficients, (int, float)):
+    if sparsity_coefficients is None or isinstance(sparsity_coefficients, int | float):
         # Populate None or the input float in a list for all modes
         sparsity_coefficients = [sparsity_coefficients] * n_modes
     # Convert None to 0

tensorly/solvers/tests/test_admm.py

@@ -2,14 +2,13 @@
 import tensorly as tl
 
 from tensorly.solvers.admm import admm
-from tensorly.testing import assert_, assert_array_equal, assert_array_almost_equal
-from tensorly import tensor_to_vec, truncated_svd
+from tensorly.testing import assert_array_almost_equal
 import pytest
 
 # Author: Jean Kossaifi
 skip_tensorflow = pytest.mark.skipif(
     (tl.get_backend() == "tensorflow"),
-    reason=f"Indexing with list not supported in TensorFlow",
+    reason="Indexing with list not supported in TensorFlow",
 )
 
 

tensorly/solvers/tests/test_nnls.py

@@ -6,14 +6,14 @@
     fista,
     active_set_nnls,
 )
-from tensorly.testing import assert_, assert_array_equal, assert_array_almost_equal
-from tensorly import tensor_to_vec, truncated_svd
+from tensorly.testing import assert_array_almost_equal
+from tensorly import tensor_to_vec
 import pytest
 
 # Author: Jean Kossaifi
 skip_tensorflow = pytest.mark.skipif(
     (tl.get_backend() == "tensorflow"),
-    reason=f"Indexing with list not supported in TensorFlow",
+    reason="Indexing with list not supported in TensorFlow",
 )
 
 

tensorly/solvers/tests/test_penalizations.py

@@ -1,4 +1,3 @@
-import numpy as np
 import tensorly as tl
 
 from tensorly.solvers.penalizations import process_regularization_weights