fix console_script and add meeteval-der

fgnt · Dec 5, 2023 · 6900193 · 6900193
1 parent fbf754d
commit 6900193
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 4 deletions.
diff --git a/meeteval/der/__main__.py b/meeteval/der/__main__.py
@@ -1,6 +1,5 @@
 
-
-if __name__ == '__main__':
+def cli():
     from meeteval.wer.__main__ import CLI
     from meeteval.der.md_eval import md_eval_22
 
@@ -9,3 +8,7 @@
     cli.add_command(md_eval_22)
 
     cli.run()
+
+
+if __name__ == '__main__':
+    cli()
diff --git a/meeteval/wer/__main__.py b/meeteval/wer/__main__.py
@@ -540,15 +540,21 @@ def run(self):
         self.parser.print_help()
 
 
-if __name__ == '__main__':
+def cli():
+
     cli = CLI()
 
     cli.add_command(wer)
     cli.add_command(cpwer)
+    cli.add_command(dicpwer)
     cli.add_command(orcwer)
     cli.add_command(mimower)
     cli.add_command(tcpwer)
     cli.add_command(merge)
     cli.add_command(average)
 
     cli.run()
+
+
+if __name__ == '__main__':
+    cli()
diff --git a/setup.py b/setup.py
@@ -115,7 +115,8 @@
     package_data={'meeteval': ['**/*.pyx', '**/*.h']},  # https://stackoverflow.com/a/60751886
     entry_points={
         'console_scripts': [
-            'meeteval-wer=meeteval.wer.__main__:cli'
+            'meeteval-wer=meeteval.wer.__main__:cli',
+            'meeteval-der=meeteval.der.__main__:cli',
         ]
     }
 )