Update cli.py

t5s/cli.py

@@ -3,30 +3,50 @@ import os
 import subprocess
 import sys
 
-arg_parser = argparse.ArgumentParser(description="T5 Summarisation Using Pytorch Lightning",
-                                     prog="t5s")
+arg_parser = argparse.ArgumentParser(
+    description="T5 Summarisation Using Pytorch Lightning", prog="t5s"
+)
 # Command choice
-command_subparser = arg_parser.add_subparsers(dest="command", help="command (refer commands section in documentation)")
+command_subparser = arg_parser.add_subparsers(
+    dest="command", help="command (refer commands section in documentation)"
+)
 
-parser_req = command_subparser.add_parser('requirements', help='Install Python Dependencies.')
+parser_req = command_subparser.add_parser(
+    "requirements", help="Install Python Dependencies."
+)
 
-parser_dirs = command_subparser.add_parser('dirs', help='Create directories that are ignored by git but required for '
-                                                        'the project')
+parser_dirs = command_subparser.add_parser(
+    "dirs",
+    help="Create directories that are ignored by git but required for " "the project",
+)
 
-parser_push = command_subparser.add_parser('push', help='Upload Data to default DVC remote')
+parser_push = command_subparser.add_parser(
+    "push", help="Upload Data to default DVC remote"
+)
 
-parser_pull = command_subparser.add_parser('pull', help="Download Data from default DVC remote")
+parser_pull = command_subparser.add_parser(
+    "pull", help="Download Data from default DVC remote"
+)
 
-parser_run = command_subparser.add_parser('run', help="run the DVC pipeline - recompute any modified outputs such as "
-                                                      "processed data or trained models")
+parser_run = command_subparser.add_parser(
+    "run",
+    help="run the DVC pipeline - recompute any modified outputs such as "
+    "processed data or trained models",
+)
 
-parser_visualize = command_subparser.add_parser('visualize', help="run the visualization using Streamlit")
+parser_visualize = command_subparser.add_parser(
+    "visualize", help="run the visualization using Streamlit"
+)
 
-parser_upload = command_subparser.add_parser('upload', help="push the trained model to HF model hub")
+parser_upload = command_subparser.add_parser(
+    "upload", help="push the trained model to HF model hub"
+)
 
-parser_lint = command_subparser.add_parser('lint', help=" Lint using flake8")
+parser_lint = command_subparser.add_parser("lint", help=" Lint using flake8")
 
-parser_clone = command_subparser.add_parser('clone', help="Clone the T5 summarisation repo")
+parser_clone = command_subparser.add_parser(
+    "clone", help="Clone the T5 summarisation repo"
+)
 
 
 class Run(object):
@@ -37,15 +57,26 @@ class Run(object):
         arguments = self.arguments
         print(f"arguments passed: {arguments['command']}")
         # os.chdir('../')
-        cmd = ['requirements', 'dirs', 'push', 'pull', 'run', 'visualize', 'upload', 'lint']
-        if arguments['command'] == "clone":
-            list_files = subprocess.run(["git", "clone", "https://dagshub.com/gagan3012/summarization.git"])
-            os.chdir('./summarization/')
+        cmd = [
+            "requirements",
+            "dirs",
+            "push",
+            "pull",
+            "run",
+            "visualize",
+            "upload",
+            "lint",
+        ]
+        if arguments["command"] == "clone":
+            list_files = subprocess.run(
+                ["git", "clone", "https://dagshub.com/gagan3012/summarization.git"]
+            )
+            os.chdir("./summarization/")
             retval = os.getcwd()
             print(retval)
             return list_files.returncode
-        elif arguments['command'] in cmd:
-            os.chdir('./summarization/')
+        elif arguments["command"] in cmd:
+            os.chdir("./summarization/")
             retval = os.getcwd()
             print(retval)
             list_files = subprocess.run(["make", arguments["command"]])