feat: update schema in example ML backend. comments

Author: mihow

ami/ml/models/pipeline.py

@@ -289,6 +289,7 @@ def save_results(results: PipelineResponse | None = None, results_json: str | No
 
             assert classification.algorithm, "No classification algorithm was specified in the returned results."
             classification_algo, _created = Algorithm.objects.get_or_create(
+                # @TODO Algorithm needs to be an object with a slug/key, name and category map
                 name=classification.algorithm,
             )
             if _created:

ml_backends/example/api/schemas.py

@@ -69,6 +69,7 @@ class Classification(pydantic.BaseModel):
     classification: str
     labels: list[str] = []
     scores: list[float] = []
+    logits: list[float] | None = None
     inference_time: float | None = None
     algorithm: str
     terminal: bool = True