changes ie api

rnugmanx · Oct 27, 2023 · b1ab9b9 · b1ab9b9
1 parent c59c81f
commit b1ab9b9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/e2e_oss/pipelines/pipeline_templates/infer_templates.py b/tests/e2e_oss/pipelines/pipeline_templates/infer_templates.py
@@ -1,7 +1,7 @@
 def common_infer_step(device, batch, api_2, input_file_path=None, skip_ir_generation=False, **additional_args):
     if api_2:
         # return 'reshape' method for setting batch because old 'batch_size' method still not implemented in new API
-        infer_api = {"ie_sync": {"device": device,
+        infer_api = {"ie_sync_api_2": {"device": device,
                                        "network_modifiers": {"set_batch_using_reshape_api_2": {"batch": batch}},
                                        **additional_args}}
     elif not api_2 and skip_ir_generation: