From 7a0537afdd74e2e362f69af20b6799ca1e684bb4 Mon Sep 17 00:00:00 2001
From: Theodore Chang <theodore.chang@physik.hu-berlin.de>
Date: Tue, 3 Sep 2024 16:24:17 +0000
Subject: [PATCH] Serialize enum using its value
---
nomad/metainfo/metainfo.py | 1 +
tests/datamodel/test_schema.py | 2 +-
2 files changed, 2 insertions(+), 1 deletion(-)
diff --git a/nomad/metainfo/metainfo.py b/nomad/metainfo/metainfo.py
index 5e384bb67d..92d0de6e32 100644
--- a/nomad/metainfo/metainfo.py
+++ b/nomad/metainfo/metainfo.py
@@ -4680,6 +4680,7 @@ class AnnotationModel(Annotation, BaseModel):
validate_assignment = True
arbitrary_types_allowed = True
+ use_enum_values = True
AnnotationModel.update_forward_refs()
diff --git a/tests/datamodel/test_schema.py b/tests/datamodel/test_schema.py
index d67ee3b0a2..17ccdea3ab 100644
--- a/tests/datamodel/test_schema.py
+++ b/tests/datamodel/test_schema.py
@@ -125,7 +125,7 @@ def test_eln_annotation_validation(eln_type, eln_component):
assert isinstance(exception.value, MetainfoError)
error_str = (
- f'The component {eln_component} '
+ f'The component {eln_component.value} '
f'is not compatible with the quantity quantity_name of the type {type_name}. '
f'Accepted components: {", ".join(valid_eln_components[eln_type])}'
)
--
GitLab