feat: delete not implemented error in convolutional module

PythonOT · Jan 8, 2025 · e8d4de5 · e8d4de5
1 parent 80d9542
commit e8d4de5
Showing 1 changed file with 1 addition and 10 deletions.
diff --git a/ot/bregman/_convolutional.py b/ot/bregman/_convolutional.py
@@ -246,11 +246,6 @@ def _convolutional_barycenter2d_log(
     A = list_to_array(A)
 
     nx = get_backend(A)
-    if nx.__name__ in ("jax", "tf"):
-        raise NotImplementedError(
-            "Log-domain functions are not yet implemented"
-            " for Jax and TF. Use numpy or torch arrays instead."
-        )
 
     n_hists, width, height = A.shape
 
@@ -483,11 +478,7 @@ def _convolutional_barycenter2d_debiased_log(
     A = list_to_array(A)
     n_hists, width, height = A.shape
     nx = get_backend(A)
-    if nx.__name__ in ("jax", "tf"):
-        raise NotImplementedError(
-            "Log-domain functions are not yet implemented"
-            " for Jax and TF. Use numpy or torch arrays instead."
-        )
+
     if weights is None:
         weights = nx.ones((n_hists,), type_as=A) / n_hists
     else: