diff --git a/backlogger/find.py b/backlogger/find.py index b3f7996..7591c75 100644 --- a/backlogger/find.py +++ b/backlogger/find.py @@ -38,7 +38,6 @@ def _db_lookup(db, ensemble, correlator_name, project=None, code=None, parameter search_expr += f" AND updated_at < '{updated_before}'" if updated_after: search_expr += f" AND updated_at > '{updated_after}'" - print(search_expr) conn = sqlite3.connect(db) results = pd.read_sql(search_expr, conn) conn.close() @@ -52,9 +51,7 @@ def filter_results(results, **kwargs): if result['code'] == 'sfcf': param = json.loads(result['parameters']) if 'offset' in kwargs: - print("checking offset") if kwargs.get('offset') != param['offset']: - print("dropping") drops.append(ind) continue if 'quark_masses' in kwargs: @@ -81,11 +78,11 @@ def filter_results(results, **kwargs): return results.drop(drops) -def find_correlator(path, ensemble, correlator_name, project=None, code=None, parameters=None, created_before=None, created_after=None, updated_before=None, updated_after=None, revision=None, **kwargs): +def find_record(path, ensemble, correlator_name, project=None, code=None, parameters=None, created_before=None, created_after=None, updated_before=None, updated_after=None, revision=None, **kwargs): db = path + '/backlogger.db' if os.path.exists(db): dl.get(db, dataset=path) results = _db_lookup(db, ensemble, correlator_name, project, code=None, parameters=None, created_before=created_before, created_after=created_after, updated_before=updated_before, updated_after=updated_after, revision=revision) results = filter_results(results, **kwargs) print("Found " + str(len(results)) + " results") - return results + return results.reset_index()