added keras suffix

microsoft · Mar 13, 2024 · b36845f · b36845f
1 parent 0a3173a
commit b36845f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/responsibleai_vision/responsibleai_vision/rai_vision_insights/rai_vision_insights.py b/responsibleai_vision/responsibleai_vision/rai_vision_insights/rai_vision_insights.py
@@ -767,7 +767,7 @@ def save(self, path):
         :param path: The directory path to save the RAIInsights to.
         :type path: str
         """
-        super(RAIVisionInsights, self).save(path)
+        super(RAIVisionInsights, self).save(path.with_suffix(".keras"))
         # Save extracted features data
         self._save_ext_data(path)
         self._save_transformations(path)