feat(components): Add nargs to allow for empty string input by component

PiperOrigin-RevId: 462727038
kubeflow · Jul 22, 2022 · 555447f · 555447f
1 parent d3fe514
commit 555447f
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/...le_cloud_pipeline_components/container/experimental/evaluation/import_model_evaluation.py b/...le_cloud_pipeline_components/container/experimental/evaluation/import_model_evaluation.py
@@ -62,13 +62,13 @@ def _make_parent_dirs_and_return_path(file_path: str):
     required=True,
     default=argparse.SUPPRESS)
 parser.add_argument(
-    '--display_name', dest='display_name', type=str, default=None)
+    '--display_name', nargs='?', dest='display_name', type=str, default=None)
 parser.add_argument(
     '--pipeline_job_id', dest='pipeline_job_id', type=str, default=None)
 parser.add_argument(
-    '--dataset_path', dest='dataset_path', type=str, default=None)
+    '--dataset_path', nargs='?', dest='dataset_path', type=str, default=None)
 parser.add_argument(
-    '--dataset_type', dest='dataset_type', type=str, default=None)
+    '--dataset_type', nargs='?', dest='dataset_type', type=str, default=None)
 parser.add_argument(
     '--model_name',
     dest='model_name',