wasserstein distance return type float (#386)

scverse · Oct 1, 2023 · 3bd9e56 · 3bd9e56
1 parent a625208
commit 3bd9e56
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pertpy/tools/_distances/_distances.py b/pertpy/tools/_distances/_distances.py
@@ -363,7 +363,7 @@ def solve_ot_problem(self, geom: Geometry, **kwargs):
         solver = Sinkhorn()
         # Solve OT problem
         ot = solver(ot_prob, **kwargs)
-        return ot.reg_ot_cost
+        return ot.reg_ot_cost.item()
 
 
 class PseudobulkDistance(AbstractDistance):