flyte 0.2.0b1__py3-none-any.whl → 2.0.0b46__py3-none-any.whl

flyte/_datastructures.py

@@ -1,342 +0,0 @@
-from __future__ import annotations
-
-import inspect
-import os
-import pathlib
-import tempfile
-from dataclasses import dataclass, field, replace
-from typing import TYPE_CHECKING, Any, Callable, Dict, Optional, Tuple, Type
-
-from flyte._docstring import Docstring
-from flyte._interface import extract_return_annotation
-from flyte._logging import logger
-from flyte._utils.helpers import base36_encode
-
-if TYPE_CHECKING:
-    from flyte._internal.imagebuild.image_builder import ImageCache
-    from flyte.report import Report
-
-
-def generate_random_name() -> str:
-    """
-    Generate a random name for the task. This is used to create unique names for tasks.
-    TODO we can use unique-namer in the future, for now its just guids
-    """
-    from uuid import uuid4
-
-    return str(uuid4())  # Placeholder for actual random name generation logic
-
-
-@dataclass(frozen=True, kw_only=True)
-class ActionID:
-    """
-    A class representing the ID of an Action, nested within a Run. This is used to identify a specific action on a task.
-    """
-
-    name: str
-    run_name: str | None = None
-    project: str | None = None
-    domain: str | None = None
-    org: str | None = None
-
-    def __post_init__(self):
-        if self.run_name is None:
-            object.__setattr__(self, "run_name", self.name)
-
-    @classmethod
-    def create_random(cls):
-        name = generate_random_name()
-        return cls(name=name, run_name=name)
-
-    def new_sub_action(self, name: str | None = None) -> ActionID:
-        """
-        Create a new sub-run with the given name. If  name is None, a random name will be generated.
-        """
-        if name is None:
-            name = generate_random_name()
-        return replace(self, name=name)
-
-    def new_sub_action_from(self, task_name: str, input_hash: str, group: str | None) -> ActionID:
-        """Make a deterministic name"""
-        import hashlib
-
-        components = f"{self.run_name}-{self.name}-{input_hash}-{task_name}" + (f"-{group}" if group else "")
-        # has the components into something deterministic
-        bytes_digest = hashlib.md5(components.encode()).digest()
-        new_name = base36_encode(bytes_digest)
-        return self.new_sub_action(new_name)
-
-
-@dataclass(frozen=True, kw_only=True)
-class RawDataPath:
-    """
-    A class representing the raw data path for a task. This is used to store the raw data for the task execution and
-    also get mutations on the path.
-    """
-
-    path: str
-
-    @classmethod
-    def from_local_folder(cls, local_folder: str | pathlib.Path | None = None) -> RawDataPath:
-        """
-        Create a new context attribute object, with local path given. Will be created if it doesn't exist.
-        :return: Path to the temporary directory
-        """
-        match local_folder:
-            case pathlib.Path():
-                local_folder.mkdir(parents=True, exist_ok=True)
-                return RawDataPath(path=str(local_folder))
-            case None:
-                # Create a temporary directory for data storage
-                p = tempfile.mkdtemp()
-                logger.debug(f"Creating temporary directory for data storage: {p}")
-                return RawDataPath(path=p)
-            case str():
-                return RawDataPath(path=local_folder)
-            case _:
-                raise ValueError(f"Invalid local path {local_folder}")
-
-    def get_random_remote_path(self, file_name: Optional[str] = None) -> str:
-        """
-        Returns a random path for uploading a file/directory to.
-
-        :param file_name: If given, will be joined after a randomly generated portion.
-        :return:
-        """
-        import random
-        from uuid import UUID
-
-        import fsspec
-        from fsspec.utils import get_protocol
-
-        random_string = UUID(int=random.getrandbits(128)).hex
-        file_prefix = self.path
-
-        protocol = get_protocol(file_prefix)
-        if "file" in protocol:
-            local_path = pathlib.Path(file_prefix) / random_string
-            if file_name:
-                # Only if file name is given do we create the parent, because it may be needed as a folder otherwise
-                local_path = local_path / file_name
-                if not local_path.exists():
-                    local_path.parent.mkdir(exist_ok=True, parents=True)
-                    local_path.touch()
-            return str(local_path.absolute())
-
-        fs = fsspec.filesystem(protocol)
-        if file_prefix.endswith(fs.sep):
-            file_prefix = file_prefix[:-1]
-        remote_path = fs.sep.join([file_prefix, random_string])
-        if file_name:
-            remote_path = fs.sep.join([remote_path, file_name])
-        return remote_path
-
-
-@dataclass(frozen=True)
-class GroupData:
-    name: str
-
-
-@dataclass(frozen=True, kw_only=True)
-class TaskContext:
-    """
-    A context class to hold the current task executions context.
-    This can be used to access various contextual parameters in the task execution by the user.
-
-    :param action: The action ID of the current execution. This is always set, within a run.
-    :param version: The version of the executed task. This is set when the task is executed by an action and will be
-      set on all sub-actions.
-    """
-
-    action: ActionID
-    version: str
-    raw_data_path: RawDataPath
-    output_path: str
-    run_base_dir: str
-    report: Report
-    group_data: GroupData | None = None
-    checkpoints: Checkpoints | None = None
-    code_bundle: CodeBundle | None = None
-    compiled_image_cache: ImageCache | None = None
-    data: Dict[str, Any] = field(default_factory=dict)
-
-    def replace(self, **kwargs) -> TaskContext:
-        if "data" in kwargs:
-            rec_data = kwargs.pop("data")
-            if rec_data is None:
-                return replace(self, **kwargs)
-            data = {}
-            if self.data is not None:
-                data = self.data.copy()
-            data.update(rec_data)
-            kwargs.update({"data": data})
-        return replace(self, **kwargs)
-
-    def __getitem__(self, key: str) -> Optional[Any]:
-        return self.data.get(key)
-
-
-@dataclass(frozen=True, kw_only=True)
-class CodeBundle:
-    """
-    A class representing a code bundle for a task. This is used to package the code and the inflation path.
-    The code bundle computes the version of the code using the hash of the code.
-
-    :param computed_version: The version of the code bundle. This is the hash of the code.
-    :param destination: The destination path for the code bundle to be inflated to.
-    :param tgz: Optional path to the tgz file.
-    :param pkl: Optional path to the pkl file.
-    :param downloaded_path: The path to the downloaded code bundle. This is only available during runtime, when
-        the code bundle has been downloaded and inflated.
-    """
-
-    computed_version: str
-    destination: str = "."
-    tgz: str | None = None
-    pkl: str | None = None
-    downloaded_path: pathlib.Path | None = None
-
-    # runtime_dependencies: Tuple[str, ...] = field(default_factory=tuple)  In the future if we want we could add this
-    # but this messes up actors, spark etc
-
-    def __post_init__(self):
-        if self.tgz is None and self.pkl is None:
-            raise ValueError("Either tgz or pkl must be provided")
-
-    def with_downloaded_path(self, path: pathlib.Path) -> CodeBundle:
-        """
-        Create a new CodeBundle with the given downloaded path.
-        """
-        return replace(self, downloaded_path=path)
-
-
-@dataclass(frozen=True)
-class Checkpoints:
-    """
-    A class representing the checkpoints for a task. This is used to store the checkpoints for the task execution.
-    """
-
-    prev_checkpoint_path: str | None
-    checkpoint_path: str | None
-
-
-@dataclass(frozen=True)
-class NativeInterface:
-    """
-    A class representing the native interface for a task. This is used to interact with the task and its execution
-    context.
-    """
-
-    inputs: Dict[str, Tuple[Type, Any]]
-    outputs: Dict[str, Type]
-    docstring: Optional[Docstring] = field(default=None)
-
-    def has_outputs(self) -> bool:
-        """
-        Check if the task has outputs. This is used to determine if the task has outputs or not.
-        """
-        return self.outputs is not None and len(self.outputs) > 0
-
-    @classmethod
-    def from_types(cls, inputs: Dict[str, Type], outputs: Dict[str, Type]) -> NativeInterface:
-        """
-        Create a new NativeInterface from the given types. This is used to create a native interface for the task.
-        """
-        return cls(inputs={k: (v, inspect.Parameter.empty) for k, v in inputs.items()}, outputs=outputs)
-
-    @classmethod
-    def from_callable(cls, func: Callable) -> NativeInterface:
-        """
-        Extract the native interface from the given function. This is used to create a native interface for the task.
-        """
-        sig = inspect.signature(func)
-
-        # Extract parameter details (name, type, default value)
-        param_info = {name: (param.annotation, param.default) for name, param in sig.parameters.items()}
-
-        # Get return type
-        outputs = extract_return_annotation(sig.return_annotation)
-        return cls(inputs=param_info, outputs=outputs)
-
-    def convert_to_kwargs(self, *args, **kwargs) -> Dict[str, Any]:
-        """
-        Convert the given arguments to keyword arguments based on the native interface. This is used to convert the
-        arguments to the correct types for the task execution.
-        """
-        # Convert positional arguments to keyword arguments
-        if len(args) > len(self.inputs):
-            raise ValueError(f"Too many positional arguments provided, inputs {self.inputs.keys()}, args {len(args)}")
-        for arg, input_name in zip(args, self.inputs.keys()):
-            kwargs[input_name] = arg
-        return kwargs
-
-    def get_input_types(self) -> Dict[str, Type]:
-        """
-        Get the input types for the task. This is used to get the types of the inputs for the task execution.
-        """
-        return {k: v[0] for k, v in self.inputs.items()}
-
-    def __repr__(self):
-        """
-        Returns a string representation of the task interface.
-        """
-        i = "("
-        if self.inputs:
-            initial = True
-            for key, tpe in self.inputs.items():
-                if not initial:
-                    i += ", "
-                initial = False
-                tp = tpe[0] if isinstance(tpe[0], str) else tpe[0].__name__
-                i += f"{key}: {tp}"
-                if tpe[1] is not inspect.Parameter.empty:
-                    i += f" = {tpe[1]}"
-        i += ")"
-        if self.outputs:
-            initial = True
-            multi = len(self.outputs) > 1
-            i += " -> "
-            if multi:
-                i += "("
-            for key, tpe in self.outputs.items():
-                if not initial:
-                    i += ", "
-                initial = False
-                tp = tpe.__name__ if isinstance(tpe, type) else tpe
-                i += f"{key}: {tp}"
-            if multi:
-                i += ")"
-        return i + ":"
-
-
-@dataclass
-class SerializationContext:
-    """
-    This object holds serialization time contextual information, that can be used when serializing the task and
-    various parameters of a tasktemplate. This is only available when the task is being serialized and can be
-    during a deployment or runtime.
-
-    :param version: The version of the task
-    :param code_bundle: The code bundle for the task. This is used to package the code and the inflation path.
-    :param input_path: The path to the inputs for the task. This is used to determine where the inputs will be located
-    :param output_path: The path to the outputs for the task. This is used to determine where the outputs will be
-     located
-    """
-
-    version: str
-    project: str | None = None
-    domain: str | None = None
-    org: str | None = None
-    code_bundle: Optional[CodeBundle] = None
-    input_path: str = "{{.input}}"
-    output_path: str = "{{.outputPrefix}}"
-    _entrypoint_path: str = field(default="_bin/runtime.py", init=False)
-    image_cache: ImageCache | None = None
-    root_dir: Optional[pathlib.Path] = None
-
-    def get_entrypoint_path(self, interpreter_path: str) -> str:
-        """
-        Get the entrypoint path for the task. This is used to determine the entrypoint for the task execution.
-        :param interpreter_path: The path to the interpreter (python)
-        """
-        return os.path.join(os.path.dirname(os.path.dirname(interpreter_path)), self._entrypoint_path)

