From 0a3173a2e46cb18673a772beef92ee57a03f001a Mon Sep 17 00:00:00 2001 From: Advitya Gemawat Date: Tue, 12 Mar 2024 15:50:34 -0700 Subject: [PATCH] other gate test fixes --- responsibleai_vision/tests/rai_vision_insights_validator.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/responsibleai_vision/tests/rai_vision_insights_validator.py b/responsibleai_vision/tests/rai_vision_insights_validator.py index a0832706d9..b3366ce1bb 100644 --- a/responsibleai_vision/tests/rai_vision_insights_validator.py +++ b/responsibleai_vision/tests/rai_vision_insights_validator.py @@ -91,6 +91,7 @@ def run_and_validate_serialization( image_downloader=image_downloader) # Save it + save_1 = save_1.with_suffix(".keras") rai_insights.save(save_1) assert len(os.listdir(save_1 / ManagerNames.EXPLAINER)) == 0 assert not os.path.exists(save_1 / DATA / TRAIN_JSON) @@ -109,6 +110,7 @@ def run_and_validate_serialization( rai_2.get_data() # Save again + save_2 = save_2.with_suffix(".keras") rai_2.save(save_2) assert len(os.listdir(save_2 / ManagerNames.EXPLAINER)) == 0 assert not os.path.exists(save_2 / DATA / TRAIN_JSON)