[#345] Fix model selector

legion-platform · Oct 8, 2018 · 58cac7a · 58cac7a
1 parent dd0586c
commit 58cac7a
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/legion/legion/k8s/enclave.py b/legion/legion/k8s/enclave.py
@@ -211,8 +211,13 @@ def get_models_strict(self, model_id, model_version=None, ignore_not_found=False
         """
         model_services = self.get_models(model_id, model_version)
 
-        if len(model_services) > 1 and not model_version:
-            raise Exception('Please specify version of model')
+        ignore_strictness = model_id == '*' or model_version == '*'
+
+        if len(model_services) > 1:
+            LOGGER.info('More than one model was found for filter: id={!r} version={!r}'
+                        .format(model_id, model_version))
+            if not ignore_strictness and not model_version:
+                raise Exception('Please specify version of model')
 
         if not model_services:
             if not ignore_not_found: