Fix task deprecation endpoint

materialsproject · Jun 1, 2021 · 0c5c4af · 0c5c4af
1 parent 045f3ad
commit 0c5c4af
Showing 1 changed file with 18 additions and 4 deletions.
diff --git a/src/mp_api/routes/tasks/query_operators.py b/src/mp_api/routes/tasks/query_operators.py
@@ -12,7 +12,10 @@ class MultipleTaskIDsQuery(QueryOperator):
     """
 
     def query(
-        self, task_ids: Optional[str] = Query(None, description="Comma-separated list of task_ids to query on"),
+        self,
+        task_ids: Optional[str] = Query(
+            None, description="Comma-separated list of task_ids to query on"
+        ),
     ) -> STORE_PARAMS:
 
         crit = {}
@@ -29,7 +32,10 @@ class TrajectoryQuery(QueryOperator):
     """
 
     def query(
-        self, task_ids: Optional[str] = Query(None, description="Comma-separated list of task_ids to query on"),
+        self,
+        task_ids: Optional[str] = Query(
+            None, description="Comma-separated list of task_ids to query on"
+        ),
     ) -> STORE_PARAMS:
 
         crit = {}
@@ -45,7 +51,12 @@ def post_process(self, docs):
         """
 
         d = [
-            {"task_id": doc["task_id"], "trajectories": jsanitize(calcs_reversed_to_trajectory(doc["calcs_reversed"])),}
+            {
+                "task_id": doc["task_id"],
+                "trajectories": jsanitize(
+                    calcs_reversed_to_trajectory(doc["calcs_reversed"])
+                ),
+            }
             for doc in docs
         ]
 
@@ -58,7 +69,10 @@ class DeprecationQuery(QueryOperator):
     """
 
     def query(
-        self, task_ids: str = Query(None, description="Comma-separated list of task_ids to query on"),
+        self,
+        task_ids: str = Query(
+            ..., description="Comma-separated list of task_ids to query on"
+        ),
     ) -> STORE_PARAMS:
 
         self.task_ids = task_ids.split(",")