Patch the escape for filter field name

marqo-ai · Jan 7, 2024 · 562bb24 · 562bb24
1 parent 9b5dc00
commit 562bb24
Showing 1 changed file with 31 additions and 31 deletions.
diff --git a/tests/cloud_test_logic/cloud_test_index.py b/tests/cloud_test_logic/cloud_test_index.py
@@ -96,46 +96,46 @@ class CloudTestIndex(str, Enum):
     #     "inferenceType": "marqo.GPU",
     #     "storageClass": "marqo.balanced",
     # },
-    CloudTestIndex.structured_image_custom: {
-        "type": "structured",
-        "treatUrlsAndPointersAsImages": True,
-        "model": "test-image-model",
-        "modelProperties": {
-            "name": "ViT-B-32-quickgelu",
-            "dimensions": 512,
-            "url": "https://github.com/mlfoundations/open_clip/releases/download/v0.2-weights/vit_b_32-quickgelu-laion400m_avg-8a00ab3c.pt",
-            "type": "open_clip",
-        },
-        "allFields": [
-            {"name": "text_field_1", "type": "text", "features": ["lexical_search", "filter"]},
-            {"name": "text_field_2", "type": "text", "features": ["filter"]},
-            {"name": "image_field_1", "type": "image_pointer"},
-            {"name": "array_field_1", "type": "array<text>", "features": ["filter"]},
-            {"name": "float_field_1", "type": "float", "features": ["filter", "score_modifier"]},
-            {"name": "int_field_1", "type": "int", "features": ["filter", "score_modifier"]},
-            {"name": "bool_field_1", "type": "bool", "features": ["filter"]},
-        ],
-        "tensorFields": ["text_field_1", "image_field_1", "text_field_2"],
-
-        "inferenceType": "marqo.CPU.large",
-        "numberOfInferences": 2,
-    },
-    # CloudTestIndex.structured_text: {
+    # CloudTestIndex.structured_image_custom: {
     #     "type": "structured",
-    #     "treatUrlsAndPointersAsImages": False,
-    #     "model": "hf/all_datasets_v4_MiniLM-L6",
+    #     "treatUrlsAndPointersAsImages": True,
+    #     "model": "test-image-model",
+    #     "modelProperties": {
+    #         "name": "ViT-B-32-quickgelu",
+    #         "dimensions": 512,
+    #         "url": "https://github.com/mlfoundations/open_clip/releases/download/v0.2-weights/vit_b_32-quickgelu-laion400m_avg-8a00ab3c.pt",
+    #         "type": "open_clip",
+    #     },
     #     "allFields": [
     #         {"name": "text_field_1", "type": "text", "features": ["lexical_search", "filter"]},
     #         {"name": "text_field_2", "type": "text", "features": ["filter"]},
-    #         {"name": "text_field_3", "type": "text", "features": ["lexical_search"]},
+    #         {"name": "image_field_1", "type": "image_pointer"},
     #         {"name": "array_field_1", "type": "array<text>", "features": ["filter"]},
     #         {"name": "float_field_1", "type": "float", "features": ["filter", "score_modifier"]},
     #         {"name": "int_field_1", "type": "int", "features": ["filter", "score_modifier"]},
     #         {"name": "bool_field_1", "type": "bool", "features": ["filter"]},
     #     ],
-    #     "tensorFields": ["text_field_1", "text_field_2", "text_field_3"],
+    #     "tensorFields": ["text_field_1", "image_field_1", "text_field_2"],
     #
-    #     "storageClass": "marqo.balanced",
-    #     "numberOfShards": 2,
+    #     "inferenceType": "marqo.CPU.large",
+    #     "numberOfInferences": 2,
     # },
+    CloudTestIndex.structured_text: {
+        "type": "structured",
+        "treatUrlsAndPointersAsImages": False,
+        "model": "hf/all_datasets_v4_MiniLM-L6",
+        "allFields": [
+            {"name": "text_field_1", "type": "text", "features": ["lexical_search", "filter"]},
+            {"name": "text_field_2", "type": "text", "features": ["filter"]},
+            {"name": "text_field_3", "type": "text", "features": ["lexical_search"]},
+            {"name": "array_field_1", "type": "array<text>", "features": ["filter"]},
+            {"name": "float_field_1", "type": "float", "features": ["filter", "score_modifier"]},
+            {"name": "int_field_1", "type": "int", "features": ["filter", "score_modifier"]},
+            {"name": "bool_field_1", "type": "bool", "features": ["filter"]},
+        ],
+        "tensorFields": ["text_field_1", "text_field_2", "text_field_3"],
+
+        "storageClass": "marqo.balanced",
+        "numberOfShards": 2,
+    },
 }