diff --git a/api/py/ai/chronon/repo/run.py b/api/py/ai/chronon/repo/run.py index 02b0b291d..cc3bf8c3b 100755 --- a/api/py/ai/chronon/repo/run.py +++ b/api/py/ai/chronon/repo/run.py @@ -343,9 +343,8 @@ def __init__(self, args, jar_path): ) else: self.conf_type = args.conf_type - self.ds = args.end_ds if args.end_ds else args.ds - self.start_ds = args.start_ds - self.parallelism = args.parallelism + self.ds = args.end_ds if hasattr(args, 'end_ds') else args.ds + self.parallelism = args.parallelism if hasattr(args, 'parallelism') else 1 self.jar_path = jar_path self.args = args.args if args.args else "" self.online_class = args.online_class @@ -475,8 +474,8 @@ def set_defaults(parser): "--app-name", help="app name. Default to {}".format(APP_NAME_TEMPLATE) ) parser.add_argument( - "--start-ds", help="overwrite the original start partition for a range backfill. It only supports staging query, " - "group by backfill and join jobs" + "--start-ds", help="overwrite the original start partition for a range backfill. " + "It only supports staging query, group by backfill and join jobs" ) parser.add_argument( "--end-ds", help="the end ds for a range backfill"