changing bulkuploadobject operation dict key temporarily to a string

arcalot · Jan 16, 2024 · 70c15fb · 70c15fb
1 parent 70b5963
commit 70c15fb
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 4 deletions.
diff --git a/arcaflow_plugin_opensearch/opensearch_schema.py b/arcaflow_plugin_opensearch/opensearch_schema.py
@@ -31,7 +31,10 @@ class BulkUploadOperationMeta:
 @dataclass
 class BulkUploadObject:
     operation: typing.Annotated[
-        typing.Dict[Operation, BulkUploadOperationMeta],
+        # typing.Dict[Operation, BulkUploadOperationMeta],
+        # Temporarily changing the key to a string in order to work
+        # around a workflow validation failure for the enum as a key
+        typing.Dict[str, BulkUploadOperationMeta],
         schema.name("operation"),
         schema.description("The operation type and associated operation metadata."),
     ]

diff --git a/inputs/opensearch_example.yaml b/inputs/opensearch_example.yaml
@@ -1,5 +1,5 @@
 # Default Docker network IP
-url: "http://172.16.0.203:9200"
+url: "http://localhost:9200"
 # username: "elastic"
 # password: "topsecret"
 default_index: "myindex"

diff --git a/tests/unit/test_arcaflow_plugin_opensearch.py b/tests/unit/test_arcaflow_plugin_opensearch.py
@@ -22,7 +22,9 @@ def test_serialization():
                 bulk_upload_list=[
                     BulkUploadObject(
                         operation={
-                            Operation.INDEX: BulkUploadOperationMeta(
+                            # Operation.INDEX: BulkUploadOperationMeta(
+                            # Temporarily changing the key to a string to work around a problem
+                            "index": BulkUploadOperationMeta(
                                 _index="myotherindex",
                                 _id="abc123",
                             ),
@@ -34,7 +36,9 @@ def test_serialization():
                     ),
                     BulkUploadObject(
                         operation={
-                            Operation.CREATE: BulkUploadOperationMeta(),
+                            # Operation.CREATE: BulkUploadOperationMeta(),
+                            # Temporarily changing the key to a string to work around a problem
+                            "create": BulkUploadOperationMeta(),
                         },
                         data={
                             "key1": "item 2 data value 1",