flyte/_internal/controllers/pbhash.py

@@ -1,39 +0,0 @@
-# This is a module that provides hashing utilities for Protobuf objects.
-import base64
-import hashlib
-import json
-
-from google.protobuf import json_format
-from google.protobuf.message import Message
-
-
-def compute_hash(pb: Message) -> bytes:
-    """
-    Computes a deterministic hash in bytes for the Protobuf object.
-    """
-    try:
-        pb_dict = json_format.MessageToDict(pb)
-        # json.dumps with sorted keys to ensure stability
-        stable_json_str = json.dumps(
-            pb_dict, sort_keys=True, separators=(",", ":")
-        )  # separators to ensure no extra spaces
-    except Exception as e:
-        raise ValueError(f"Failed to marshal Protobuf object {pb} to JSON with error: {e}")
-
-    try:
-        # Deterministically hash the JSON object to a byte array. Using SHA-256 for hashing here,
-        # assuming it provides a consistent hash output.
-        hash_obj = hashlib.sha256(stable_json_str.encode("utf-8"))
-    except Exception as e:
-        raise ValueError(f"Failed to hash JSON for Protobuf object {pb} with error: {e}")
-
-    # The digest is guaranteed to be 32 bytes long
-    return hash_obj.digest()
-
-
-def compute_hash_string(pb: Message) -> str:
-    """
-    Computes a deterministic hash in base64 encoded string for the Protobuf object
-    """
-    hash_bytes = compute_hash(pb)
-    return base64.b64encode(hash_bytes).decode("utf-8")

