TaskRester primary key fix

materialsproject · May 7, 2021 · 0b85a2c · 0b85a2c
1 parent 0d063b5
commit 0b85a2c
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/mp_api/routes/tasks/client.py b/src/mp_api/routes/tasks/client.py
@@ -9,6 +9,7 @@ class TaskRester(BaseRester):
 
     suffix = "tasks"
     document_model = TaskDoc
+    primary_key = "task_id"
 
     def get_trajectory(self, task_id):
         """