null check

airbnb · Nov 9, 2023 · 21f94a8 · 21f94a8
1 parent ffe6ec6
commit 21f94a8
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/api/py/ai/chronon/repo/run.py b/api/py/ai/chronon/repo/run.py
@@ -343,8 +343,8 @@ def __init__(self, args, jar_path):
             )
         else:
             self.conf_type = args.conf_type
-        self.ds = args.end_ds if hasattr(args, 'end_ds') else args.ds
-        self.parallelism = args.parallelism if hasattr(args, 'parallelism') else 1
+        self.ds = args.end_ds if hasattr(args, 'end_ds') and args.end_ds else args.ds
+        self.parallelism = args.parallelism if hasattr(args, 'parallelism') and args.parallelism else 1
         self.jar_path = jar_path
         self.args = args.args if args.args else ""
         self.online_class = args.online_class
@@ -474,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"