From ffe6ec69672c20295ea9fb4378b90f08c6c8fab5 Mon Sep 17 00:00:00 2001
From: Pengyu <3771747+better365@users.noreply.github.com>
Date: Thu, 9 Nov 2023 10:29:42 -0800
Subject: [PATCH] fix flake8 and test

---
 api/py/ai/chronon/repo/run.py | 9 ++++-----
 1 file changed, 4 insertions(+), 5 deletions(-)

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"