diff --git a/netbone/compare.py b/netbone/compare.py index 847ddfcf06ed5859b5753644af5f5640566ed6f8..5d59704768004e91c9433749843735cea2189e65 100644 --- a/netbone/compare.py +++ b/netbone/compare.py @@ -107,12 +107,12 @@ class Compare: extracted_backbone = self.filter(backbone, value=self.filter_values[i], narrate=False) values1 = method(extracted_backbone) dist[backbone.name] = cumulative_dist(name, extracted_backbone.name, values1, increasing) - vals.append(kstest(values0, values1)) + vals.append(kstest(values0, values1)[0]) ks_statistics = pd.DataFrame(index=['Original'] + [backbone.name for backbone in self.backbones]) ks_statistics['KS Statistic'] = vals - return ks_statistics.T, dist + return ks_statistics, dist