[PY] Fix bad rebase

Signed-off-by: julian <[email protected]>
giotto-ai · Aug 17, 2022 · 31015b0 · 31015b0
1 parent 38e9246
commit 31015b0
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/gph/python/ripser_interface.py b/gph/python/ripser_interface.py
@@ -34,14 +34,15 @@
 MAX_COEFF_SUPPORTED = gph_ripser.get_max_coefficient_field_supported()
 
 
-def _compute_ph_vr_dense(DParam, maxHomDim, thresh=-1, coeff=2, n_threads=1,
-                         return_generators=False, return_cocycles=False):
+def _compute_ph_vr_dense(DParam, diagonal, maxHomDim, thresh=-1, coeff=2,
+                         n_threads=1, return_generators=False,
+                         return_cocycles=False):
     if coeff == 2:
-        ret = gph_ripser.rips_dm(DParam, DParam.shape[0], coeff,
+        ret = gph_ripser.rips_dm(DParam, diagonal, coeff,
                                  maxHomDim, thresh, n_threads,
                                  return_generators, return_cocycles)
     else:
-        ret = gph_ripser_coeff.rips_dm(DParam, DParam.shape[0], coeff,
+        ret = gph_ripser_coeff.rips_dm(DParam, diagonal, coeff,
                                        maxHomDim, thresh, n_threads,
                                        return_generators, return_cocycles)
     return ret