@@ -120,39 +120,39 @@ def test_get_model_graph(self, mocked_tensorflow_graph_reader: MagicMock) -> Non
120120
121121 def test_domain_object_detection_domain (self ) -> None :
122122 """Test getting domain of a model."""
123- self .assert_model_domain_matches_expected (
123+ self .assert_model_domain_matches_expected ( # pylint: disable=E1120
124124 node_names = ["boxes" , "scores" , "classes" ],
125125 expected_domain = Domains .OBJECT_DETECTION .value ,
126126 expected_domain_flavour = DomainFlavours .NONE .value ,
127127 )
128128
129129 def test_domain_object_detection_domain_ssd (self ) -> None :
130130 """Test getting domain of a model."""
131- self .assert_model_domain_matches_expected (
131+ self .assert_model_domain_matches_expected ( # pylint: disable=E1120
132132 node_names = ["bboxes" , "scores" , "classes" , "ssd" ],
133133 expected_domain = Domains .OBJECT_DETECTION .value ,
134134 expected_domain_flavour = DomainFlavours .SSD .value ,
135135 )
136136
137137 def test_domain_object_detection_domain_yolo (self ) -> None :
138138 """Test getting domain of a model."""
139- self .assert_model_domain_matches_expected (
139+ self .assert_model_domain_matches_expected ( # pylint: disable=E1120
140140 node_names = ["boxes" , "yolo" ],
141141 expected_domain = Domains .OBJECT_DETECTION .value ,
142142 expected_domain_flavour = DomainFlavours .YOLO .value ,
143143 )
144144
145145 def test_domain_image_recognition_resnet (self ) -> None :
146146 """Test getting domain of a model."""
147- self .assert_model_domain_matches_expected (
147+ self .assert_model_domain_matches_expected ( # pylint: disable=E1120
148148 node_names = ["resnet_model/Pad" ],
149149 expected_domain = Domains .IMAGE_RECOGNITION .value ,
150150 expected_domain_flavour = DomainFlavours .NONE .value ,
151151 )
152152
153153 def test_domain_unknown (self ) -> None :
154154 """Test getting domain of a model."""
155- self .assert_model_domain_matches_expected (
155+ self .assert_model_domain_matches_expected ( # pylint: disable=E1120
156156 node_names = ["foo" , "bar" , "baz" , "ssd" ],
157157 expected_domain = "" ,
158158 expected_domain_flavour = "" ,
0 commit comments