G3-226 add publication return model

TheJacksonLaboratory · May 1, 2024 · a48b463 · a48b463
1 parent 97db411
commit a48b463
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/geneweaver/api/controller/publications.py b/src/geneweaver/api/controller/publications.py
@@ -4,7 +4,7 @@
 
 from fastapi import APIRouter, Depends, HTTPException, Path, Security
 from geneweaver.api import dependencies as deps
-from geneweaver.api.schemas.apimodels import NewPubMedRecord
+from geneweaver.api.schemas.apimodels import NewPubmedRecord
 from geneweaver.api.schemas.auth import UserInternal
 from geneweaver.api.services import publications as publication_service
 from geneweaver.core.schema.publication import Publication
@@ -44,7 +44,7 @@ def add_publication(
     ],
     user: UserInternal = Security(deps.full_user),
     cursor: Optional[deps.Cursor] = Depends(deps.cursor),
-) -> NewPubMedRecord:
+) -> NewPubmedRecord:
     """Add pubmed publication endpoint."""
     response = publication_service.add_pubmed_record(
         cursor=cursor, user=user, pubmed_id=publication_id

diff --git a/src/geneweaver/api/schemas/apimodels.py b/src/geneweaver/api/schemas/apimodels.py
@@ -84,7 +84,7 @@ class GeneValueReturn(BaseModel):
     data: List[GeneValueSchema]
 
 
-class NewPubMedRecord(BaseModel):
+class NewPubmedRecord(BaseModel):
     """Model returned for adding new pubmed info into DB."""
 
     pub_id: int