def METHOD_NAME(cls, *args, **kwargs):
    if cls._args_schema is not None:
        return cls._args_schema
    cls._args_schema = super().METHOD_NAME(*args, **kwargs)
    # define Arg Group ""
    # define Arg Group "RequiredAMLFilesystemSubnetsSizeInfo"
    _args_schema = cls._args_schema
    _args_schema.storage_capacity = AAZFloatArg(
        options=["--storage-capacity"],
        arg_group="RequiredAMLFilesystemSubnetsSizeInfo",
        help="The size of the AML file system, in TiB.",
    )
    # define Arg Group "Sku"
    _args_schema = cls._args_schema
    _args_schema.sku = AAZStrArg(
        options=["--sku"],
        arg_group="Sku",
        help="SKU name for this resource.",
    )
    return cls._args_schema