From ae1c232a3be18e8a07ab4ab35448cf50472db8e7 Mon Sep 17 00:00:00 2001 From: Erick Benitez-Ramos Date: Wed, 26 Mar 2025 20:53:59 -0700 Subject: [PATCH] fix: fix flaky clarify model monitor test --- tests/unit/sagemaker/monitor/test_clarify_model_monitor.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/tests/unit/sagemaker/monitor/test_clarify_model_monitor.py b/tests/unit/sagemaker/monitor/test_clarify_model_monitor.py index 53119e532a..026e1a2d54 100644 --- a/tests/unit/sagemaker/monitor/test_clarify_model_monitor.py +++ b/tests/unit/sagemaker/monitor/test_clarify_model_monitor.py @@ -568,11 +568,12 @@ def test_clarify_model_monitor(): # The subclass should has monitoring_type() defined # noinspection PyAbstractClass - class DummyClarifyModelMonitoir(ClarifyModelMonitor): + class DummyClarifyModelMonitor(ClarifyModelMonitor): + _TEST_CLASS = True pass with pytest.raises(TypeError): - DummyClarifyModelMonitoir.monitoring_type() + DummyClarifyModelMonitor.monitoring_type() def test_clarify_model_monitor_invalid_update(clarify_model_monitors): @@ -593,6 +594,8 @@ def test_clarify_model_monitor_invalid_attach(sagemaker_session): ) # attach, invalid monitoring type for clarify_model_monitor_cls in ClarifyModelMonitor.__subclasses__(): + if hasattr(clarify_model_monitor_cls, "_TEST_CLASS"): + continue with pytest.raises(TypeError): clarify_model_monitor_cls.attach(SCHEDULE_NAME, sagemaker_session)