mirrors/mim
1
0
Fork 0
mirror of https://github.com/open-mmlab/mim.git synced 2025-06-03 14:59:11 +08:00
Code Issues Projects Releases Wiki Activity

[Enhancement] Support full match (#79)

Browse Source
This commit is contained in:
Zaida Zhou 2021-09-07 21:49:25 +08:00 committed by GitHub
parent c156e060ee
commit dd97fdf226
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 24 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
mim/commands/search.py
View File

@ -564,6 +564,9 @@ def sort_by(dataframe: DataFrame,
matched_fields = [] matched_fields = []
invalid_fields = set() invalid_fields = set()
for input_field in input_fields: for input_field in input_fields:
if any(valid_fields.isin([input_field])):
matched_fields.append(input_field)
else:
contain_index = valid_fields.str.contains(input_field) contain_index = valid_fields.str.contains(input_field)
contain_fields = valid_fields[contain_index] contain_fields = valid_fields[contain_index]
if len(contain_fields) == 1: if len(contain_fields) == 1:
@ -620,6 +623,9 @@ def select_by(dataframe: DataFrame,
# not consistent with the input_fields # not consistent with the input_fields
seen_fields = set() seen_fields = set()
for input_field in input_fields: for input_field in input_fields:
if any(valid_fields.isin([input_field])):
matched_fields.append(input_field)
else:
contain_index = valid_fields.str.contains(input_field) contain_index = valid_fields.str.contains(input_field)
contain_fields = valid_fields[contain_index] contain_fields = valid_fields[contain_index]
if len(contain_fields) > 0: if len(contain_fields) > 0: