diff --git a/tests/unit/test_arcaflow_plugin_opensearch.py b/tests/unit/test_arcaflow_plugin_opensearch.py index 9a4310a..26d61eb 100644 --- a/tests/unit/test_arcaflow_plugin_opensearch.py +++ b/tests/unit/test_arcaflow_plugin_opensearch.py @@ -2,7 +2,11 @@ import unittest import opensearch_plugin -from opensearch_schema import Operation, BulkUploadObject, BulkUploadOperationMeta +from opensearch_schema import ( + # Operation, + BulkUploadObject, + BulkUploadOperationMeta, +) from arcaflow_plugin_sdk import plugin @@ -23,7 +27,8 @@ def test_serialization(): BulkUploadObject( operation={ # Operation.INDEX: BulkUploadOperationMeta( - # Temporarily changing the key to a string to work around a problem + # Temporarily changing the key to a string in order to work + # around a workflow validation failure for the enum as a key "index": BulkUploadOperationMeta( _index="myotherindex", _id="abc123", @@ -37,7 +42,8 @@ def test_serialization(): BulkUploadObject( operation={ # Operation.CREATE: BulkUploadOperationMeta(), - # Temporarily changing the key to a string to work around a problem + # Temporarily changing the key to a string in order to work + # around a workflow validation failure for the enum as a key "create": BulkUploadOperationMeta(), }, data={