From de854360d894260dc860f220e0714a113c413f38 Mon Sep 17 00:00:00 2001 From: kkaris Date: Wed, 25 Oct 2023 14:08:16 -0700 Subject: [PATCH] Handle NaN for mapping_set_version and mapping_set_license --- src/semra/io.py | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/src/semra/io.py b/src/semra/io.py index 0e2bc8d..5ae0f73 100644 --- a/src/semra/io.py +++ b/src/semra/io.py @@ -264,13 +264,18 @@ def _parse_sssom_row(row, mapping_set_name=None) -> Mapping: else: n = mapping_set_name confidence = None + mapping_set_version = None + mapping_set_license = None if "mapping_set_confidence" in row and pd.notna(row["mapping_set_confidence"]): confidence = row["mapping_set_confidence"] - + if "mapping_set_version" in row and pd.notna(row["mapping_set_version"]): + mapping_set_version = row["mapping_set_version"] + if "mapping_set_license" in row and pd.notna(row["mapping_set_license"]): + mapping_set_license = row["mapping_set_license"] mapping_set = MappingSet( name=n, - version=row.get("mapping_set_version"), - license=row.get("mapping_set_license"), + version=mapping_set_version, + license=mapping_set_license, confidence=confidence, ) if "mapping_justification" in row and pd.notna(row["mapping_justification"]):