diff --git a/src/mp_api/routes/tasks/query_operators.py b/src/mp_api/routes/tasks/query_operators.py index 1917c815..d5ec1f4d 100644 --- 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(",")