release 0.5.3

nlp_architect/cli/__init__.py

@@ -18,38 +18,53 @@
 
 # register all procedures by importing
 import nlp_architect.procedures  # noqa: F401
-from nlp_architect.cli.cli_commands import cli_train_cmd, cli_run_cmd
+from nlp_architect.cli.cmd_registry import CMD_REGISTRY
 from nlp_architect.version import NLP_ARCHITECT_VERSION
 
 logging.basicConfig(format="%(asctime)s : %(levelname)s : %(message)s", level=logging.INFO)
 
 
-def run_cli():
-    """ Run nlp_architect command line application
-    """
-    prog_name = "nlp_architect"
-    desc = "NLP Architect CLI [{}]".format(NLP_ARCHITECT_VERSION)
+def nlp_train_cli():
+    prog_name = "nlp-train"
+    desc = "NLP Architect Train CLI [{}]".format(NLP_ARCHITECT_VERSION)
     parser = argparse.ArgumentParser(description=desc, prog=prog_name)
     parser.add_argument(
         "-v", "--version", action="version", version="%(prog)s v{}".format(NLP_ARCHITECT_VERSION)
     )
-
     parser.set_defaults(func=lambda _: parser.print_help())
-    subparsers = parser.add_subparsers(title="commands", metavar="")
-    for command in sub_commands:
-        command(subparsers)
+    subparsers = parser.add_subparsers(title="Models", metavar="")
+    for model in CMD_REGISTRY["train"]:
+        sp = subparsers.add_parser(
+            model["name"], description=model["description"], help=model["description"]
+        )
+        model["arg_adder"](sp)
+        sp.set_defaults(func=model["fn"])
+
     args = parser.parse_args()
     if hasattr(args, "func"):
         args.func(args)
     else:
         parser.print_help()
 
 
-# sub commands list
-sub_commands = [
-    cli_train_cmd,
-    cli_run_cmd,
-]
+def nlp_inference_cli():
+    prog_name = "nlp-inference"
+    desc = "NLP Architect Inference CLI [{}]".format(NLP_ARCHITECT_VERSION)
+    parser = argparse.ArgumentParser(description=desc, prog=prog_name)
+    parser.add_argument(
+        "-v", "--version", action="version", version="%(prog)s v{}".format(NLP_ARCHITECT_VERSION)
+    )
+    parser.set_defaults(func=lambda _: parser.print_help())
+    subparsers = parser.add_subparsers(title="Models", metavar="")
+    for model in CMD_REGISTRY["inference"]:
+        sp = subparsers.add_parser(
+            model["name"], description=model["description"], help=model["description"]
+        )
+        model["arg_adder"](sp)
+        sp.set_defaults(func=model["fn"])
 
-if __name__ == "__main__":
-    run_cli()
+    args = parser.parse_args()
+    if hasattr(args, "func"):
+        args.func(args)
+    else:
+        parser.print_help()

nlp_architect/version.py

@@ -14,7 +14,7 @@
 # limitations under the License.
 # ******************************************************************************
 
-MAJOR_V, MINOR_V, PATCH_V, STAGE = 0, 5, 2, ""
+MAJOR_V, MINOR_V, PATCH_V, STAGE = 0, 5, 3, ""
 
 
 def nlp_architect_version():

setup.py

@@ -80,7 +80,7 @@
     license="Apache 2.0",
     python_requires=">=3.6.*",
     packages=find_packages(
-        exclude=["tests.*", "tests", "server.*", "server", "examples.*", "examples"]
+        exclude=["tests.*", "tests", "server.*", "server", "examples.*", "examples", "solutions.*", "solutions"]
     ),
     install_requires=install_requirements,
     extras_require=extras,