File tree 1 file changed +5
-2
lines changed
tests/unit/sagemaker/monitor
1 file changed +5
-2
lines changed Original file line number Diff line number Diff line change @@ -568,11 +568,12 @@ def test_clarify_model_monitor():
568
568
569
569
# The subclass should has monitoring_type() defined
570
570
# noinspection PyAbstractClass
571
- class DummyClarifyModelMonitoir (ClarifyModelMonitor ):
571
+ class DummyClarifyModelMonitor (ClarifyModelMonitor ):
572
+ _TEST_CLASS = True
572
573
pass
573
574
574
575
with pytest .raises (TypeError ):
575
- DummyClarifyModelMonitoir .monitoring_type ()
576
+ DummyClarifyModelMonitor .monitoring_type ()
576
577
577
578
578
579
def test_clarify_model_monitor_invalid_update (clarify_model_monitors ):
@@ -593,6 +594,8 @@ def test_clarify_model_monitor_invalid_attach(sagemaker_session):
593
594
)
594
595
# attach, invalid monitoring type
595
596
for clarify_model_monitor_cls in ClarifyModelMonitor .__subclasses__ ():
597
+ if hasattr (clarify_model_monitor_cls , "_TEST_CLASS" ):
598
+ continue
596
599
with pytest .raises (TypeError ):
597
600
clarify_model_monitor_cls .attach (SCHEDULE_NAME , sagemaker_session )
598
601
You can’t perform that action at this time.
0 commit comments