flyte/_protos/common/authorization_pb2.py

@@ -1,66 +0,0 @@
-# -*- coding: utf-8 -*-
-# Generated by the protocol buffer compiler.  DO NOT EDIT!
-# source: common/authorization.proto
-"""Generated protocol buffer code."""
-from google.protobuf import descriptor as _descriptor
-from google.protobuf import descriptor_pool as _descriptor_pool
-from google.protobuf import symbol_database as _symbol_database
-from google.protobuf.internal import builder as _builder
-# @@protoc_insertion_point(imports)
-
-_sym_db = _symbol_database.Default()
-
-
-from flyte._protos.common import identifier_pb2 as common_dot_identifier__pb2
-from flyte._protos.validate.validate import validate_pb2 as validate_dot_validate__pb2
-
-
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1a\x63ommon/authorization.proto\x12\x0f\x63loudidl.common\x1a\x17\x63ommon/identifier.proto\x1a\x17validate/validate.proto\"+\n\x0cOrganization\x12\x1b\n\x04name\x18\x01 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x04name\"i\n\x06\x44omain\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12K\n\x0corganization\x18\x02 \x01(\x0b\x32\x1d.cloudidl.common.OrganizationB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x0corganization\"a\n\x07Project\x12\x1b\n\x04name\x18\x01 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x04name\x12\x39\n\x06\x64omain\x18\x02 \x01(\x0b\x32\x17.cloudidl.common.DomainB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x06\x64omain\"e\n\x08Workflow\x12\x1b\n\x04name\x18\x01 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x04name\x12<\n\x07project\x18\x02 \x01(\x0b\x32\x18.cloudidl.common.ProjectB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x07project\"g\n\nLaunchPlan\x12\x1b\n\x04name\x18\x01 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x04name\x12<\n\x07project\x18\x02 \x01(\x0b\x32\x18.cloudidl.common.ProjectB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x07project\"\xfd\x02\n\x08Resource\x12\x43\n\x0corganization\x18\x01 \x01(\x0b\x32\x1d.cloudidl.common.OrganizationH\x00R\x0corganization\x12\x31\n\x06\x64omain\x18\x02 \x01(\x0b\x32\x17.cloudidl.common.DomainH\x00R\x06\x64omain\x12\x34\n\x07project\x18\x03 \x01(\x0b\x32\x18.cloudidl.common.ProjectH\x00R\x07project\x12\x37\n\x08workflow\x18\x04 \x01(\x0b\x32\x19.cloudidl.common.WorkflowH\x00R\x08workflow\x12>\n\x0blaunch_plan\x18\x05 \x01(\x0b\x32\x1b.cloudidl.common.LaunchPlanH\x00R\nlaunchPlan\x12>\n\x07\x63luster\x18\x06 \x01(\x0b\x32\".cloudidl.common.ClusterIdentifierH\x00R\x07\x63lusterB\n\n\x08resource\"v\n\nPermission\x12\x35\n\x08resource\x18\x01 \x01(\x0b\x32\x19.cloudidl.common.ResourceR\x08resource\x12\x31\n\x07\x61\x63tions\x18\x02 \x03(\x0e\x32\x17.cloudidl.common.ActionR\x07\x61\x63tions*\x94\x04\n\x06\x41\x63tion\x12\x0f\n\x0b\x41\x43TION_NONE\x10\x00\x12\x15\n\rACTION_CREATE\x10\x01\x1a\x02\x08\x01\x12\x13\n\x0b\x41\x43TION_READ\x10\x02\x1a\x02\x08\x01\x12\x15\n\rACTION_UPDATE\x10\x03\x1a\x02\x08\x01\x12\x15\n\rACTION_DELETE\x10\x04\x1a\x02\x08\x01\x12\x1f\n\x1b\x41\x43TION_VIEW_FLYTE_INVENTORY\x10\x05\x12 \n\x1c\x41\x43TION_VIEW_FLYTE_EXECUTIONS\x10\x06\x12#\n\x1f\x41\x43TION_REGISTER_FLYTE_INVENTORY\x10\x07\x12\"\n\x1e\x41\x43TION_CREATE_FLYTE_EXECUTIONS\x10\x08\x12\x1d\n\x19\x41\x43TION_ADMINISTER_PROJECT\x10\t\x12\x1d\n\x19\x41\x43TION_MANAGE_PERMISSIONS\x10\n\x12\x1d\n\x19\x41\x43TION_ADMINISTER_ACCOUNT\x10\x0b\x12\x19\n\x15\x41\x43TION_MANAGE_CLUSTER\x10\x0c\x12,\n(ACTION_EDIT_EXECUTION_RELATED_ATTRIBUTES\x10\r\x12*\n&ACTION_EDIT_CLUSTER_RELATED_ATTRIBUTES\x10\x0e\x12!\n\x1d\x41\x43TION_EDIT_UNUSED_ATTRIBUTES\x10\x0f\x12\x1e\n\x1a\x41\x43TION_SUPPORT_SYSTEM_LOGS\x10\x10\x42\xb3\x01\n\x13\x63om.cloudidl.commonB\x12\x41uthorizationProtoH\x02P\x01Z)github.com/unionai/cloud/gen/pb-go/common\xa2\x02\x03\x43\x43X\xaa\x02\x0f\x43loudidl.Common\xca\x02\x0f\x43loudidl\\Common\xe2\x02\x1b\x43loudidl\\Common\\GPBMetadata\xea\x02\x10\x43loudidl::Commonb\x06proto3')
-
-_globals = globals()
-_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
-_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'common.authorization_pb2', _globals)
-if _descriptor._USE_C_DESCRIPTORS == False:
-  DESCRIPTOR._options = None
-  DESCRIPTOR._serialized_options = b'\n\023com.cloudidl.commonB\022AuthorizationProtoH\002P\001Z)github.com/unionai/cloud/gen/pb-go/common\242\002\003CCX\252\002\017Cloudidl.Common\312\002\017Cloudidl\\Common\342\002\033Cloudidl\\Common\\GPBMetadata\352\002\020Cloudidl::Common'
-  _ACTION.values_by_name["ACTION_CREATE"]._options = None
-  _ACTION.values_by_name["ACTION_CREATE"]._serialized_options = b'\010\001'
-  _ACTION.values_by_name["ACTION_READ"]._options = None
-  _ACTION.values_by_name["ACTION_READ"]._serialized_options = b'\010\001'
-  _ACTION.values_by_name["ACTION_UPDATE"]._options = None
-  _ACTION.values_by_name["ACTION_UPDATE"]._serialized_options = b'\010\001'
-  _ACTION.values_by_name["ACTION_DELETE"]._options = None
-  _ACTION.values_by_name["ACTION_DELETE"]._serialized_options = b'\010\001'
-  _ORGANIZATION.fields_by_name['name']._options = None
-  _ORGANIZATION.fields_by_name['name']._serialized_options = b'\372B\004r\002\020\001'
-  _DOMAIN.fields_by_name['organization']._options = None
-  _DOMAIN.fields_by_name['organization']._serialized_options = b'\372B\005\212\001\002\020\001'
-  _PROJECT.fields_by_name['name']._options = None
-  _PROJECT.fields_by_name['name']._serialized_options = b'\372B\004r\002\020\001'
-  _PROJECT.fields_by_name['domain']._options = None
-  _PROJECT.fields_by_name['domain']._serialized_options = b'\372B\005\212\001\002\020\001'
-  _WORKFLOW.fields_by_name['name']._options = None
-  _WORKFLOW.fields_by_name['name']._serialized_options = b'\372B\004r\002\020\001'
-  _WORKFLOW.fields_by_name['project']._options = None
-  _WORKFLOW.fields_by_name['project']._serialized_options = b'\372B\005\212\001\002\020\001'
-  _LAUNCHPLAN.fields_by_name['name']._options = None
-  _LAUNCHPLAN.fields_by_name['name']._serialized_options = b'\372B\004r\002\020\001'
-  _LAUNCHPLAN.fields_by_name['project']._options = None
-  _LAUNCHPLAN.fields_by_name['project']._serialized_options = b'\372B\005\212\001\002\020\001'
-  _globals['_ACTION']._serialized_start=1061
-  _globals['_ACTION']._serialized_end=1593
-  _globals['_ORGANIZATION']._serialized_start=97
-  _globals['_ORGANIZATION']._serialized_end=140
-  _globals['_DOMAIN']._serialized_start=142
-  _globals['_DOMAIN']._serialized_end=247
-  _globals['_PROJECT']._serialized_start=249
-  _globals['_PROJECT']._serialized_end=346
-  _globals['_WORKFLOW']._serialized_start=348
-  _globals['_WORKFLOW']._serialized_end=449
-  _globals['_LAUNCHPLAN']._serialized_start=451
-  _globals['_LAUNCHPLAN']._serialized_end=554
-  _globals['_RESOURCE']._serialized_start=557
-  _globals['_RESOURCE']._serialized_end=938
-  _globals['_PERMISSION']._serialized_start=940
-  _globals['_PERMISSION']._serialized_end=1058
-# @@protoc_insertion_point(module_scope)

