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(