We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 309d783 commit 9d7c5d9Copy full SHA for 9d7c5d9
1 file changed
src/geist/datastore/clingo.py
@@ -38,7 +38,12 @@ def format_dicts(dicts, returnformat):
38
def filter_by_predicate(dicts, predicate):
39
filtered_dicts = []
40
for one_case in dicts:
41
- filtered_dict = {predicate: one_case[predicate]} if predicate in one_case else {predicate: []}
+ if type(predicate) == str:
42
+ filtered_dict = {predicate: one_case.get(predicate, [])}
43
+ elif type(predicate) == list:
44
+ filtered_dict = {p: one_case.get(p, []) for p in predicate}
45
+ else:
46
+ raise ValueError(f"Unsupported predicate type: {type(predicate)}")
47
filtered_dicts.append(filtered_dict)
48
return filtered_dicts
49
0 commit comments