Skip to content

Commit

Permalink
Merge pull request #2377 from devitocodes/filter_ordered
Browse files Browse the repository at this point in the history
dsl: Correct retention of wrong branch in filter_ordered
  • Loading branch information
mloubout authored May 28, 2024
2 parents 779aa09 + 5368999 commit ffd8ee0
Showing 1 changed file with 2 additions and 13 deletions.
15 changes: 2 additions & 13 deletions devito/tools/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -169,21 +169,10 @@ def filter_ordered(elements, key=None):
if isinstance(elements, types.GeneratorType):
elements = list(elements)

seen = set()
if key is None:
try:
unordered, inds = np.unique(elements, return_index=True)
return unordered[np.argsort(inds)].tolist()
except:
return sorted(list(set(elements)), key=elements.index)
return list(dict.fromkeys(elements))
else:
ret = []
for e in elements:
k = key(e)
if k not in seen:
ret.append(e)
seen.add(k)
return ret
return list(dict(zip([key(i) for i in elements], elements)).values())


def filter_sorted(elements, key=None):
Expand Down

0 comments on commit ffd8ee0

Please sign in to comment.