Merge pull request #2 from yudhiesh/fix-spelling

fix spelling issue
yudhiesh · Oct 16, 2024 · adf3bba · adf3bba
2 parents 6906ff5 + 8166034
commit adf3bba
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/project/src/data_models.py b/project/src/data_models.py
@@ -5,6 +5,7 @@
 # NOTE: If you're using a different model ensure that you add in the Results and ModelResponse
 # Pydantic models below!
 
+
 class SimpleModelRequest(BaseModel):
     review: str
 
@@ -20,7 +21,7 @@ def process_labels(cls, data: dict[int, float]) -> dict[str, float]:
         return {LABEL_CLASS_TO_NAME[key]: value for key, value in data.items()}
 
 
-class SimpleModelRespone(BaseModel):
+class SimpleModelResponse(BaseModel):
     label: SentimentLabel
     score: float
 

diff --git a/project/src/server.py b/project/src/server.py
@@ -2,7 +2,7 @@
 from ray import serve
 from ray.serve.handle import DeploymentHandle
 
-from src.data_models import SimpleModelRequest, SimpleModelRespone, SimpleModelResults
+from src.data_models import SimpleModelRequest, SimpleModelResponse, SimpleModelResults
 from src.model import Model
 
 app = FastAPI(
@@ -28,7 +28,7 @@ def __init__(self, simple_model_handle: DeploymentHandle) -> None:
     async def predict(self, request: SimpleModelRequest):
         # TODO: Use the handle.predict which is a remote function
         # to get the result
-        return SimpleModelRespone.model_validate(result.model_dump())
+        return SimpleModelResponse.model_validate(result.model_dump())
 
 
 @serve.deployment(