from math import sqrt def alpha_shapes_diagrams(points): f = Filtration() fill_alpha_complex(points, f) f.sort(dim_data_cmp) p = StaticPersistence(f) p.pair_simplices() dgms = init_diagrams(p, f, lambda s: sqrt(s.data[0])) return dgms