Update according to 'main' function change

mvafin · May 13, 2021 · 3d03f02 · 3d03f02
1 parent 7390672
commit 3d03f02
Show file tree

Hide file tree

Showing 6 changed files with 6 additions and 6 deletions.
diff --git a/model-optimizer/mo/main.py b/model-optimizer/mo/main.py
@@ -451,4 +451,4 @@ def main(cli_parser: argparse.ArgumentParser, fem, framework: str):
 
 if __name__ == "__main__":
     from mo.utils.cli_parser import get_all_cli_parser
-    sys.exit(main(get_all_cli_parser(), None))
+    sys.exit(main(*get_all_cli_parser(), None))
diff --git a/model-optimizer/mo/main_caffe.py b/model-optimizer/mo/main_caffe.py
@@ -7,4 +7,4 @@
 
 if __name__ == "__main__":
     from mo.main import main
-    sys.exit(main(get_caffe_cli_parser(), 'caffe'))
+    sys.exit(main(get_caffe_cli_parser(), None, 'caffe'))
diff --git a/model-optimizer/mo/main_kaldi.py b/model-optimizer/mo/main_kaldi.py
@@ -7,4 +7,4 @@
 
 if __name__ == "__main__":
     from mo.main import main
-    sys.exit(main(get_kaldi_cli_parser(), 'kaldi'))
+    sys.exit(main(get_kaldi_cli_parser(), None, 'kaldi'))
diff --git a/model-optimizer/mo/main_mxnet.py b/model-optimizer/mo/main_mxnet.py
@@ -7,4 +7,4 @@
 
 if __name__ == "__main__":
     from mo.main import main
-    sys.exit(main(get_mxnet_cli_parser(), 'mxnet'))
+    sys.exit(main(get_mxnet_cli_parser(), None, 'mxnet'))
diff --git a/model-optimizer/mo/main_onnx.py b/model-optimizer/mo/main_onnx.py
@@ -7,4 +7,4 @@
 
 if __name__ == "__main__":
     from mo.main import main
-    sys.exit(main(get_onnx_cli_parser(), 'onnx'))
+    sys.exit(main(get_onnx_cli_parser(), None, 'onnx'))
diff --git a/model-optimizer/mo/main_tf.py b/model-optimizer/mo/main_tf.py
@@ -7,4 +7,4 @@
 
 if __name__ == "__main__":
     from mo.main import main
-    sys.exit(main(get_tf_cli_parser(), 'tf'))
+    sys.exit(main(get_tf_cli_parser(), None, 'tf'))