Add beam_search parser as parent

k2-fsa · Jul 25, 2022 · e9ec2aa · e9ec2aa
1 parent 9d4d1c8
commit e9ec2aa
Show file tree

Hide file tree

Showing 5 changed files with 12 additions and 6 deletions.
diff --git a/sherpa/bin/conv_emformer_transducer_stateless2/streaming_server.py b/sherpa/bin/conv_emformer_transducer_stateless2/streaming_server.py
@@ -49,6 +49,7 @@
 def get_args():
     beam_search_parser = add_beam_search_arguments()
     parser = argparse.ArgumentParser(
+        parents=[beam_search_parser],
         formatter_class=argparse.ArgumentDefaultsHelpFormatter,
     )
 
@@ -172,7 +173,7 @@ def get_args():
     )
 
     return (
-        parser.parse_known_args()[0],
+        parser.parse_args(),
         beam_search_parser.parse_known_args()[0],
     )
 

diff --git a/sherpa/bin/pruned_stateless_emformer_rnnt2/streaming_server.py b/sherpa/bin/pruned_stateless_emformer_rnnt2/streaming_server.py
@@ -49,6 +49,7 @@
 def get_args():
     beam_search_parser = add_beam_search_arguments()
     parser = argparse.ArgumentParser(
+        parents=[beam_search_parser],
         formatter_class=argparse.ArgumentDefaultsHelpFormatter,
     )
 
@@ -172,7 +173,7 @@ def get_args():
     )
 
     return (
-        parser.parse_known_args()[0],
+        parser.parse_args(),
         beam_search_parser.parse_known_args()[0],
     )
 

diff --git a/sherpa/bin/pruned_transducer_statelessX/offline_server.py b/sherpa/bin/pruned_transducer_statelessX/offline_server.py
@@ -47,7 +47,8 @@
 def get_args():
     beam_search_parser = add_beam_search_arguments()
     parser = argparse.ArgumentParser(
-        formatter_class=argparse.ArgumentDefaultsHelpFormatter
+        parents=[beam_search_parser],
+        formatter_class=argparse.ArgumentDefaultsHelpFormatter,
     )
 
     parser.add_argument(
@@ -169,7 +170,7 @@ def get_args():
     )
 
     return (
-        parser.parse_known_args()[0],
+        parser.parse_args(),
         beam_search_parser.parse_known_args()[0],
     )
 

diff --git a/sherpa/bin/streaming_pruned_transducer_statelessX/streaming_server.py b/sherpa/bin/streaming_pruned_transducer_statelessX/streaming_server.py
@@ -51,6 +51,7 @@
 def get_args():
     beam_search_parser = add_beam_search_arguments()
     parser = argparse.ArgumentParser(
+        parents=[beam_search_parser],
         formatter_class=argparse.ArgumentDefaultsHelpFormatter,
     )
 
@@ -174,7 +175,7 @@ def get_args():
     )
 
     return (
-        parser.parse_known_args()[0],
+        parser.parse_args(),
         beam_search_parser.parse_known_args()[0],
     )
 

diff --git a/sherpa/python/sherpa/utils.py b/sherpa/python/sherpa/utils.py
@@ -2,7 +2,9 @@
 
 
 def add_beam_search_arguments():
-    parser = argparse.ArgumentParser(description="Parameters for beam search")
+    parser = argparse.ArgumentParser(
+        description="Parameters for beam search", add_help=False
+    )
 
     parser.add_argument(
         "--decoding-method",