flyte/_protos/common/authorization_pb2.pyi

@@ -1,108 +0,0 @@
-from flyte._protos.common import identifier_pb2 as _identifier_pb2
-from flyte._protos.validate.validate import validate_pb2 as _validate_pb2
-from google.protobuf.internal import containers as _containers
-from google.protobuf.internal import enum_type_wrapper as _enum_type_wrapper
-from google.protobuf import descriptor as _descriptor
-from google.protobuf import message as _message
-from typing import ClassVar as _ClassVar, Iterable as _Iterable, Mapping as _Mapping, Optional as _Optional, Union as _Union
-
-DESCRIPTOR: _descriptor.FileDescriptor
-
-class Action(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
-    __slots__ = []
-    ACTION_NONE: _ClassVar[Action]
-    ACTION_CREATE: _ClassVar[Action]
-    ACTION_READ: _ClassVar[Action]
-    ACTION_UPDATE: _ClassVar[Action]
-    ACTION_DELETE: _ClassVar[Action]
-    ACTION_VIEW_FLYTE_INVENTORY: _ClassVar[Action]
-    ACTION_VIEW_FLYTE_EXECUTIONS: _ClassVar[Action]
-    ACTION_REGISTER_FLYTE_INVENTORY: _ClassVar[Action]
-    ACTION_CREATE_FLYTE_EXECUTIONS: _ClassVar[Action]
-    ACTION_ADMINISTER_PROJECT: _ClassVar[Action]
-    ACTION_MANAGE_PERMISSIONS: _ClassVar[Action]
-    ACTION_ADMINISTER_ACCOUNT: _ClassVar[Action]
-    ACTION_MANAGE_CLUSTER: _ClassVar[Action]
-    ACTION_EDIT_EXECUTION_RELATED_ATTRIBUTES: _ClassVar[Action]
-    ACTION_EDIT_CLUSTER_RELATED_ATTRIBUTES: _ClassVar[Action]
-    ACTION_EDIT_UNUSED_ATTRIBUTES: _ClassVar[Action]
-    ACTION_SUPPORT_SYSTEM_LOGS: _ClassVar[Action]
-ACTION_NONE: Action
-ACTION_CREATE: Action
-ACTION_READ: Action
-ACTION_UPDATE: Action
-ACTION_DELETE: Action
-ACTION_VIEW_FLYTE_INVENTORY: Action
-ACTION_VIEW_FLYTE_EXECUTIONS: Action
-ACTION_REGISTER_FLYTE_INVENTORY: Action
-ACTION_CREATE_FLYTE_EXECUTIONS: Action
-ACTION_ADMINISTER_PROJECT: Action
-ACTION_MANAGE_PERMISSIONS: Action
-ACTION_ADMINISTER_ACCOUNT: Action
-ACTION_MANAGE_CLUSTER: Action
-ACTION_EDIT_EXECUTION_RELATED_ATTRIBUTES: Action
-ACTION_EDIT_CLUSTER_RELATED_ATTRIBUTES: Action
-ACTION_EDIT_UNUSED_ATTRIBUTES: Action
-ACTION_SUPPORT_SYSTEM_LOGS: Action
-
-class Organization(_message.Message):
-    __slots__ = ["name"]
-    NAME_FIELD_NUMBER: _ClassVar[int]
-    name: str
-    def __init__(self, name: _Optional[str] = ...) -> None: ...
-
-class Domain(_message.Message):
-    __slots__ = ["name", "organization"]
-    NAME_FIELD_NUMBER: _ClassVar[int]
-    ORGANIZATION_FIELD_NUMBER: _ClassVar[int]
-    name: str
-    organization: Organization
-    def __init__(self, name: _Optional[str] = ..., organization: _Optional[_Union[Organization, _Mapping]] = ...) -> None: ...
-
-class Project(_message.Message):
-    __slots__ = ["name", "domain"]
-    NAME_FIELD_NUMBER: _ClassVar[int]
-    DOMAIN_FIELD_NUMBER: _ClassVar[int]
-    name: str
-    domain: Domain
-    def __init__(self, name: _Optional[str] = ..., domain: _Optional[_Union[Domain, _Mapping]] = ...) -> None: ...
-
-class Workflow(_message.Message):
-    __slots__ = ["name", "project"]
-    NAME_FIELD_NUMBER: _ClassVar[int]
-    PROJECT_FIELD_NUMBER: _ClassVar[int]
-    name: str
-    project: Project
-    def __init__(self, name: _Optional[str] = ..., project: _Optional[_Union[Project, _Mapping]] = ...) -> None: ...
-
-class LaunchPlan(_message.Message):
-    __slots__ = ["name", "project"]
-    NAME_FIELD_NUMBER: _ClassVar[int]
-    PROJECT_FIELD_NUMBER: _ClassVar[int]
-    name: str
-    project: Project
-    def __init__(self, name: _Optional[str] = ..., project: _Optional[_Union[Project, _Mapping]] = ...) -> None: ...
-
-class Resource(_message.Message):
-    __slots__ = ["organization", "domain", "project", "workflow", "launch_plan", "cluster"]
-    ORGANIZATION_FIELD_NUMBER: _ClassVar[int]
-    DOMAIN_FIELD_NUMBER: _ClassVar[int]
-    PROJECT_FIELD_NUMBER: _ClassVar[int]
-    WORKFLOW_FIELD_NUMBER: _ClassVar[int]
-    LAUNCH_PLAN_FIELD_NUMBER: _ClassVar[int]
-    CLUSTER_FIELD_NUMBER: _ClassVar[int]
-    organization: Organization
-    domain: Domain
-    project: Project
-    workflow: Workflow
-    launch_plan: LaunchPlan
-    cluster: _identifier_pb2.ClusterIdentifier
-    def __init__(self, organization: _Optional[_Union[Organization, _Mapping]] = ..., domain: _Optional[_Union[Domain, _Mapping]] = ..., project: _Optional[_Union[Project, _Mapping]] = ..., workflow: _Optional[_Union[Workflow, _Mapping]] = ..., launch_plan: _Optional[_Union[LaunchPlan, _Mapping]] = ..., cluster: _Optional[_Union[_identifier_pb2.ClusterIdentifier, _Mapping]] = ...) -> None: ...
-
-class Permission(_message.Message):
-    __slots__ = ["resource", "actions"]
-    RESOURCE_FIELD_NUMBER: _ClassVar[int]
-    ACTIONS_FIELD_NUMBER: _ClassVar[int]
-    resource: Resource
-    actions: _containers.RepeatedScalarFieldContainer[Action]
-    def __init__(self, resource: _Optional[_Union[Resource, _Mapping]] = ..., actions: _Optional[_Iterable[_Union[Action, str]]] = ...) -> None: ...

flyte/_protos/common/authorization_pb2_grpc.py

@@ -1,4 +0,0 @@
-# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
-"""Client and server classes corresponding to protobuf-defined services."""
-import grpc
-

flyte/_protos/common/identifier_pb2.py

@@ -1,71 +0,0 @@
-# -*- coding: utf-8 -*-
-# Generated by the protocol buffer compiler.  DO NOT EDIT!
-# source: common/identifier.proto
-"""Generated protocol buffer code."""
-from google.protobuf import descriptor as _descriptor
-from google.protobuf import descriptor_pool as _descriptor_pool
-from google.protobuf import symbol_database as _symbol_database
-from google.protobuf.internal import builder as _builder
-# @@protoc_insertion_point(imports)
-
-_sym_db = _symbol_database.Default()
-
-
-from flyte._protos.validate.validate import validate_pb2 as validate_dot_validate__pb2
-
-
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x17\x63ommon/identifier.proto\x12\x0f\x63loudidl.common\x1a\x17validate/validate.proto\"~\n\x11ProjectIdentifier\x12+\n\x0corganization\x18\x01 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x0corganization\x12\x1f\n\x06\x64omain\x18\x02 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x06\x64omain\x12\x1b\n\x04name\x18\x03 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x04name\"T\n\x11\x43lusterIdentifier\x12\"\n\x0corganization\x18\x01 \x01(\tR\x0corganization\x12\x1b\n\x04name\x18\x02 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x04name\"O\n\x15\x43lusterPoolIdentifier\x12\"\n\x0corganization\x18\x01 \x01(\tR\x0corganization\x12\x12\n\x04name\x18\x02 \x01(\tR\x04name\"_\n\x17\x43lusterConfigIdentifier\x12+\n\x0corganization\x18\x01 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x0corganization\x12\x17\n\x02id\x18\x02 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x02id\"3\n\x0eUserIdentifier\x12!\n\x07subject\x18\x01 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x07subject\":\n\x15\x41pplicationIdentifier\x12!\n\x07subject\x18\x01 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x07subject\"Q\n\x0eRoleIdentifier\x12\"\n\x0corganization\x18\x01 \x01(\tR\x0corganization\x12\x1b\n\x04name\x18\x02 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x04name\"O\n\rOrgIdentifier\x12>\n\x04name\x18\x01 \x01(\tB*\xfa\x42\'r%\x10\x01\x18?2\x1f^[a-z0-9]([-a-z0-9]*[a-z0-9])?$R\x04name\"y\n\x18ManagedClusterIdentifier\x12\x1b\n\x04name\x18\x02 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x04name\x12:\n\x03org\x18\x03 \x01(\x0b\x32\x1e.cloudidl.common.OrgIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x03orgJ\x04\x08\x01\x10\x02\"S\n\x10PolicyIdentifier\x12\"\n\x0corganization\x18\x01 \x01(\tR\x0corganization\x12\x1b\n\x04name\x18\x02 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x04nameB\xb0\x01\n\x13\x63om.cloudidl.commonB\x0fIdentifierProtoH\x02P\x01Z)github.com/unionai/cloud/gen/pb-go/common\xa2\x02\x03\x43\x43X\xaa\x02\x0f\x43loudidl.Common\xca\x02\x0f\x43loudidl\\Common\xe2\x02\x1b\x43loudidl\\Common\\GPBMetadata\xea\x02\x10\x43loudidl::Commonb\x06proto3')
-
-_globals = globals()
-_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
-_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'common.identifier_pb2', _globals)
-if _descriptor._USE_C_DESCRIPTORS == False:
-  DESCRIPTOR._options = None
-  DESCRIPTOR._serialized_options = b'\n\023com.cloudidl.commonB\017IdentifierProtoH\002P\001Z)github.com/unionai/cloud/gen/pb-go/common\242\002\003CCX\252\002\017Cloudidl.Common\312\002\017Cloudidl\\Common\342\002\033Cloudidl\\Common\\GPBMetadata\352\002\020Cloudidl::Common'
-  _PROJECTIDENTIFIER.fields_by_name['organization']._options = None
-  _PROJECTIDENTIFIER.fields_by_name['organization']._serialized_options = b'\372B\004r\002\020\001'
-  _PROJECTIDENTIFIER.fields_by_name['domain']._options = None
-  _PROJECTIDENTIFIER.fields_by_name['domain']._serialized_options = b'\372B\004r\002\020\001'
-  _PROJECTIDENTIFIER.fields_by_name['name']._options = None
-  _PROJECTIDENTIFIER.fields_by_name['name']._serialized_options = b'\372B\004r\002\020\001'
-  _CLUSTERIDENTIFIER.fields_by_name['name']._options = None
-  _CLUSTERIDENTIFIER.fields_by_name['name']._serialized_options = b'\372B\004r\002\020\001'
-  _CLUSTERCONFIGIDENTIFIER.fields_by_name['organization']._options = None
-  _CLUSTERCONFIGIDENTIFIER.fields_by_name['organization']._serialized_options = b'\372B\004r\002\020\001'
-  _CLUSTERCONFIGIDENTIFIER.fields_by_name['id']._options = None
-  _CLUSTERCONFIGIDENTIFIER.fields_by_name['id']._serialized_options = b'\372B\004r\002\020\001'
-  _USERIDENTIFIER.fields_by_name['subject']._options = None
-  _USERIDENTIFIER.fields_by_name['subject']._serialized_options = b'\372B\004r\002\020\001'
-  _APPLICATIONIDENTIFIER.fields_by_name['subject']._options = None
-  _APPLICATIONIDENTIFIER.fields_by_name['subject']._serialized_options = b'\372B\004r\002\020\001'
-  _ROLEIDENTIFIER.fields_by_name['name']._options = None
-  _ROLEIDENTIFIER.fields_by_name['name']._serialized_options = b'\372B\004r\002\020\001'
-  _ORGIDENTIFIER.fields_by_name['name']._options = None
-  _ORGIDENTIFIER.fields_by_name['name']._serialized_options = b'\372B\'r%\020\001\030?2\037^[a-z0-9]([-a-z0-9]*[a-z0-9])?$'
-  _MANAGEDCLUSTERIDENTIFIER.fields_by_name['name']._options = None
-  _MANAGEDCLUSTERIDENTIFIER.fields_by_name['name']._serialized_options = b'\372B\004r\002\020\001'
-  _MANAGEDCLUSTERIDENTIFIER.fields_by_name['org']._options = None
-  _MANAGEDCLUSTERIDENTIFIER.fields_by_name['org']._serialized_options = b'\372B\005\212\001\002\020\001'
-  _POLICYIDENTIFIER.fields_by_name['name']._options = None
-  _POLICYIDENTIFIER.fields_by_name['name']._serialized_options = b'\372B\004r\002\020\001'
-  _globals['_PROJECTIDENTIFIER']._serialized_start=69
-  _globals['_PROJECTIDENTIFIER']._serialized_end=195
-  _globals['_CLUSTERIDENTIFIER']._serialized_start=197
-  _globals['_CLUSTERIDENTIFIER']._serialized_end=281
-  _globals['_CLUSTERPOOLIDENTIFIER']._serialized_start=283
-  _globals['_CLUSTERPOOLIDENTIFIER']._serialized_end=362
-  _globals['_CLUSTERCONFIGIDENTIFIER']._serialized_start=364
-  _globals['_CLUSTERCONFIGIDENTIFIER']._serialized_end=459
-  _globals['_USERIDENTIFIER']._serialized_start=461
-  _globals['_USERIDENTIFIER']._serialized_end=512
-  _globals['_APPLICATIONIDENTIFIER']._serialized_start=514
-  _globals['_APPLICATIONIDENTIFIER']._serialized_end=572
-  _globals['_ROLEIDENTIFIER']._serialized_start=574
-  _globals['_ROLEIDENTIFIER']._serialized_end=655
-  _globals['_ORGIDENTIFIER']._serialized_start=657
-  _globals['_ORGIDENTIFIER']._serialized_end=736
-  _globals['_MANAGEDCLUSTERIDENTIFIER']._serialized_start=738
-  _globals['_MANAGEDCLUSTERIDENTIFIER']._serialized_end=859
-  _globals['_POLICYIDENTIFIER']._serialized_start=861
-  _globals['_POLICYIDENTIFIER']._serialized_end=944
-# @@protoc_insertion_point(module_scope)