Fix test

sony · Dec 22, 2024 · 7bb7891 · 7bb7891
1 parent 11d8535
commit 7bb7891
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/keras_tests/non_parallel_tests/test_keras_tp_model.py b/tests/keras_tests/non_parallel_tests/test_keras_tp_model.py
@@ -231,6 +231,7 @@ def test_opset_not_in_tp(self):
                                         tpc_minor_version=None,
                                         tpc_patch_version=None,
                                         tpc_platform_type=None,
+                                        operator_set=tuple([schema.OperatorsSet("opA")]),
                                         add_metadata=False)
         hm_keras = tp.TargetPlatformCapabilities(hm)
         with self.assertRaises(Exception) as e: