Electrodes now uses numeric query

materialsproject · May 28, 2021 · 4d05347 · 4d05347
1 parent 5b3cdd5
commit 4d05347
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/src/mp_api/routes/electrodes/resources.py b/src/mp_api/routes/electrodes/resources.py
@@ -1,3 +1,4 @@
+from maggma.api.query_operator.dynamic import NumericQuery
 from maggma.api.resource import ReadOnlyResource
 from emmet.core.electrode import InsertionElectrodeDoc
 
@@ -18,12 +19,12 @@ def insertion_electrodes_resource(insertion_electrodes_store):
         query_operators=[
             ElectrodeFormulaQuery(),
             ElementsQuery(),
-            VoltageStepQuery(),
-            InsertionVoltageStepQuery(),
-            InsertionElectrodeQuery(),
+            NumericQuery(model=InsertionElectrodeDoc),
             SortQuery(),
             PaginationQuery(),
-            SparseFieldsQuery(InsertionElectrodeDoc, default_fields=["battery_id", "last_updated"],),
+            SparseFieldsQuery(
+                InsertionElectrodeDoc, default_fields=["battery_id", "last_updated"],
+            ),
         ],
         tags=["Electrodes"],
     )