From d08ae39582d2662b03d5fce5fe4471bbdb03bbb5 Mon Sep 17 00:00:00 2001 From: Ilya Matiach Date: Fri, 16 Dec 2022 12:36:17 -0500 Subject: [PATCH] fix loading older versions of RAIInsights due to _FEATURE_METADATA key error --- responsibleai/responsibleai/rai_insights/rai_insights.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/responsibleai/responsibleai/rai_insights/rai_insights.py b/responsibleai/responsibleai/rai_insights/rai_insights.py index 08fdf5f960..65b31c28ca 100644 --- a/responsibleai/responsibleai/rai_insights/rai_insights.py +++ b/responsibleai/responsibleai/rai_insights/rai_insights.py @@ -884,7 +884,7 @@ def _load_metadata(inst, path): inst.__dict__['_' + _FEATURE_COLUMNS] = meta[_FEATURE_COLUMNS] inst.__dict__['_' + _FEATURE_RANGES] = meta[_FEATURE_RANGES] - if meta[_FEATURE_METADATA] is None: + if _FEATURE_METADATA not in meta or meta[_FEATURE_METADATA] is None: inst.__dict__['_' + _FEATURE_METADATA] = None else: inst.__dict__['_' + _FEATURE_METADATA] = FeatureMetadata(