add version to argparse

ycrc · Mar 15, 2023 · e28f702 · e28f702
1 parent 09c8df2
commit e28f702
Showing 1 changed file with 1 addition and 6 deletions.
diff --git a/seff-array.py b/seff-array.py
@@ -179,12 +179,7 @@ def job_eff(job_id=0, cluster=os.getenv('SLURM_CLUSTER_NAME')):
     )
     parser.add_argument("jobid")
     parser.add_argument("-c", "--cluster", action="store", dest="cluster")
-#    parser.add_argument("-f", "--figures", action="store_true", dest=figures, help="display figures of histograms")
-    #parser.add_argument("-m", "--mem", action="store_true", dest="m", help="show memory usage stats")
-    #parser.add_argument("-t", "--time", action="store_true", dest="t", help="show time usage stats")
-    #parser.add_argument("-c", "--cpu", action="store_true", dest="c", help="show cpu usage stats")
+    parser.add_argument('--version', action='version',  version='%(prog)s {version}'.format(version=__version__))
     args = parser.parse_args()
-    #if not (args.m or args.t or args.c):
-    #    args.m, args.t, args.c = True, True, True
 
     job_eff(args.jobid, args.cluster)