Rename KerasBackend -> TFKerasBackend to avoid confusion with keras 3.

arogozhnikov · arogozhnikov · commit b67cfea95639 · 2024-04-28T03:47:02.000Z
diff --git a/einops/_backends.py b/einops/_backends.py
@@ -466,7 +466,7 @@ def einsum(self, pattern, *x):
         return self.tf.einsum(pattern, *x)
 
 
-class KerasBackend(AbstractBackend):
+class TFKerasBackend(AbstractBackend):
     framework_name = "tensorflow.keras"
 
     def __init__(self):
diff --git a/tests/__init__.py b/tests/__init__.py
@@ -86,7 +86,7 @@ def collect_test_backends(symbolic=False, layers=False) -> List[_backends.Abstra
             backend_types = []
         else:
             backend_types = [
-                _backends.KerasBackend,
+                _backends.TFKerasBackend,
             ]
 
     backend_names_to_test = parse_backends_to_test()
diff --git a/tests/test_ops.py b/tests/test_ops.py
@@ -351,13 +351,13 @@ def logsumexp_numpy(x, tuple_of_axes):
         y = numpy.sum(y, axis=tuple_of_axes)
         return numpy.log(y) + minused
 
-    from einops._backends import TorchBackend, ChainerBackend, TensorflowBackend, KerasBackend, NumpyBackend
+    from einops._backends import TorchBackend, ChainerBackend, TensorflowBackend, TFKerasBackend, NumpyBackend
 
     backend2callback = {
         TorchBackend.framework_name: logsumexp_torch,
         ChainerBackend.framework_name: logsumexp_chainer,
         TensorflowBackend.framework_name: logsumexp_tf,
-        KerasBackend.framework_name: logsumexp_keras,
+        TFKerasBackend.framework_name: logsumexp_keras,
         NumpyBackend.framework_name: logsumexp_numpy,
     }
 

Original file line number	Diff line number	Diff line change
`@@ -86,7 +86,7 @@ def collect_test_backends(symbolic=False, layers=False) -> List[_backends.Abstra`
`86`	`86`	`backend_types = []`
`87`	`87`	`else:`
`88`	`88`	`backend_types = [`
`89`		`- _backends.KerasBackend,`
	`89`	`+ _backends.TFKerasBackend,`
`90`	`90`	`]`
`91`	`91`
`92`	`92`	`backend_names_to_test = parse_backends_to_test()`