Fixes type for plugin (flyteorg#2065)

Signed-off-by: Thomas J. Fan <[email protected]>
aswanthkrishna · Dec 21, 2023 · 274549f · 274549f
1 parent e45fdc2
commit 274549f
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/flytekit/configuration/plugin.py b/flytekit/configuration/plugin.py
@@ -19,7 +19,7 @@
 """
 from typing import Optional, Protocol, runtime_checkable
 
-from click import Command
+from click import Group
 from importlib_metadata import entry_points
 
 from flytekit.configuration import Config, get_config_file
@@ -36,7 +36,7 @@ def get_remote(
         """Get FlyteRemote object for CLI session."""
 
     @staticmethod
-    def configure_pyflyte_cli(main: Command) -> Command:
+    def configure_pyflyte_cli(main: Group) -> Group:
         """Configure pyflyte's CLI."""
 
 
@@ -58,7 +58,7 @@ def get_remote(
         )
 
     @staticmethod
-    def configure_pyflyte_cli(main: Command) -> Command:
+    def configure_pyflyte_cli(main: Group) -> Group:
         """Configure pyflyte's CLI."""
         return main