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)