From 4d1ea92b198bad10eb1f01c4130098154d78cf2c Mon Sep 17 00:00:00 2001 From: Justus Kuhlmann Date: Wed, 16 Apr 2025 15:34:41 +0000 Subject: [PATCH] make param serach a bit fuzzier --- corrlib/find.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/corrlib/find.py b/corrlib/find.py index 0daf9f3..1d6f93d 100644 --- a/corrlib/find.py +++ b/corrlib/find.py @@ -3,6 +3,7 @@ import datalad.api as dl import os import json import pandas as pd +import numpy as np # this will implement the search functionality @@ -76,12 +77,12 @@ def filter_results(results, **kwargs): # careful, this is not save, when multiple contributions are present! if 'wf1' in kwargs: wf1 = kwargs['wf1'] - if (wf1[0][0] != param['wf1'][0][0]) or (wf1[0][1][0] != param['wf1'][0][1][0]) or (wf1[0][1][1] != param['wf1'][0][1][1]): + if not (np.isclose(wf1[0][0], param['wf1'][0][0], 1e-8) and np.isclose(wf1[0][1][0], param['wf1'][0][1][0], 1e-8) and np.isclose(wf1[0][1][1], param['wf1'][0][1][1], 1e-8)): drops.append(ind) continue if 'wf2' in kwargs: wf2 = kwargs['wf2'] - if (wf2[0][0] != param['wf2'][0][0]) or (wf2[0][1][0] != param['wf2'][0][1][0]) or (wf2[0][1][1] != param['wf2'][0][1][1]): + if not (np.isclose(wf2[0][0], param['wf2'][0][0], 1e-8) and np.isclose(wf2[0][1][0], param['wf2'][0][1][0], 1e-8) and np.isclose(wf2[0][1][1], param['wf2'][0][1][1], 1e-8)): drops.append(ind) continue