Skip to content

Commit d335cc9

Browse files
committed
fix: improve CLI behavior of gen-op-dataset
1 parent 89c3f6d commit d335cc9

1 file changed

Lines changed: 16 additions & 3 deletions

File tree

developer/gen-op-dataset.py

Lines changed: 16 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22

33
import argparse
44
import re
5+
import sys
56
from abc import ABC, abstractmethod
67

78
import numpy as np
@@ -236,12 +237,24 @@ def main_sims() -> None:
236237

237238

238239
if __name__ == "__main__":
239-
parser = argparse.ArgumentParser()
240-
parser.add_argument("--exps", action="store_true", help="Run experiments")
241-
parser.add_argument("--sims", action="store_true", help="Run simulations")
240+
parser = argparse.ArgumentParser(
241+
prog="OP Dataset Generator",
242+
description="""
243+
CI helper script for delivering dataset for Kaggle in HDF5 format.
244+
Two DataFrames are stored for each Sim/Exp-lipid pair:
245+
1. SMILES-aligned values for each atom of the molecule
246+
2. Composition table of membrane part of the system""",
247+
formatter_class=argparse.RawDescriptionHelpFormatter,
248+
)
249+
parser.add_argument("--exps", action="store_true", help="Generate DS from experiments")
250+
parser.add_argument("--sims", action="store_true", help="Generate DS from simulations")
242251
args = parser.parse_args()
243252

244253
if args.exps:
245254
main_exps()
246255
if args.sims:
247256
main_sims()
257+
258+
if not args.exps and not args.sims:
259+
parser.print_usage()
260+
sys.exit(1)

0 commit comments

Comments
 (0)