Fix desc sorting in query op

materialsproject · Jun 11, 2024 · 524bd97 · 524bd97
1 parent 786c331
commit 524bd97
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/src/maggma/api/query_operator/sorting.py b/src/maggma/api/query_operator/sorting.py
@@ -42,12 +42,11 @@ def query(
                 )
 
             for sort_field in field_list:
+                query_entry = {sort_field[1:]: -1} if sort_field.startswith("-") else {sort_field: 1}
+
                 if self.fields and sort_field not in self.fields:
                     continue
 
-                if sort_field[0] == "-":
-                    sort.update({sort_field[1:]: -1})
-                else:
-                    sort.update({sort_field: 1})
+                sort.update(query_entry)
 
         return {"sort": sort}