add errors

Signed-off-by: Yee Hing Tong <[email protected]>
flyteorg · Sep 25, 2024 · 18bc033 · 18bc033
1 parent 4f3332e
commit 18bc033
Show file tree

Hide file tree

Showing 3 changed files with 23 additions and 0 deletions.
diff --git a/flytekit/clis/sdk_in_container/package.py b/flytekit/clis/sdk_in_container/package.py
@@ -1,4 +1,5 @@
 import os
+import sys
 import typing
 
 import rich_click as click
@@ -131,6 +132,13 @@ def package(
     """
     # Ensure that the two flags are consistent
     if fast:
+        if "--copy" in sys.argv:
+            raise click.BadParameter(
+                click.style(
+                    "Cannot use both --fast and --copy flags together. Please move to --copy",
+                    fg="red",
+                )
+            )
         click.secho("The --fast flag is deprecated, please use --copy all instead", fg="yellow")
         copy = CopyFileDetection.ALL
 

diff --git a/flytekit/clis/sdk_in_container/register.py b/flytekit/clis/sdk_in_container/register.py
@@ -1,4 +1,5 @@
 import os
+import sys
 import typing
 
 import rich_click as click
@@ -169,6 +170,13 @@ def register(
     # that the copy flag uses.
     if non_fast:
         click.secho("The --non-fast flag is deprecated, please use --copy none instead", fg="yellow")
+        if "--copy" in sys.argv:
+            raise click.BadParameter(
+                click.style(
+                    "Cannot use both --non-fast and --copy flags together. Please move to --copy.",
+                    fg="red",
+                )
+            )
         copy = CopyFileDetection.NO_COPY
     if copy == CopyFileDetection.NO_COPY and not version:
         raise ValueError("Version is a required parameter in case --copy none is specified.")

diff --git a/flytekit/clis/sdk_in_container/run.py b/flytekit/clis/sdk_in_container/run.py
@@ -647,6 +647,13 @@ def _run(*args, **kwargs):
                     "The --copy_all flag is now deprecated. Please use --copy all instead.",
                     fg="yellow",
                 )
+                if "--copy" in sys.argv:
+                    raise click.BadParameter(
+                        click.style(
+                            "Cannot use both --copy-all and --copy flags together. Please move to --copy.",
+                            fg="red",
+                        )
+                    )
 
             with context_manager.FlyteContextManager.with_context(remote.context.new_builder()):
                 show_files = run_level_params.verbose > 0