@@ -120,39 +120,39 @@ def test_get_model_graph(self, mocked_tensorflow_graph_reader: MagicMock) -> Non
120
120
121
121
def test_domain_object_detection_domain (self ) -> None :
122
122
"""Test getting domain of a model."""
123
- self .assert_model_domain_matches_expected (
123
+ self .assert_model_domain_matches_expected ( # pylint: disable=E1120
124
124
node_names = ["boxes" , "scores" , "classes" ],
125
125
expected_domain = Domains .OBJECT_DETECTION .value ,
126
126
expected_domain_flavour = DomainFlavours .NONE .value ,
127
127
)
128
128
129
129
def test_domain_object_detection_domain_ssd (self ) -> None :
130
130
"""Test getting domain of a model."""
131
- self .assert_model_domain_matches_expected (
131
+ self .assert_model_domain_matches_expected ( # pylint: disable=E1120
132
132
node_names = ["bboxes" , "scores" , "classes" , "ssd" ],
133
133
expected_domain = Domains .OBJECT_DETECTION .value ,
134
134
expected_domain_flavour = DomainFlavours .SSD .value ,
135
135
)
136
136
137
137
def test_domain_object_detection_domain_yolo (self ) -> None :
138
138
"""Test getting domain of a model."""
139
- self .assert_model_domain_matches_expected (
139
+ self .assert_model_domain_matches_expected ( # pylint: disable=E1120
140
140
node_names = ["boxes" , "yolo" ],
141
141
expected_domain = Domains .OBJECT_DETECTION .value ,
142
142
expected_domain_flavour = DomainFlavours .YOLO .value ,
143
143
)
144
144
145
145
def test_domain_image_recognition_resnet (self ) -> None :
146
146
"""Test getting domain of a model."""
147
- self .assert_model_domain_matches_expected (
147
+ self .assert_model_domain_matches_expected ( # pylint: disable=E1120
148
148
node_names = ["resnet_model/Pad" ],
149
149
expected_domain = Domains .IMAGE_RECOGNITION .value ,
150
150
expected_domain_flavour = DomainFlavours .NONE .value ,
151
151
)
152
152
153
153
def test_domain_unknown (self ) -> None :
154
154
"""Test getting domain of a model."""
155
- self .assert_model_domain_matches_expected (
155
+ self .assert_model_domain_matches_expected ( # pylint: disable=E1120
156
156
node_names = ["foo" , "bar" , "baz" , "ssd" ],
157
157
expected_domain = "" ,
158
158
expected_domain_flavour = "" ,
0 commit comments