[partition] Improves partition error message (#826)

deepjavalibrary · Jun 10, 2023 · 1c1426e · 1c1426e
1 parent eabaeab
commit 1c1426e
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/serving/docker/partition/properties_manager.py b/serving/docker/partition/properties_manager.py
@@ -140,12 +140,14 @@ def set_and_validate_entry_point(self):
                     self.properties['option.entryPoint'] = 'model.py'
                 else:
                     engine = self.properties.get('engine')
-                    if engine == "DeepSpeed":
+                    if engine is None:
+                        raise ValueError("Please specify engine")
+                    elif engine.lower() == "deepspeed":
                         entry_point = "djl_python.deepspeed"
-                    elif engine == "FasterTransformer":
+                    elif engine.lower() == "fastertransformer":
                         entry_point = "djl_python.fastertransformer"
                     else:
-                        raise ValueError("Please specify engine")
+                        raise ValueError(f"Invalid engine: {engine}")
                     self.properties['option.entryPoint'] = entry_point
         elif entry_point.lower().startswith('http'):
             logging.info(f'Downloading entrypoint file.')