Similarity: re-key to material_id

FYI @munrojm
materialsproject · Jul 24, 2021 · 8b252c5 · 8b252c5
1 parent ab5f536
commit 8b252c5
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/app.py b/app.py
@@ -127,7 +127,7 @@
     similarity_store = MongoURIStore(
         uri=f"mongodb+srv://{db_uri}",
         database="mp_core",
-        key="task_id",
+        key="material_id",
         collection_name="similarity",
     )
 

diff --git a/src/mp_api/routes/similarity/client.py b/src/mp_api/routes/similarity/client.py
@@ -6,4 +6,4 @@ class SimilarityRester(BaseRester):
 
     suffix = "similarity"
     document_model = SimilarityDoc  # type: ignore
-    primary_key = "task_id"
+    primary_key = "material_id"
diff --git a/src/mp_api/routes/similarity/models.py b/src/mp_api/routes/similarity/models.py
@@ -36,7 +36,7 @@ class SimilarityDoc(BaseModel):
         description="List containing similar structure data for a given material.",
     )
 
-    task_id: str = Field(
+    material_id: str = Field(
         None,
         description="The Materials Project ID for the material. This comes in the form: mp-******",
     )
diff --git a/src/mp_api/routes/similarity/resources.py b/src/mp_api/routes/similarity/resources.py
@@ -9,7 +9,7 @@ def similarity_resource(similarity_store):
         SimilarityDoc,
         query_operators=[
             PaginationQuery(),
-            SparseFieldsQuery(SimilarityDoc, default_fields=["task_id"]),
+            SparseFieldsQuery(SimilarityDoc, default_fields=["material_id"]),
         ],
         tags=["Similarity"],
         enable_default_search=False,