diff --git a/src/ai/backend/client/cli/vfolder.py b/src/ai/backend/client/cli/vfolder.py index 584c900843c..65b3d9eac51 100644 --- a/src/ai/backend/client/cli/vfolder.py +++ b/src/ai/backend/client/cli/vfolder.py @@ -79,9 +79,11 @@ def list_allowed_types(): @click.option( "--unmanaged", "host_path", - is_flag=True, + metavar="HOST_PATH", + type=str, + default=None, help=( - "Treats HOST as a mount point of unmanaged virtual folder. " + "Map a given host path to a virtual folder. " "This option can only be used by Admin or Superadmin." ), ) @@ -124,24 +126,15 @@ def create(name, host, group, host_path, usage_mode, permission, cloneable): """ with Session() as session: try: - if host_path: - result = session.VFolder.create( - name=name, - unmanaged_path=host, - group=group, - usage_mode=usage_mode, - permission=permission, - cloneable=cloneable, - ) - else: - result = session.VFolder.create( - name=name, - host=host, - group=group, - usage_mode=usage_mode, - permission=permission, - cloneable=cloneable, - ) + result = session.VFolder.create( + name=name, + host=host, + unmanaged_path=host_path, + group=group, + usage_mode=usage_mode, + permission=permission, + cloneable=cloneable, + ) print('Virtual folder "{0}" is created.'.format(result["name"])) except Exception as e: print_error(e)