============================= test session starts ============================== platform linux -- Python 3.9.12, pytest-7.1.2, pluggy-1.0.0 rootdir: /opt/projects/helmholtz/ott_mk plugins: cov-3.0.0, forked-1.4.0, xdist-2.5.0, mock-3.8.1, memray-1.2.0 collected 441 items tests/core/continuous_barycenter_test.py ........................ [ 5%] tests/core/discrete_barycenter_test.py .... [ 6%] tests/core/fused_gromov_wasserstein_test.py ............... [ 9%] tests/core/gromov_wasserstein_test.py ................ [ 13%] tests/core/gromov_wasserstein_unbalanced_test.py ..... [ 14%] tests/core/icnn_test.py .. [ 14%] tests/core/neuraldual_test.py .. [ 15%] tests/core/sinkhorn_anderson_acceleration_test.py ................ [ 19%] tests/core/sinkhorn_bures_test.py .. [ 19%] tests/core/sinkhorn_diff_grid_loc_test.py .. [ 19%] tests/core/sinkhorn_diff_grid_weights_test.py .. [ 20%] tests/core/sinkhorn_diff_precond_test.py ............................... [ 27%] ................. [ 31%] tests/core/sinkhorn_diff_test.py ........... [ 33%] tests/core/sinkhorn_grid_test.py ....... [ 35%] tests/core/sinkhorn_hessian_test.py ................ [ 39%] tests/core/sinkhorn_implicit_lse_test.py . [ 39%] tests/core/sinkhorn_implicit_test.py . [ 39%] tests/core/sinkhorn_jacobian_apply_test.py ............................. [ 46%] ... [ 46%] tests/core/sinkhorn_jit_test.py ... [ 47%] tests/core/sinkhorn_lr_test.py ...... [ 48%] tests/core/sinkhorn_online_large_test.py ........ [ 50%] tests/core/sinkhorn_potentials_jacobian_test.py ........................ [ 56%] ........ [ 57%] tests/core/sinkhorn_test.py ......................... [ 63%] tests/core/sinkhorn_unbalanced_test.py .... [ 64%] tests/geometry/geometry_costs_test.py . [ 64%] tests/geometry/geometry_lr_test.py ........... [ 67%] tests/geometry/geometry_lse_test.py . [ 67%] tests/geometry/geometry_pointcloud_apply_test.py ... [ 68%] tests/geometry/geometry_pointcloud_generalcost_test.py . [ 68%] tests/geometry/matrix_square_root_test.py ......... [ 70%] tests/geometry/scaling_cost_test.py ........................... [ 76%] tests/tools/sinkhorn_divergence_differentiability_test.py . [ 76%] tests/tools/sinkhorn_divergence_test.py ................ [ 80%] tests/tools/soft_sort_test.py ................... [ 84%] tests/tools/transport_test.py .... [ 85%] tests/tools/gaussian_mixture/fit_gmm_pair_test.py .... [ 86%] tests/tools/gaussian_mixture/fit_gmm_test.py . [ 86%] tests/tools/gaussian_mixture/gaussian_mixture_pair_test.py ............. [ 89%] [ 89%] tests/tools/gaussian_mixture/gaussian_mixture_test.py ......... [ 91%] tests/tools/gaussian_mixture/gaussian_test.py .......... [ 93%] tests/tools/gaussian_mixture/linalg_test.py .......... [ 96%] tests/tools/gaussian_mixture/probabilities_test.py ....... [ 97%] tests/tools/gaussian_mixture/scale_tril_test.py .......... [100%] =============================== warnings summary =============================== ../../../../home/michal/.miniconda3/envs/ott/lib/python3.9/site-packages/flatbuffers/compat.py:19 /home/michal/.miniconda3/envs/ott/lib/python3.9/site-packages/flatbuffers/compat.py:19: DeprecationWarning: the imp module is deprecated in favour of importlib; see the module's documentation for alternative uses import imp -- Docs: https://docs.pytest.org/en/stable/how-to/capture-warnings.html ================================ MEMRAY REPORT ================================= Allocations results for tests/core/fused_gromov_wasserstein_test.py::test_fgw_lr_memory[True] 📦 Total memory allocated: 108.8MiB 📏 Total allocations: 16421332 📊 Histogram of allocation sizes: | ▂ █ | 🥇 Biggest allocating functions: - backend_compile:/home/michal/.miniconda3/envs/ott/lib/python3.9/site-packages/jax/_src/dispatch.py:713 -> 90.8MiB - make_ir_context:/home/michal/.miniconda3/envs/ott/lib/python3.9/site-packages/jax/interpreters/mlir.py:321 -> 775.5KiB - module_to_string:/home/michal/.miniconda3/envs/ott/lib/python3.9/site-packages/jax/interpreters/mlir.py:569 -> 761.4KiB - wrapper:/home/michal/.miniconda3/envs/ott/lib/python3.9/site-packages/jax/_src/profiler.py:206 -> 360.5KiB - deferring_binary_op:/home/michal/.miniconda3/envs/ott/lib/python3.9/site-packages/jax/_src/numpy/lax_numpy.py:4557 -> 295.1KiB Allocations results for tests/core/fused_gromov_wasserstein_test.py::test_fgw_lr_memory[False] 📦 Total memory allocated: 78.6MiB 📏 Total allocations: 23156954 📊 Histogram of allocation sizes: | ▃ █ | 🥇 Biggest allocating functions: - backend_compile:/home/michal/.miniconda3/envs/ott/lib/python3.9/site-packages/jax/_src/dispatch.py:713 -> 42.5MiB - backend_compile:/home/michal/.miniconda3/envs/ott/lib/python3.9/site-packages/jax/_src/dispatch.py:713 -> 2.4MiB - backend_compile:/home/michal/.miniconda3/envs/ott/lib/python3.9/site-packages/jax/_src/dispatch.py:713 -> 2.4MiB - _execute_compiled:/home/michal/.miniconda3/envs/ott/lib/python3.9/site-packages/jax/_src/dispatch.py:615 -> 1.7MiB - backend_compile:/home/michal/.miniconda3/envs/ott/lib/python3.9/site-packages/jax/_src/dispatch.py:713 -> 807.4KiB ============================== slowest durations =============================== 53.15s call tests/tools/soft_sort_test.py::SoftSortTest::test_quantile_on_several_axes 47.51s call tests/core/sinkhorn_hessian_test.py::SinkhornHessianTest::test_hessian_sinkhorn13 46.81s call tests/core/sinkhorn_hessian_test.py::SinkhornHessianTest::test_hessian_sinkhorn7 46.30s call tests/core/sinkhorn_hessian_test.py::SinkhornHessianTest::test_hessian_sinkhorn11 45.52s call tests/core/sinkhorn_hessian_test.py::SinkhornHessianTest::test_hessian_sinkhorn4 45.39s call tests/core/sinkhorn_hessian_test.py::SinkhornHessianTest::test_hessian_sinkhorn5 42.79s call tests/core/sinkhorn_jacobian_apply_test.py::SinkhornJacobianTest::test_apply_transport_jacobian14 41.71s call tests/core/sinkhorn_hessian_test.py::SinkhornHessianTest::test_hessian_sinkhorn2 40.81s call tests/core/sinkhorn_hessian_test.py::SinkhornHessianTest::test_hessian_sinkhorn6 40.63s call tests/core/sinkhorn_jacobian_apply_test.py::SinkhornJacobianTest::test_apply_transport_jacobian9 40.63s call tests/core/sinkhorn_hessian_test.py::SinkhornHessianTest::test_hessian_sinkhorn14 40.51s call tests/core/sinkhorn_hessian_test.py::SinkhornHessianTest::test_hessian_sinkhorn15 40.08s call tests/core/sinkhorn_hessian_test.py::SinkhornHessianTest::test_hessian_sinkhorn9 39.89s call tests/core/sinkhorn_hessian_test.py::SinkhornHessianTest::test_hessian_sinkhorn12 39.80s call tests/core/sinkhorn_hessian_test.py::SinkhornHessianTest::test_hessian_sinkhorn3 39.33s call tests/core/sinkhorn_hessian_test.py::SinkhornHessianTest::test_hessian_sinkhorn0 39.15s call tests/core/sinkhorn_jacobian_apply_test.py::SinkhornJacobianTest::test_apply_transport_jacobian8 39.01s call tests/core/sinkhorn_jacobian_apply_test.py::SinkhornJacobianTest::test_apply_transport_jacobian7 38.04s call tests/core/sinkhorn_jacobian_apply_test.py::SinkhornJacobianTest::test_apply_transport_jacobian11 36.58s call tests/core/sinkhorn_hessian_test.py::SinkhornHessianTest::test_hessian_sinkhorn1 36.07s call tests/core/sinkhorn_hessian_test.py::SinkhornHessianTest::test_hessian_sinkhorn10 34.94s call tests/core/sinkhorn_jacobian_apply_test.py::SinkhornJacobianTest::test_apply_transport_jacobian10 33.14s call tests/core/sinkhorn_jacobian_apply_test.py::SinkhornJacobianTest::test_apply_transport_jacobian6 32.93s call tests/core/sinkhorn_jacobian_apply_test.py::SinkhornJacobianTest::test_apply_transport_jacobian13 31.53s call tests/core/sinkhorn_jacobian_apply_test.py::SinkhornJacobianTest::test_apply_transport_jacobian15 31.45s call tests/core/sinkhorn_implicit_lse_test.py::SinkhornTest::test_implicit_differentiation_versus_autodiff0 31.17s call tests/core/sinkhorn_jacobian_apply_test.py::SinkhornJacobianTest::test_apply_transport_jacobian12 29.48s call tests/core/sinkhorn_jacobian_apply_test.py::SinkhornJacobianTest::test_apply_transport_jacobian4 29.31s call tests/core/sinkhorn_jacobian_apply_test.py::SinkhornJacobianTest::test_apply_transport_jacobian5 28.65s call tests/tools/soft_sort_test.py::SoftSortTest::test_soft_sort_jacobian0 27.19s call tests/core/sinkhorn_hessian_test.py::SinkhornHessianTest::test_hessian_sinkhorn8 26.33s call tests/core/sinkhorn_jacobian_apply_test.py::SinkhornJacobianTest::test_apply_transport_jacobian1 25.25s call tests/core/sinkhorn_jacobian_apply_test.py::SinkhornJacobianTest::test_apply_transport_jacobian0 24.38s call tests/tools/gaussian_mixture/fit_gmm_pair_test.py::FitGmmPairTest::test_fit_gmm_balanced_unweighted 23.35s call tests/core/continuous_barycenter_test.py::Barycenter::test_bures_barycenter2 23.21s call tests/core/continuous_barycenter_test.py::Barycenter::test_bures_barycenter0 23.10s call tests/core/sinkhorn_jacobian_apply_test.py::SinkhornJacobianTest::test_apply_transport_jacobian2 22.85s call tests/core/sinkhorn_jacobian_apply_test.py::SinkhornJacobianTest::test_apply_transport_jacobian3 22.37s call tests/core/continuous_barycenter_test.py::Barycenter::test_bures_barycenter3 21.09s call tests/core/sinkhorn_potentials_jacobian_test.py::SinkhornJacobianTest::test_potential_jacobian_sinkhorn4 20.35s call tests/core/sinkhorn_implicit_test.py::SinkhornTest::test_implicit_differentiation_versus_autodiff0 19.82s call tests/core/continuous_barycenter_test.py::Barycenter::test_bures_barycenter4 19.46s call tests/core/sinkhorn_diff_precond_test.py::SinkhornJacobianPreconditioningTest::test_potential_jacobian_sinkhorn23 19.35s call tests/core/continuous_barycenter_test.py::Barycenter::test_bures_barycenter1 19.34s call tests/core/sinkhorn_diff_precond_test.py::SinkhornJacobianPreconditioningTest::test_potential_jacobian_sinkhorn5 18.98s call tests/core/sinkhorn_diff_precond_test.py::SinkhornJacobianPreconditioningTest::test_potential_jacobian_sinkhorn4 18.68s call tests/core/sinkhorn_diff_precond_test.py::SinkhornJacobianPreconditioningTest::test_potential_jacobian_sinkhorn11 18.57s call tests/core/sinkhorn_jacobian_apply_test.py::SinkhornJacobianTest::test_apply_transport_jacobian26 18.49s call tests/core/continuous_barycenter_test.py::Barycenter::test_bures_barycenter7 18.03s call tests/core/fused_gromov_wasserstein_test.py::test_fgw_lr_memory[False] 17.81s call tests/core/sinkhorn_diff_precond_test.py::SinkhornJacobianPreconditioningTest::test_potential_jacobian_sinkhorn10 17.65s call tests/core/sinkhorn_diff_precond_test.py::SinkhornJacobianPreconditioningTest::test_potential_jacobian_sinkhorn16 17.64s call tests/tools/sinkhorn_divergence_test.py::SinkhornDivergenceTest::test_segment_sinkhorn_different_segment_sizes 17.54s call tests/core/sinkhorn_diff_precond_test.py::SinkhornJacobianPreconditioningTest::test_potential_jacobian_sinkhorn17 17.10s call tests/core/continuous_barycenter_test.py::Barycenter::test_bures_barycenter5 16.61s call tests/core/sinkhorn_potentials_jacobian_test.py::SinkhornJacobianTest::test_potential_jacobian_sinkhorn18 16.55s call tests/core/sinkhorn_diff_precond_test.py::SinkhornJacobianPreconditioningTest::test_potential_jacobian_sinkhorn9 15.77s call tests/core/continuous_barycenter_test.py::Barycenter::test_bures_barycenter6 15.73s call tests/core/sinkhorn_diff_precond_test.py::SinkhornJacobianPreconditioningTest::test_potential_jacobian_sinkhorn22 15.59s call tests/tools/gaussian_mixture/fit_gmm_pair_test.py::FitGmmPairTest::test_fit_gmm_unbalanced_unweighted 15.48s call tests/tools/soft_sort_test.py::SoftSortTest::test_soft_sort_jacobian1 15.27s call tests/core/sinkhorn_jit_test.py::SinkhornTest::test_jit_vs_non_jit_bwd1 15.09s call tests/core/sinkhorn_potentials_jacobian_test.py::SinkhornJacobianTest::test_potential_jacobian_sinkhorn25 14.81s call tests/core/sinkhorn_diff_precond_test.py::SinkhornJacobianPreconditioningTest::test_potential_jacobian_sinkhorn13 14.55s call tests/core/fused_gromov_wasserstein_test.py::test_fgw_lr_memory[True] 14.52s call tests/core/sinkhorn_potentials_jacobian_test.py::SinkhornJacobianTest::test_potential_jacobian_sinkhorn14 14.36s call tests/core/sinkhorn_potentials_jacobian_test.py::SinkhornJacobianTest::test_potential_jacobian_sinkhorn3 13.96s call tests/core/sinkhorn_potentials_jacobian_test.py::SinkhornJacobianTest::test_potential_jacobian_sinkhorn7 13.90s call tests/core/sinkhorn_potentials_jacobian_test.py::SinkhornJacobianTest::test_potential_jacobian_sinkhorn13 13.86s call tests/core/sinkhorn_potentials_jacobian_test.py::SinkhornJacobianTest::test_potential_jacobian_sinkhorn11 13.62s call tests/core/sinkhorn_potentials_jacobian_test.py::SinkhornJacobianTest::test_potential_jacobian_sinkhorn10 13.47s call tests/core/sinkhorn_potentials_jacobian_test.py::SinkhornJacobianTest::test_potential_jacobian_sinkhorn5 13.45s call tests/core/sinkhorn_grid_test.py::SinkhornGridTest::test_apply_transport_grid0 13.24s call tests/core/sinkhorn_potentials_jacobian_test.py::SinkhornJacobianTest::test_potential_jacobian_sinkhorn6 12.82s call tests/core/sinkhorn_diff_precond_test.py::SinkhornJacobianPreconditioningTest::test_potential_jacobian_sinkhorn7 12.77s call tests/tools/gaussian_mixture/fit_gmm_pair_test.py::FitGmmPairTest::test_fit_gmm_balanced_weighted 12.65s call tests/core/sinkhorn_diff_precond_test.py::SinkhornJacobianPreconditioningTest::test_potential_jacobian_sinkhorn8 12.61s call tests/core/sinkhorn_potentials_jacobian_test.py::SinkhornJacobianTest::test_potential_jacobian_sinkhorn12 12.57s call tests/core/sinkhorn_potentials_jacobian_test.py::SinkhornJacobianTest::test_potential_jacobian_sinkhorn30 12.46s call tests/core/sinkhorn_potentials_jacobian_test.py::SinkhornJacobianTest::test_potential_jacobian_sinkhorn26 12.42s call tests/core/sinkhorn_diff_precond_test.py::SinkhornJacobianPreconditioningTest::test_potential_jacobian_sinkhorn15 12.42s call tests/tools/gaussian_mixture/fit_gmm_pair_test.py::FitGmmPairTest::test_fit_gmm_unbalanced_weighted 12.36s call tests/core/sinkhorn_potentials_jacobian_test.py::SinkhornJacobianTest::test_potential_jacobian_sinkhorn31 12.35s call tests/core/sinkhorn_potentials_jacobian_test.py::SinkhornJacobianTest::test_potential_jacobian_sinkhorn27 12.30s call tests/core/sinkhorn_diff_precond_test.py::SinkhornJacobianPreconditioningTest::test_potential_jacobian_sinkhorn34 12.20s call tests/core/gromov_wasserstein_test.py::GromovWassersteinTest::test_gw_lr_fused 12.00s call tests/core/sinkhorn_diff_precond_test.py::SinkhornJacobianPreconditioningTest::test_potential_jacobian_sinkhorn42 11.92s call tests/core/sinkhorn_potentials_jacobian_test.py::SinkhornJacobianTest::test_potential_jacobian_sinkhorn24 11.44s call tests/core/sinkhorn_diff_precond_test.py::SinkhornJacobianPreconditioningTest::test_potential_jacobian_sinkhorn2 11.22s call tests/core/sinkhorn_potentials_jacobian_test.py::SinkhornJacobianTest::test_potential_jacobian_sinkhorn21 11.06s call tests/core/sinkhorn_diff_precond_test.py::SinkhornJacobianPreconditioningTest::test_potential_jacobian_sinkhorn21 11.01s call tests/core/sinkhorn_potentials_jacobian_test.py::SinkhornJacobianTest::test_potential_jacobian_sinkhorn1 10.97s call tests/core/sinkhorn_jacobian_apply_test.py::SinkhornJacobianTest::test_apply_transport_jacobian30 10.95s call tests/core/sinkhorn_diff_precond_test.py::SinkhornJacobianPreconditioningTest::test_potential_jacobian_sinkhorn18 10.90s call tests/core/sinkhorn_diff_precond_test.py::SinkhornJacobianPreconditioningTest::test_potential_jacobian_sinkhorn1 10.83s call tests/core/sinkhorn_diff_precond_test.py::SinkhornJacobianPreconditioningTest::test_potential_jacobian_sinkhorn19 10.74s call tests/core/sinkhorn_jacobian_apply_test.py::SinkhornJacobianTest::test_apply_transport_jacobian31 10.74s call tests/core/sinkhorn_diff_precond_test.py::SinkhornJacobianPreconditioningTest::test_potential_jacobian_sinkhorn14 10.74s call tests/core/sinkhorn_potentials_jacobian_test.py::SinkhornJacobianTest::test_potential_jacobian_sinkhorn29 10.72s call tests/core/sinkhorn_potentials_jacobian_test.py::SinkhornJacobianTest::test_potential_jacobian_sinkhorn0 10.68s call tests/core/sinkhorn_potentials_jacobian_test.py::SinkhornJacobianTest::test_potential_jacobian_sinkhorn23 10.67s call tests/core/fused_gromov_wasserstein_test.py::FusedGromovWassersteinTest::test_adaptive_threshold_fused 10.66s call tests/core/sinkhorn_potentials_jacobian_test.py::SinkhornJacobianTest::test_potential_jacobian_sinkhorn2 10.62s call tests/core/sinkhorn_diff_precond_test.py::SinkhornJacobianPreconditioningTest::test_potential_jacobian_sinkhorn12 10.57s call tests/core/sinkhorn_potentials_jacobian_test.py::SinkhornJacobianTest::test_potential_jacobian_sinkhorn15 10.56s call tests/core/sinkhorn_diff_precond_test.py::SinkhornJacobianPreconditioningTest::test_potential_jacobian_sinkhorn20 10.53s call tests/core/sinkhorn_diff_precond_test.py::SinkhornJacobianPreconditioningTest::test_potential_jacobian_sinkhorn26 10.20s call tests/core/gromov_wasserstein_test.py::GromovWassersteinTest::test_gw_lr 10.19s call tests/core/sinkhorn_potentials_jacobian_test.py::SinkhornJacobianTest::test_potential_jacobian_sinkhorn28 10.08s call tests/core/sinkhorn_potentials_jacobian_test.py::SinkhornJacobianTest::test_potential_jacobian_sinkhorn22 10.02s call tests/tools/sinkhorn_divergence_test.py::SinkhornDivergenceTest::test_segment_sinkhorn_result0 10.02s call tests/tools/sinkhorn_divergence_differentiability_test.py::SinkhornDivergenceGradTest::test_gradient_generic_point_cloud_wrapper 9.97s call tests/core/sinkhorn_test.py::SinkhornTest::test_apply_transport_geometry_from_potentials 9.97s call tests/core/sinkhorn_potentials_jacobian_test.py::SinkhornJacobianTest::test_potential_jacobian_sinkhorn9 9.92s call tests/core/sinkhorn_diff_precond_test.py::SinkhornJacobianPreconditioningTest::test_potential_jacobian_sinkhorn46 9.78s call tests/core/sinkhorn_jacobian_apply_test.py::SinkhornJacobianTest::test_apply_transport_jacobian21 9.59s call tests/core/sinkhorn_diff_precond_test.py::SinkhornJacobianPreconditioningTest::test_potential_jacobian_sinkhorn41 9.57s call tests/core/sinkhorn_potentials_jacobian_test.py::SinkhornJacobianTest::test_potential_jacobian_sinkhorn8 9.54s call tests/core/sinkhorn_diff_precond_test.py::SinkhornJacobianPreconditioningTest::test_potential_jacobian_sinkhorn6 9.52s call tests/core/sinkhorn_bures_test.py::SinkhornTest::test_bures_point_cloud_ker-batch 9.41s call tests/core/sinkhorn_diff_grid_weights_test.py::SinkhornGradGridTest::test_diff_sinkhorn_x_grid_weights_perturbation0 9.33s call tests/core/sinkhorn_diff_precond_test.py::SinkhornJacobianPreconditioningTest::test_potential_jacobian_sinkhorn35 9.32s call tests/core/sinkhorn_diff_grid_loc_test.py::SinkhornGradGridTest::test_diff_sinkhorn_x_grid_x_perturbation0 9.26s call tests/core/sinkhorn_diff_precond_test.py::SinkhornJacobianPreconditioningTest::test_potential_jacobian_sinkhorn0 9.22s call tests/core/sinkhorn_diff_precond_test.py::SinkhornJacobianPreconditioningTest::test_potential_jacobian_sinkhorn40 9.13s call tests/core/sinkhorn_jacobian_apply_test.py::SinkhornJacobianTest::test_apply_transport_jacobian19 9.13s call tests/core/sinkhorn_potentials_jacobian_test.py::SinkhornJacobianTest::test_potential_jacobian_sinkhorn20 9.06s call tests/core/sinkhorn_diff_precond_test.py::SinkhornJacobianPreconditioningTest::test_potential_jacobian_sinkhorn47 9.05s call tests/core/sinkhorn_diff_precond_test.py::SinkhornJacobianPreconditioningTest::test_potential_jacobian_sinkhorn3 8.93s call tests/core/sinkhorn_potentials_jacobian_test.py::SinkhornJacobianTest::test_potential_jacobian_sinkhorn16 8.91s call tests/core/sinkhorn_jacobian_apply_test.py::SinkhornJacobianTest::test_apply_transport_jacobian23 8.90s call tests/core/sinkhorn_jacobian_apply_test.py::SinkhornJacobianTest::test_apply_transport_jacobian24 8.83s call tests/core/sinkhorn_jacobian_apply_test.py::SinkhornJacobianTest::test_apply_transport_jacobian29 8.81s call tests/core/sinkhorn_jacobian_apply_test.py::SinkhornJacobianTest::test_apply_transport_jacobian27 8.70s call tests/core/sinkhorn_diff_precond_test.py::SinkhornJacobianPreconditioningTest::test_potential_jacobian_sinkhorn36 8.62s call tests/core/sinkhorn_jacobian_apply_test.py::SinkhornJacobianTest::test_apply_transport_jacobian25 8.51s call tests/core/sinkhorn_jacobian_apply_test.py::SinkhornJacobianTest::test_apply_transport_jacobian20 8.40s call tests/core/sinkhorn_diff_precond_test.py::SinkhornJacobianPreconditioningTest::test_potential_jacobian_sinkhorn33 8.30s call tests/core/sinkhorn_jacobian_apply_test.py::SinkhornJacobianTest::test_apply_transport_jacobian22 8.29s call tests/core/sinkhorn_jacobian_apply_test.py::SinkhornJacobianTest::test_apply_transport_jacobian16 8.28s call tests/core/sinkhorn_jacobian_apply_test.py::SinkhornJacobianTest::test_apply_transport_jacobian18 8.24s call tests/core/sinkhorn_diff_precond_test.py::SinkhornJacobianPreconditioningTest::test_potential_jacobian_sinkhorn39 8.22s call tests/core/icnn_test.py::ICNNTest::test_icnn_convexity0 8.21s call tests/core/sinkhorn_jacobian_apply_test.py::SinkhornJacobianTest::test_apply_transport_jacobian28 8.13s call tests/core/sinkhorn_jacobian_apply_test.py::SinkhornJacobianTest::test_apply_transport_jacobian17 8.12s call tests/core/fused_gromov_wasserstein_test.py::FusedGromovWassersteinTest::test_gradient_fused_gromov_wasserstein_pointcloud0 8.00s call tests/core/gromov_wasserstein_unbalanced_test.py::GromovWassersteinUnbalancedTest::test_gradient_gromov_wasserstein_pointcloud0 7.97s call tests/core/sinkhorn_diff_precond_test.py::SinkhornJacobianPreconditioningTest::test_potential_jacobian_sinkhorn43 7.91s call tests/core/sinkhorn_diff_test.py::SinkhornJacobianTest::test_gradient_sinkhorn_euclidean_lse-backprop 7.88s call tests/core/sinkhorn_diff_precond_test.py::SinkhornJacobianPreconditioningTest::test_potential_jacobian_sinkhorn27 7.87s call tests/core/sinkhorn_diff_precond_test.py::SinkhornJacobianPreconditioningTest::test_potential_jacobian_sinkhorn24 7.86s call tests/tools/soft_sort_test.py::SoftSortTest::test_sort_one_array0 7.69s call tests/core/sinkhorn_diff_precond_test.py::SinkhornJacobianPreconditioningTest::test_potential_jacobian_sinkhorn25 7.69s call tests/core/continuous_barycenter_test.py::Barycenter::test_euclidean_barycenter9 7.58s call tests/core/continuous_barycenter_test.py::Barycenter::test_euclidean_barycenter13 7.52s call tests/core/sinkhorn_potentials_jacobian_test.py::SinkhornJacobianTest::test_potential_jacobian_sinkhorn17 7.48s call tests/core/sinkhorn_potentials_jacobian_test.py::SinkhornJacobianTest::test_potential_jacobian_sinkhorn19 7.45s call tests/core/gromov_wasserstein_test.py::GromovWassersteinTest::test_adaptive_threshold 7.39s call tests/core/continuous_barycenter_test.py::Barycenter::test_euclidean_barycenter11 7.29s call tests/core/sinkhorn_diff_precond_test.py::SinkhornJacobianPreconditioningTest::test_potential_jacobian_sinkhorn29 7.25s call tests/core/sinkhorn_diff_precond_test.py::SinkhornJacobianPreconditioningTest::test_potential_jacobian_sinkhorn37 7.24s call tests/core/sinkhorn_diff_precond_test.py::SinkhornJacobianPreconditioningTest::test_potential_jacobian_sinkhorn45 7.15s call tests/core/sinkhorn_diff_precond_test.py::SinkhornJacobianPreconditioningTest::test_potential_jacobian_sinkhorn44 7.12s call tests/core/sinkhorn_diff_precond_test.py::SinkhornJacobianPreconditioningTest::test_potential_jacobian_sinkhorn38 7.06s call tests/core/continuous_barycenter_test.py::Barycenter::test_euclidean_barycenter10 6.97s call tests/core/continuous_barycenter_test.py::Barycenter::test_euclidean_barycenter15 6.96s call tests/core/continuous_barycenter_test.py::Barycenter::test_euclidean_barycenter14 6.95s call tests/core/sinkhorn_diff_precond_test.py::SinkhornJacobianPreconditioningTest::test_potential_jacobian_sinkhorn28 6.82s call tests/tools/gaussian_mixture/gaussian_mixture_pair_test.py::GaussianMixturePairTest::test_get_cost_matrix 6.77s call tests/core/continuous_barycenter_test.py::Barycenter::test_euclidean_barycenter12 6.77s call tests/core/sinkhorn_diff_precond_test.py::SinkhornJacobianPreconditioningTest::test_potential_jacobian_sinkhorn32 6.75s call tests/core/fused_gromov_wasserstein_test.py::FusedGromovWassersteinTest::test_gradient_fused_gromov_wasserstein_geometry0 6.72s call tests/core/sinkhorn_diff_precond_test.py::SinkhornJacobianPreconditioningTest::test_potential_jacobian_sinkhorn31 6.70s call tests/core/continuous_barycenter_test.py::Barycenter::test_euclidean_barycenter8 6.61s call tests/core/gromov_wasserstein_test.py::GromovWassersteinTest::test_gradient_gromov_wasserstein_pointcloud0 6.54s call tests/core/sinkhorn_diff_precond_test.py::SinkhornJacobianPreconditioningTest::test_potential_jacobian_sinkhorn30 6.53s call tests/core/gromov_wasserstein_unbalanced_test.py::GromovWassersteinUnbalancedTest::test_gradient_gromov_wasserstein_geometry0 6.47s call tests/tools/sinkhorn_divergence_test.py::SinkhornDivergenceTest::test_euclidean_point_cloud 6.44s call tests/tools/sinkhorn_divergence_test.py::SinkhornDivergenceTest::test_segment_sinkhorn_result1 6.44s call tests/core/neuraldual_test.py::NeuralDualTest::test_neural_dual_convergence0 6.42s call tests/core/fused_gromov_wasserstein_test.py::FusedGromovWassersteinTest::test_gradient_marginals_fused_gromov_wasserstein0 6.40s call tests/core/sinkhorn_lr_test.py::SinkhornLRTest::test_euclidean_point_cloud1 6.28s call tests/tools/soft_sort_test.py::SoftSortTest::test_sort_array_squashing_momentum 6.27s call tests/core/continuous_barycenter_test.py::Barycenter::test_euclidean_barycenter0 6.18s call tests/core/sinkhorn_diff_test.py::SinkhornJacobianTest::test_differentiability_with_jit 6.11s call tests/core/discrete_barycenter_test.py::DiscreteBarycenterTest::test_discrete_barycenter_grid_lse-deb 6.08s call tests/tools/sinkhorn_divergence_test.py::SinkhornDivergenceTest::test_generic_point_cloud_wrapper 5.96s call tests/core/neuraldual_test.py::NeuralDualTest::test_neural_dual_jit0 5.71s call tests/core/sinkhorn_anderson_acceleration_test.py::SinkhornAndersonTest::test_anderson0 5.70s call tests/core/fused_gromov_wasserstein_test.py::FusedGromovWassersteinTest::test_gradient_fused_gromov_wasserstein_pointcloud1 5.65s call tests/geometry/matrix_square_root_test.py::MatrixSquareRootTest::test_grad_test_sqrtm_inv_sqrtm 5.64s call tests/core/sinkhorn_diff_test.py::SinkhornJacobianTest::test_gradient_sinkhorn_euclidean_lse-backprop-force_scan 5.57s call tests/core/gromov_wasserstein_unbalanced_test.py::GromovWassersteinUnbalancedTest::test_gradient_gromov_wasserstein_pointcloud1 5.57s call tests/core/sinkhorn_online_large_test.py::SinkhornOnlineTest::test_online_matches_offline_size0 5.51s call tests/core/fused_gromov_wasserstein_test.py::FusedGromovWassersteinTest::test_gradient_marginals_fused_gromov_wasserstein1 5.35s call tests/core/gromov_wasserstein_test.py::GromovWassersteinTest::test_gradient_marginals_gromov_wasserstein0 5.27s call tests/core/continuous_barycenter_test.py::Barycenter::test_euclidean_barycenter4 5.25s call tests/core/sinkhorn_lr_test.py::SinkhornLRTest::test_euclidean_point_cloud3 5.05s call tests/core/sinkhorn_lr_test.py::SinkhornLRTest::test_euclidean_point_cloud0 5.00s call tests/geometry/matrix_square_root_test.py::MatrixSquareRootTest::test_grad_test_inv_sqrtm_only 4.99s call tests/core/fused_gromov_wasserstein_test.py::FusedGromovWassersteinTest::test_gradient_fused_gromov_wasserstein_penalty0 4.93s call tests/core/gromov_wasserstein_test.py::GromovWassersteinTest::test_gw_lr_apply0 4.93s call tests/core/fused_gromov_wasserstein_test.py::FusedGromovWassersteinTest::test_gradient_fused_gromov_wasserstein_geometry1 4.91s call tests/core/continuous_barycenter_test.py::Barycenter::test_euclidean_barycenter1 4.90s call tests/core/gromov_wasserstein_unbalanced_test.py::GromovWassersteinUnbalancedTest::test_gradient_gromov_wasserstein_geometry1 4.86s call tests/geometry/scaling_cost_test.py::ScaleCostTest::test_scale_cost_pointcloud_online1 4.73s call tests/core/continuous_barycenter_test.py::Barycenter::test_euclidean_barycenter6 4.73s call tests/core/continuous_barycenter_test.py::Barycenter::test_euclidean_barycenter5 4.72s call tests/core/sinkhorn_online_large_test.py::SinkhornOnlineTest::test_online_matches_offline_size2 4.69s call tests/core/gromov_wasserstein_test.py::GromovWassersteinTest::test_gradient_gromov_wasserstein_pointcloud1 4.67s call tests/tools/sinkhorn_divergence_test.py::SinkhornDivergenceTest::test_euclidean_autoepsilon 4.65s call tests/core/continuous_barycenter_test.py::Barycenter::test_euclidean_barycenter7 4.64s call tests/geometry/scaling_cost_test.py::ScaleCostTest::test_scale_cost_pointcloud_online0 4.62s call tests/core/sinkhorn_lr_test.py::SinkhornLRTest::test_euclidean_point_cloud2 4.58s call tests/core/sinkhorn_online_large_test.py::SinkhornOnlineTest::test_online_matches_offline_size1 4.54s call tests/core/sinkhorn_online_large_test.py::SinkhornOnlineTest::test_euclidean_point_cloud0 4.46s call tests/core/fused_gromov_wasserstein_test.py::FusedGromovWassersteinTest::test_gradient_fused_gromov_wasserstein_penalty1 4.43s call tests/core/gromov_wasserstein_test.py::GromovWassersteinTest::test_gradient_marginals_gromov_wasserstein1 4.40s call tests/tools/soft_sort_test.py::SoftSortTest::test_sort_batch0 4.34s call tests/core/continuous_barycenter_test.py::Barycenter::test_euclidean_barycenter2 4.30s call tests/core/sinkhorn_anderson_acceleration_test.py::SinkhornAndersonTest::test_anderson2 4.29s call tests/core/continuous_barycenter_test.py::Barycenter::test_euclidean_barycenter3 4.25s call tests/tools/soft_sort_test.py::SoftSortTest::test_quantize 4.22s call tests/core/fused_gromov_wasserstein_test.py::FusedGromovWassersteinTest::test_flag_store_errors_fused 4.17s call tests/core/sinkhorn_anderson_acceleration_test.py::SinkhornAndersonTest::test_anderson4 4.15s call tests/core/sinkhorn_anderson_acceleration_test.py::SinkhornAndersonTest::test_anderson3 4.13s call tests/tools/soft_sort_test.py::SoftSortTest::test_sort_with 4.10s call tests/core/gromov_wasserstein_test.py::GromovWassersteinTest::test_gradient_gromov_wasserstein_geometry1 4.08s call tests/core/sinkhorn_online_large_test.py::SinkhornOnlineTest::test_online_matches_offline_size3 4.05s call tests/core/sinkhorn_anderson_acceleration_test.py::SinkhornAndersonTest::test_anderson5 4.05s call tests/core/sinkhorn_test.py::SinkhornTest::test_apply_transport_geometry_from_scalings 4.01s call tests/core/gromov_wasserstein_test.py::GromovWassersteinTest::test_gradient_gromov_wasserstein_geometry0 3.93s call tests/core/sinkhorn_test.py::SinkhornTest::test_online_vs_batch_euclidean_point_cloud0 3.92s call tests/tools/sinkhorn_divergence_test.py::SinkhornDivergenceTest::test_euclidean_point_cloud_unbalanced_wrapper 3.88s call tests/core/gromov_wasserstein_test.py::GromovWassersteinTest::test_gw_fused_scale_cost0 3.83s call tests/core/sinkhorn_jit_test.py::SinkhornTest::test_jit_vs_non_jit_bwd0 3.78s call tests/core/sinkhorn_anderson_acceleration_test.py::SinkhornAndersonTest::test_anderson6 3.73s call tests/core/fused_gromov_wasserstein_test.py::FusedGromovWassersteinTest::test_effect_fused_penalty 3.72s call tests/core/gromov_wasserstein_test.py::GromovWassersteinTest::test_gw_fused_scale_cost2 3.64s call tests/tools/soft_sort_test.py::SoftSortTest::test_sort_batch2 3.64s call tests/tools/sinkhorn_divergence_test.py::SinkhornDivergenceTest::test_euclidean_momentum_params3 3.62s call tests/tools/soft_sort_test.py::SoftSortTest::test_topk_one_array1 3.61s call tests/core/gromov_wasserstein_test.py::GromovWassersteinTest::test_gw_fused_scale_cost1 3.59s call tests/geometry/scaling_cost_test.py::ScaleCostTest::test_scale_cost_pointcloud_online3 3.55s call tests/tools/soft_sort_test.py::SoftSortTest::test_rank_one_array 3.40s call tests/core/gromov_wasserstein_test.py::GromovWassersteinTest::test_flag_store_errors 3.40s call tests/tools/soft_sort_test.py::SoftSortTest::test_quantile0 3.39s call tests/tools/sinkhorn_divergence_test.py::SinkhornDivergenceTest::test_euclidean_momentum_params4 3.39s call tests/core/sinkhorn_grid_test.py::SinkhornGridTest::test_grid_vs_euclidean0 3.38s call tests/core/sinkhorn_online_large_test.py::SinkhornOnlineTest::test_online_external_jit 3.30s call tests/core/sinkhorn_diff_grid_loc_test.py::SinkhornGradGridTest::test_diff_sinkhorn_x_grid_x_perturbation1 3.25s call tests/core/sinkhorn_diff_test.py::SinkhornJacobianTest::test_autograd_sinkhorn0 3.25s call tests/tools/soft_sort_test.py::SoftSortTest::test_soft_quantile_normalization 3.16s call tests/tools/sinkhorn_divergence_test.py::SinkhornDivergenceTest::test_euclidean_momentum_params5 3.14s call tests/geometry/scaling_cost_test.py::ScaleCostTest::test_max_scale_cost_low_rank_large_array 3.11s call tests/core/sinkhorn_anderson_acceleration_test.py::SinkhornAndersonTest::test_anderson1 3.00s call tests/core/sinkhorn_anderson_acceleration_test.py::SinkhornAndersonTest::test_anderson7 2.96s call tests/tools/soft_sort_test.py::SoftSortTest::test_sort_batch1 2.88s call tests/tools/transport_test.py::TransportTest::test_transport_from_geom 2.82s call tests/tools/sinkhorn_divergence_test.py::SinkhornDivergenceTest::test_euclidean_momentum_params2 2.68s call tests/geometry/matrix_square_root_test.py::MatrixSquareRootTest::test_grad_test_sqrtm_sqrtm_plus_inv_sqrtm 2.65s call tests/tools/gaussian_mixture/gaussian_test.py::GaussianTest::test_w2_dist 2.63s call tests/tools/gaussian_mixture/gaussian_mixture_pair_test.py::GaussianMixturePairTest::test_get_coupling_between_same_gmm 2.56s call tests/geometry/matrix_square_root_test.py::MatrixSquareRootTest::test_solve_bartels_stewart_batch 2.56s call tests/geometry/scaling_cost_test.py::ScaleCostTest::test_scale_cost_low_rank2 2.51s call tests/geometry/scaling_cost_test.py::ScaleCostTest::test_scale_cost_low_rank0 2.48s call tests/core/sinkhorn_grid_test.py::SinkhornGridTest::test_separable_grid0 2.47s call tests/tools/gaussian_mixture/fit_gmm_test.py::FitGmmTest::test_integration 2.43s call tests/geometry/matrix_square_root_test.py::MatrixSquareRootTest::test_grad_test_sqrtm_only 2.42s call tests/tools/sinkhorn_divergence_test.py::SinkhornDivergenceTest::test_euclidean_autoepsilon_not_share_epsilon 2.38s call tests/geometry/matrix_square_root_test.py::MatrixSquareRootTest::test_grad_test_sqrtm_sqrtm 2.30s call tests/core/sinkhorn_anderson_acceleration_test.py::SinkhornAndersonTest::test_anderson10 2.29s call tests/core/sinkhorn_diff_test.py::SinkhornJacobianTest::test_autograd_sinkhorn1 2.28s call tests/geometry/geometry_lr_test.py::LRGeometryTest::test_apply_squared 2.25s call tests/tools/sinkhorn_divergence_test.py::SinkhornDivergenceTest::test_euclidean_momentum_params0 2.24s call tests/tools/sinkhorn_divergence_test.py::SinkhornDivergenceTest::test_euclidean_momentum_params1 2.23s call tests/geometry/geometry_pointcloud_apply_test.py::ApplyTest::test_apply_cost_and_kernel 2.17s call tests/core/sinkhorn_test.py::SinkhornTest::test_autoepsilon_with_decay3 2.17s call tests/tools/sinkhorn_divergence_test.py::SinkhornDivergenceTest::test_euclidean_point_cloud_wrapper_no_weights 2.16s call tests/core/gromov_wasserstein_unbalanced_test.py::GromovWassersteinUnbalancedTest::test_gromov_wasserstein_pointcloud 2.12s call tests/tools/soft_sort_test.py::SoftSortTest::test_sort_batch3 2.09s call tests/geometry/scaling_cost_test.py::ScaleCostTest::test_scale_cost_geometry0 2.08s call tests/core/sinkhorn_online_large_test.py::SinkhornOnlineTest::test_online_sinkhorn_jit 2.08s call tests/tools/gaussian_mixture/gaussian_mixture_pair_test.py::GaussianMixturePairTest::test_get_sinkhorn_to_same_gmm_is_almost_zero 2.08s call tests/core/sinkhorn_test.py::SinkhornTest::test_autoepsilon 2.07s call tests/core/gromov_wasserstein_test.py::GromovWassersteinTest::test_gw_lr_apply1 2.03s call tests/core/sinkhorn_unbalanced_test.py::SinkhornUnbalancedTest::test_euclidean_point_cloud_lse-high-mom 2.03s call tests/core/sinkhorn_diff_test.py::SinkhornJacobianTest::test_gradient_sinkhorn_euclidean_lse-implicit 2.02s call tests/tools/gaussian_mixture/gaussian_mixture_pair_test.py::GaussianMixturePairTest::test_get_sinkhorn_to_shifted_is_almost_shift 2.02s call tests/tools/sinkhorn_divergence_test.py::SinkhornDivergenceTest::test_euclidean_point_cloud_wrapper 2.01s call tests/tools/gaussian_mixture/gaussian_mixture_pair_test.py::GaussianMixturePairTest::test_get_coupling_to_shifted 1.99s call tests/core/sinkhorn_anderson_acceleration_test.py::SinkhornAndersonTest::test_anderson14 1.92s call tests/core/sinkhorn_anderson_acceleration_test.py::SinkhornAndersonTest::test_anderson15 1.89s call tests/core/sinkhorn_test.py::SinkhornTest::test_euclidean_point_cloud_lse-Leh-1 1.87s call tests/geometry/scaling_cost_test.py::ScaleCostTest::test_scale_cost_pointcloud4 1.84s call tests/core/sinkhorn_diff_grid_weights_test.py::SinkhornGradGridTest::test_diff_sinkhorn_x_grid_weights_perturbation1 1.84s call tests/geometry/scaling_cost_test.py::ScaleCostTest::test_scale_cost_low_rank1 1.82s call tests/core/sinkhorn_grid_test.py::SinkhornGridTest::test_apply_transport_grid1 1.79s call tests/core/sinkhorn_diff_test.py::SinkhornJacobianTest::test_gradient_sinkhorn_geometry1 1.79s call tests/core/sinkhorn_test.py::SinkhornTest::test_autoepsilon_with_decay2 1.78s call tests/core/sinkhorn_diff_test.py::SinkhornJacobianTest::test_gradient_sinkhorn_geometry0 1.77s call tests/core/sinkhorn_test.py::SinkhornTest::test_autoepsilon_with_decay1 1.75s call tests/geometry/geometry_lr_test.py::LRGeometryTest::test_add_lr_geoms 1.74s call tests/core/sinkhorn_test.py::SinkhornTest::test_euclidean_point_cloud_parallel_weights0 1.74s call tests/core/sinkhorn_anderson_acceleration_test.py::SinkhornAndersonTest::test_anderson12 1.73s call tests/core/sinkhorn_diff_test.py::SinkhornJacobianTest::test_autoepsilon_differentiability 1.72s call tests/geometry/geometry_lse_test.py::GeometryLseTest::test_lse 1.71s call tests/core/sinkhorn_test.py::SinkhornTest::test_geom_vs_point_cloud 1.69s call tests/core/sinkhorn_test.py::SinkhornTest::test_online_vs_batch_euclidean_point_cloud1 1.68s call tests/geometry/scaling_cost_test.py::ScaleCostTest::test_scale_cost_low_rank3 1.67s call tests/core/sinkhorn_online_large_test.py::SinkhornOnlineTest::test_euclidean_point_cloud1 1.66s call tests/core/sinkhorn_diff_test.py::SinkhornJacobianTest::test_gradient_sinkhorn_euclidean_lse-implicit-force_scan 1.65s call tests/core/discrete_barycenter_test.py::DiscreteBarycenterTest::test_discrete_barycenter_pointcloud_lse 1.64s call tests/core/fused_gromov_wasserstein_test.py::FusedGromovWassersteinTest::test_fused_gromov_wasserstein_pointcloud0 1.62s call tests/core/sinkhorn_anderson_acceleration_test.py::SinkhornAndersonTest::test_anderson11 1.61s call tests/core/icnn_test.py::ICNNTest::test_icnn_hessian0 1.61s call tests/core/gromov_wasserstein_test.py::GromovWassersteinTest::test_gromov_wasserstein_pointcloud 1.61s call tests/geometry/matrix_square_root_test.py::MatrixSquareRootTest::test_sqrtm 1.59s call tests/core/fused_gromov_wasserstein_test.py::FusedGromovWassersteinTest::test_fused_gromov_wasserstein_pointcloud1 1.58s call tests/core/sinkhorn_anderson_acceleration_test.py::SinkhornAndersonTest::test_anderson13 1.57s call tests/tools/gaussian_mixture/gaussian_mixture_test.py::GaussianMixtureTest::test_log_prob 1.57s call tests/geometry/scaling_cost_test.py::ScaleCostTest::test_scale_cost_pointcloud0 1.56s call tests/tools/transport_test.py::TransportTest::test_transport_from_point 1.53s call tests/core/sinkhorn_test.py::SinkhornTest::test_autoepsilon_with_decay0 1.52s call tests/geometry/scaling_cost_test.py::ScaleCostTest::test_scale_cost_pointcloud_online2 1.51s call tests/core/sinkhorn_jit_test.py::SinkhornTest::test_jit_vs_non_jit_fwd 1.48s call tests/geometry/scaling_cost_test.py::ScaleCostTest::test_scale_cost_pointcloud_online4 1.45s call tests/tools/soft_sort_test.py::SoftSortTest::test_topk_one_array0 1.45s call tests/core/sinkhorn_test.py::SinkhornTest::test_restart0 1.44s call tests/geometry/geometry_costs_test.py::CostFnTest::test_cosine 1.43s call tests/core/sinkhorn_bures_test.py::SinkhornTest::test_regularized_unbalanced_bures 1.42s call tests/core/sinkhorn_grid_test.py::SinkhornGridTest::test_apply_cost 1.38s call tests/tools/transport_test.py::TransportTest::test_transport_from_problem 1.35s call tests/tools/soft_sort_test.py::SoftSortTest::test_topk_one_array2 1.34s call tests/tools/gaussian_mixture/gaussian_mixture_test.py::GaussianMixtureTest::test_sample 1.25s call tests/core/sinkhorn_diff_test.py::SinkhornJacobianTest::test_gradient_sinkhorn_euclidean_scal-implicit 1.21s call tests/tools/soft_sort_test.py::SoftSortTest::test_sort_one_array1 1.20s call tests/core/sinkhorn_test.py::SinkhornTest::test_online_euclidean_point_cloud0 1.18s call tests/geometry/scaling_cost_test.py::ScaleCostTest::test_scale_cost_pointcloud2 1.18s call tests/geometry/scaling_cost_test.py::ScaleCostTest::test_scale_cost_pointcloud1 1.16s call tests/tools/gaussian_mixture/gaussian_mixture_test.py::GaussianMixtureTest::test_get_summary_stats_from_points_and_assignment_probs 1.16s call tests/core/sinkhorn_grid_test.py::SinkhornGridTest::test_grid_vs_euclidean1 1.16s call tests/geometry/geometry_lr_test.py::LRGeometryTest::test_conversion_pointcloud0 1.15s call tests/tools/gaussian_mixture/scale_tril_test.py::ScaleTriLTest::test_w2_dist 1.15s call tests/tools/gaussian_mixture/gaussian_test.py::GaussianTest::test_transport 1.11s call tests/core/sinkhorn_anderson_acceleration_test.py::SinkhornAndersonTest::test_anderson9 1.10s call tests/core/sinkhorn_anderson_acceleration_test.py::SinkhornAndersonTest::test_anderson8 1.07s call tests/geometry/scaling_cost_test.py::ScaleCostTest::test_scale_cost_pointcloud3 1.07s call tests/geometry/scaling_cost_test.py::ScaleCostTest::test_scale_cost_pointcloud5 1.06s call tests/core/discrete_barycenter_test.py::DiscreteBarycenterTest::test_discrete_barycenter_grid_scal-no-deb 1.06s call tests/core/sinkhorn_test.py::SinkhornTest::test_restart1 0.97s call tests/geometry/scaling_cost_test.py::ScaleCostTest::test_online_matches_notonline_pointcloud0 0.93s call tests/core/sinkhorn_test.py::SinkhornTest::test_euclidean_point_cloud_lse-Leh-24 0.92s call tests/tools/soft_sort_test.py::SoftSortTest::test_quantile1 0.91s call tests/geometry/matrix_square_root_test.py::MatrixSquareRootTest::test_sqrtm_batch 0.89s call tests/core/sinkhorn_lr_test.py::SinkhornLRTest::test_output_apply_batch_size0 0.85s call tests/core/sinkhorn_test.py::SinkhornTest::test_euclidean_point_cloud_parallel_weights1 0.80s call tests/tools/gaussian_mixture/gaussian_mixture_test.py::GaussianMixtureTest::test_log_component_posterior 0.76s call tests/core/sinkhorn_unbalanced_test.py::SinkhornUnbalancedTest::test_euclidean_point_cloud_lse-no-mom 0.76s call tests/core/sinkhorn_test.py::SinkhornTest::test_euclidean_point_cloud_min_iter 0.75s call tests/core/sinkhorn_lr_test.py::SinkhornLRTest::test_output_apply_batch_size1 0.74s call tests/geometry/geometry_lr_test.py::LRGeometryTest::test_apply_affine_function_efficient0 0.73s call tests/core/sinkhorn_test.py::SinkhornTest::test_euclidean_point_cloud_scal-Leh-mom 0.71s call tests/core/sinkhorn_test.py::SinkhornTest::test_euclidean_point_cloud_lse-Leh-mom 0.70s call tests/core/sinkhorn_unbalanced_test.py::SinkhornUnbalancedTest::test_euclidean_point_cloud_scal-high-mom 0.66s call tests/core/sinkhorn_grid_test.py::SinkhornGridTest::test_separable_grid1 0.65s call tests/geometry/geometry_lr_test.py::LRGeometryTest::test_conversion_pointcloud1 0.62s call tests/tools/gaussian_mixture/gaussian_test.py::GaussianTest::test_to_z 0.61s call tests/geometry/scaling_cost_test.py::ScaleCostTest::test_scale_cost_geometry3 0.58s call tests/core/sinkhorn_unbalanced_test.py::SinkhornUnbalancedTest::test_euclidean_point_cloud_scal-no-mom 0.56s call tests/tools/gaussian_mixture/gaussian_test.py::GaussianTest::test_sample 0.55s call tests/core/sinkhorn_test.py::SinkhornTest::test_online_euclidean_point_cloud1 0.55s call tests/tools/gaussian_mixture/linalg_test.py::LinalgTest::test_apply_to_diag 0.53s call tests/tools/gaussian_mixture/scale_tril_test.py::ScaleTriLTest::test_from_cholesky 0.49s call tests/tools/gaussian_mixture/scale_tril_test.py::ScaleTriLTest::test_flatten_unflatten 0.47s call tests/geometry/scaling_cost_test.py::ScaleCostTest::test_scale_cost_geometry1 0.47s call tests/tools/gaussian_mixture/scale_tril_test.py::ScaleTriLTest::test_transport 0.47s call tests/geometry/scaling_cost_test.py::ScaleCostTest::test_scale_cost_geometry2 0.44s call tests/tools/gaussian_mixture/linalg_test.py::LinalgTest::test_matrix_powers 0.44s call tests/tools/gaussian_mixture/linalg_test.py::LinalgTest::test_get_mean_and_cov 0.43s call tests/core/discrete_barycenter_test.py::DiscreteBarycenterTest::test_discrete_barycenter_pointcloud_scal 0.38s call tests/geometry/geometry_lr_test.py::LRGeometryTest::test_apply_affine_function_efficient1 0.34s call tests/tools/gaussian_mixture/gaussian_test.py::GaussianTest::test_log_prob 0.34s call tests/tools/gaussian_mixture/linalg_test.py::LinalgTest::test_flat_to_tril 0.34s call tests/geometry/scaling_cost_test.py::ScaleCostTest::test_online_matches_notonline_pointcloud1 0.34s call tests/tools/gaussian_mixture/gaussian_test.py::GaussianTest::test_flatten_unflatten 0.29s call tests/geometry/geometry_pointcloud_generalcost_test.py::ApplyTest::test_general_cost_fn 0.27s call tests/tools/gaussian_mixture/gaussian_test.py::GaussianTest::test_from_z 0.26s call tests/geometry/matrix_square_root_test.py::MatrixSquareRootTest::test_solve_bartels_stewart 0.24s call tests/tools/gaussian_mixture/scale_tril_test.py::ScaleTriLTest::test_covariance_sqrt 0.23s call tests/tools/gaussian_mixture/linalg_test.py::LinalgTest::test_invmatvectril 0.22s call tests/geometry/scaling_cost_test.py::ScaleCostTest::test_online_matches_notonline_pointcloud3 0.22s call tests/geometry/geometry_lr_test.py::LRGeometryTest::test_conversion_pointcloud2 0.22s call tests/geometry/scaling_cost_test.py::ScaleCostTest::test_max_scale_cost_low_rank_with_batch0 0.21s call tests/tools/gaussian_mixture/gaussian_mixture_pair_test.py::GaussianMixturePairTest::test_flatten_unflatten_balanced_locked 0.21s call tests/geometry/geometry_lr_test.py::LRGeometryTest::test_apply 0.13s call tests/geometry/scaling_cost_test.py::ScaleCostTest::test_max_scale_cost_low_rank_with_batch1 0.11s call tests/tools/gaussian_mixture/linalg_test.py::LinalgTest::test_get_mean_and_cov_nonuniform_weights 0.11s call tests/tools/gaussian_mixture/gaussian_mixture_pair_test.py::GaussianMixturePairTest::test_pytree_mapping_balanced_locked 0.08s call tests/tools/gaussian_mixture/probabilities_test.py::ProbabilitiesTest::test_from_probs 0.08s call tests/tools/gaussian_mixture/linalg_test.py::LinalgTest::test_get_mean_and_var 0.08s call tests/tools/gaussian_mixture/linalg_test.py::LinalgTest::test_tril_to_flat 0.05s call tests/tools/gaussian_mixture/scale_tril_test.py::ScaleTriLTest::test_log_det_covariance 0.05s call tests/tools/gaussian_mixture/gaussian_mixture_test.py::GaussianMixtureTest::test_covariance 0.05s call tests/geometry/geometry_lr_test.py::LRGeometryTest::test_conversion_pointcloud3 0.05s call tests/tools/gaussian_mixture/probabilities_test.py::ProbabilitiesTest::test_flatten_unflatten 0.05s call tests/tools/gaussian_mixture/gaussian_mixture_test.py::GaussianMixtureTest::test_from_mean_cov_component_weights 0.05s call tests/tools/gaussian_mixture/gaussian_test.py::GaussianTest::test_from_mean_and_cov 0.04s call tests/tools/gaussian_mixture/gaussian_test.py::GaussianTest::test_pytree_mapping 0.04s call tests/tools/gaussian_mixture/probabilities_test.py::ProbabilitiesTest::test_sample 0.04s call tests/tools/gaussian_mixture/gaussian_mixture_pair_test.py::GaussianMixturePairTest::test_pytree_mapping_unbalanced_locked 0.04s call tests/geometry/scaling_cost_test.py::ScaleCostTest::test_online_matches_notonline_pointcloud2 0.04s call tests/tools/gaussian_mixture/linalg_test.py::LinalgTest::test_get_mean_and_var_nonuniform_weights 0.04s call tests/geometry/geometry_lr_test.py::LRGeometryTest::test_apply_affine_function_efficient2 0.04s call tests/geometry/geometry_lr_test.py::LRGeometryTest::test_apply_affine_function_efficient3 0.04s call tests/tools/gaussian_mixture/gaussian_mixture_pair_test.py::GaussianMixturePairTest::test_pytree_mapping_unbalanced_unlocked 0.04s call tests/tools/gaussian_mixture/gaussian_mixture_pair_test.py::GaussianMixturePairTest::test_pytree_mapping_balanced_unlocked 0.04s call tests/tools/gaussian_mixture/scale_tril_test.py::ScaleTriLTest::test_pytree_mapping 0.03s call tests/tools/gaussian_mixture/gaussian_mixture_pair_test.py::GaussianMixturePairTest::test_flatten_unflatten_unbalanced_unlocked 0.03s call tests/tools/gaussian_mixture/linalg_test.py::LinalgTest::test_get_random_orthogonal 0.03s call tests/tools/gaussian_mixture/gaussian_mixture_pair_test.py::GaussianMixturePairTest::test_flatten_unflatten_balanced_unlocked 0.03s call tests/tools/gaussian_mixture/gaussian_mixture_pair_test.py::GaussianMixturePairTest::test_flatten_unflatten_unbalanced_locked 0.02s call tests/core/sinkhorn_test.py::SinkhornTest::test_autoepsilon_with_decay4 0.02s call tests/geometry/geometry_pointcloud_apply_test.py::ApplyTest::test_shape_with_jnp_ndarrays 0.02s call tests/tools/gaussian_mixture/gaussian_mixture_test.py::GaussianMixtureTest::test_flatten_unflatten 0.02s call tests/tools/gaussian_mixture/gaussian_mixture_test.py::GaussianMixtureTest::test_from_random 0.02s call tests/core/sinkhorn_test.py::SinkhornTest::test_autoepsilon_with_decay6 0.02s call tests/core/sinkhorn_test.py::SinkhornTest::test_autoepsilon_with_decay5 0.02s call tests/core/sinkhorn_test.py::SinkhornTest::test_autoepsilon_with_decay7 0.02s call tests/tools/gaussian_mixture/probabilities_test.py::ProbabilitiesTest::test_log_probs 0.01s call tests/tools/gaussian_mixture/gaussian_mixture_test.py::GaussianMixtureTest::test_pytree_mapping 0.01s call tests/tools/gaussian_mixture/gaussian_test.py::GaussianTest::test_from_random 0.01s call tests/tools/gaussian_mixture/scale_tril_test.py::ScaleTriLTest::test_from_random 0.01s call tests/tools/gaussian_mixture/scale_tril_test.py::ScaleTriLTest::test_cholesky 0.01s call tests/tools/gaussian_mixture/scale_tril_test.py::ScaleTriLTest::test_covariance (888 durations < 0.005s hidden. Use -vv to show these durations.) ================= 441 passed, 1 warning in 3513.96s (0:58:33) ==================