test fixes

microsoft · Mar 13, 2024 · 014f771 · 014f771
1 parent b36845f
commit 014f771
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
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.with_suffix(".keras"))
+        super(RAIVisionInsights, self).save(path)
         # Save extracted features data
         self._save_ext_data(path)
         self._save_transformations(path)

diff --git a/responsibleai_vision/tests/common_vision_utils.py b/responsibleai_vision/tests/common_vision_utils.py
@@ -514,7 +514,7 @@ def __call__(self, X):
         return self.model(tmp)
 
     def save(self, path):
-        self.model.save(path)
+        self.model.save(path + '.keras')
 
     @staticmethod
     def load(path):