[Multi model deployment] Updating usages for container config (#1144)

Author: kumar-shivam-ranjan

ads/aqua/config/container_config.py

@@ -15,6 +15,7 @@ class Usage(ExtendedEnum):
     INFERENCE = "inference"
     BATCH_INFERENCE = "batch_inference"
     MULTI_MODEL = "multi_model"
+    OTHER = "other"
 
 
 class AquaContainerConfigSpec(Serializable):

ads/aqua/model/model.py

@@ -270,7 +270,10 @@ def create_multi(
         supported_container_families = [
             container_config_item.family
             for container_config_item in service_inference_containers
-            if Usage.MULTI_MODEL in container_config_item.usages
+            if any(
+                usage in container_config_item.usages
+                for usage in [Usage.MULTI_MODEL, Usage.OTHER]
+            )
         ]
 
         if not supported_container_families:

ads/aqua/modeldeployment/deployment.py

@@ -237,7 +237,10 @@ def create(
             supported_container_families = [
                 container_config_item.family
                 for container_config_item in service_inference_containers
-                if Usage.MULTI_MODEL in container_config_item.usages
+                if any(
+                    usage in container_config_item.usages
+                    for usage in [Usage.MULTI_MODEL, Usage.OTHER]
+                )
             ]
 
             if not supported_container_families:

tests/unitary/with_extras/aqua/test_data/ui/container_index.json

@@ -131,6 +131,8 @@
       ],
       "type": "inference",
       "usages": [
+        "inference",
+        "batch_inference",
         "multi_model"
       ],
       "version": "0.4.1.3"

tests/unitary/with_extras/aqua/test_ui.py

@@ -595,7 +595,7 @@ def test_list_containers(self, mock_get_container_config):
                         ],
                         "server_port": "8080",
                     },
-                    "usages": ["multi_model"],
+                    "usages": ["inference", "batch_inference", "multi_model"],
                 },
             ],
             "finetune": [