Fix typos.

faustomorales · Jul 25, 2022 · 7faae52 · 7faae52
1 parent f535adc
commit 7faae52
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/mira/detectors/assets/serve/object_detector.py b/mira/detectors/assets/serve/object_detector.py
@@ -39,7 +39,7 @@ def handle(self, data, context):
         return output
 
     def postprocess(self, data, thresholds):
-        if API_MODE == "torch":
+        if API_MODE == "torchserve":
             return super().postprocess(data)
         return [
             [

diff --git a/mira/detectors/detector.py b/mira/detectors/detector.py
@@ -360,7 +360,7 @@ def to_torchserve(
         archive_format: tx.Literal["default", "no-archive"] = "default",
         score_threshold: float = 0.5,
         model_version="1.0",
-        api_mode: tx.Literal["mira", "torchserver"] = "mira",
+        api_mode: tx.Literal["mira", "torchserve"] = "mira",
     ):
         """Build a TorchServe-compatible MAR file for this model."""
         assert (