File tree Expand file tree Collapse file tree 1 file changed +4
-0
lines changed
Expand file tree Collapse file tree 1 file changed +4
-0
lines changed Original file line number Diff line number Diff line change @@ -246,6 +246,8 @@ def _convolutional_barycenter2d_log(
246246 A = list_to_array (A )
247247
248248 nx = get_backend (A )
249+ # This error is raised because we are using mutable assignment in the line
250+ # `log_KU[k] = ...` which is not allowed in Jax and TF.
249251 if nx .__name__ in ("jax" , "tf" ):
250252 raise NotImplementedError (
251253 "Log-domain functions are not yet implemented"
@@ -483,6 +485,8 @@ def _convolutional_barycenter2d_debiased_log(
483485 A = list_to_array (A )
484486 n_hists , width , height = A .shape
485487 nx = get_backend (A )
488+ # This error is raised because we are using mutable assignment in the line
489+ # `log_KU[k] = ...` which is not allowed in Jax and TF.
486490 if nx .__name__ in ("jax" , "tf" ):
487491 raise NotImplementedError (
488492 "Log-domain functions are not yet implemented"
You can’t perform that action at this time.
0 commit comments