basedpyright 1.36.2 → 1.37.0-961d59bacff1958014fe4e9ed2ef3116fe2266e2

package/dist/typeshed-fallback/stubs/tensorflow/tensorflow/core/protobuf/transport_options_pb2.pyi

@@ -0,0 +1,28 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+
+import builtins
+import collections.abc
+import typing
+
+import google.protobuf.descriptor
+import google.protobuf.internal.containers
+import google.protobuf.message
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+@typing.final
+class RecvBufRespExtra(google.protobuf.message.Message):
+    """Extra data needed on a non-RDMA RecvBufResponse."""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    TENSOR_CONTENT_FIELD_NUMBER: builtins.int
+    @property
+    def tensor_content(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.bytes]: ...
+    def __init__(self, *, tensor_content: collections.abc.Iterable[builtins.bytes] | None = ...) -> None: ...
+    def ClearField(self, field_name: typing.Literal["tensor_content", b"tensor_content"]) -> None: ...
+
+global___RecvBufRespExtra = RecvBufRespExtra

package/dist/typeshed-fallback/stubs/tensorflow/tensorflow/core/protobuf/verifier_config_pb2.pyi

@@ -0,0 +1,65 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+
+import builtins
+import sys
+import typing
+
+import google.protobuf.descriptor
+import google.protobuf.internal.enum_type_wrapper
+import google.protobuf.message
+
+if sys.version_info >= (3, 10):
+    import typing as typing_extensions
+else:
+    import typing_extensions
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+@typing.final
+class VerifierConfig(google.protobuf.message.Message):
+    """The config for graph verifiers."""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    class _Toggle:
+        ValueType = typing.NewType("ValueType", builtins.int)
+        V: typing_extensions.TypeAlias = ValueType
+
+    class _ToggleEnumTypeWrapper(
+        google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[VerifierConfig._Toggle.ValueType], builtins.type
+    ):
+        DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
+        DEFAULT: VerifierConfig._Toggle.ValueType  # 0
+        ON: VerifierConfig._Toggle.ValueType  # 1
+        OFF: VerifierConfig._Toggle.ValueType  # 2
+
+    class Toggle(_Toggle, metaclass=_ToggleEnumTypeWrapper): ...
+    DEFAULT: VerifierConfig.Toggle.ValueType  # 0
+    ON: VerifierConfig.Toggle.ValueType  # 1
+    OFF: VerifierConfig.Toggle.ValueType  # 2
+
+    VERIFICATION_TIMEOUT_IN_MS_FIELD_NUMBER: builtins.int
+    STRUCTURE_VERIFIER_FIELD_NUMBER: builtins.int
+    verification_timeout_in_ms: builtins.int
+    """Deadline for completion of all verification i.e. all the Toggle ON
+    verifiers must complete execution within this time.
+    """
+    structure_verifier: global___VerifierConfig.Toggle.ValueType
+    """Perform structural validation on a tensorflow graph. Default is OFF."""
+    def __init__(
+        self,
+        *,
+        verification_timeout_in_ms: builtins.int | None = ...,
+        structure_verifier: global___VerifierConfig.Toggle.ValueType | None = ...,
+    ) -> None: ...
+    def ClearField(
+        self,
+        field_name: typing.Literal[
+            "structure_verifier", b"structure_verifier", "verification_timeout_in_ms", b"verification_timeout_in_ms"
+        ],
+    ) -> None: ...
+
+global___VerifierConfig = VerifierConfig

package/dist/typeshed-fallback/stubs/tensorflow/tensorflow/core/util/event_pb2.pyi

@@ -0,0 +1,419 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+
+import builtins
+import collections.abc
+import sys
+import typing
+
+import google.protobuf.descriptor
+import google.protobuf.internal.containers
+import google.protobuf.internal.enum_type_wrapper
+import google.protobuf.message
+import tensorflow.core.framework.summary_pb2
+
+if sys.version_info >= (3, 10):
+    import typing as typing_extensions
+else:
+    import typing_extensions
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+class _WorkerHealth:
+    ValueType = typing.NewType("ValueType", builtins.int)
+    V: typing_extensions.TypeAlias = ValueType
+
+class _WorkerHealthEnumTypeWrapper(
+    google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_WorkerHealth.ValueType], builtins.type
+):
+    DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
+    OK: _WorkerHealth.ValueType  # 0
+    """By default a worker is healthy."""
+    RECEIVED_SHUTDOWN_SIGNAL: _WorkerHealth.ValueType  # 1
+    INTERNAL_ERROR: _WorkerHealth.ValueType  # 2
+    SHUTTING_DOWN: _WorkerHealth.ValueType  # 3
+    """Worker has been instructed to shutdown after a timeout."""
+
+class WorkerHealth(_WorkerHealth, metaclass=_WorkerHealthEnumTypeWrapper):
+    """Worker heartbeat messages.  Support for these operations is currently
+    internal and expected to change.
+
+    Current health status of a worker.
+    """
+
+OK: WorkerHealth.ValueType  # 0
+"""By default a worker is healthy."""
+RECEIVED_SHUTDOWN_SIGNAL: WorkerHealth.ValueType  # 1
+INTERNAL_ERROR: WorkerHealth.ValueType  # 2
+SHUTTING_DOWN: WorkerHealth.ValueType  # 3
+"""Worker has been instructed to shutdown after a timeout."""
+global___WorkerHealth = WorkerHealth
+
+class _WorkerShutdownMode:
+    ValueType = typing.NewType("ValueType", builtins.int)
+    V: typing_extensions.TypeAlias = ValueType
+
+class _WorkerShutdownModeEnumTypeWrapper(
+    google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_WorkerShutdownMode.ValueType], builtins.type
+):
+    DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
+    DEFAULT: _WorkerShutdownMode.ValueType  # 0
+    NOT_CONFIGURED: _WorkerShutdownMode.ValueType  # 1
+    WAIT_FOR_COORDINATOR: _WorkerShutdownMode.ValueType  # 2
+    SHUTDOWN_AFTER_TIMEOUT: _WorkerShutdownMode.ValueType  # 3
+
+class WorkerShutdownMode(_WorkerShutdownMode, metaclass=_WorkerShutdownModeEnumTypeWrapper):
+    """Indicates the behavior of the worker when an internal error or shutdown
+    signal is received.
+    """
+
+DEFAULT: WorkerShutdownMode.ValueType  # 0
+NOT_CONFIGURED: WorkerShutdownMode.ValueType  # 1
+WAIT_FOR_COORDINATOR: WorkerShutdownMode.ValueType  # 2
+SHUTDOWN_AFTER_TIMEOUT: WorkerShutdownMode.ValueType  # 3
+global___WorkerShutdownMode = WorkerShutdownMode
+
+@typing.final
+class Event(google.protobuf.message.Message):
+    """Protocol buffer representing an event that happened during
+    the execution of a Brain model.
+    """
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    WALL_TIME_FIELD_NUMBER: builtins.int
+    STEP_FIELD_NUMBER: builtins.int
+    FILE_VERSION_FIELD_NUMBER: builtins.int
+    GRAPH_DEF_FIELD_NUMBER: builtins.int
+    SUMMARY_FIELD_NUMBER: builtins.int
+    LOG_MESSAGE_FIELD_NUMBER: builtins.int
+    SESSION_LOG_FIELD_NUMBER: builtins.int
+    TAGGED_RUN_METADATA_FIELD_NUMBER: builtins.int
+    META_GRAPH_DEF_FIELD_NUMBER: builtins.int
+    SOURCE_METADATA_FIELD_NUMBER: builtins.int
+    wall_time: builtins.float
+    """Timestamp of the event."""
+    step: builtins.int
+    """Global step of the event."""
+    file_version: builtins.str
+    """An event file was started, with the specified version.
+    This is use to identify the contents of the record IO files
+    easily.  Current version is "brain.Event:2".  All versions
+    start with "brain.Event:".
+    """
+    graph_def: builtins.bytes
+    """An encoded version of a GraphDef."""
+    meta_graph_def: builtins.bytes
+    """An encoded version of a MetaGraphDef."""
+    @property
+    def summary(self) -> tensorflow.core.framework.summary_pb2.Summary:
+        """A summary was generated."""
+
+    @property
+    def log_message(self) -> global___LogMessage:
+        """The user output a log message. This was theoretically used by the defunct
+        tensorboard_logging module, which has since been removed; this field is
+        now deprecated and should not be used.
+        """
+
+    @property
+    def session_log(self) -> global___SessionLog:
+        """The state of the session which can be used for restarting after crashes."""
+
+    @property
+    def tagged_run_metadata(self) -> global___TaggedRunMetadata:
+        """The metadata returned by running a session.run() call."""
+
+    @property
+    def source_metadata(self) -> global___SourceMetadata:
+        """Information of the source that writes the events, this is only logged in
+        the very first event along with the `file_version` field.
+        """
+
+    def __init__(
+        self,
+        *,
+        wall_time: builtins.float | None = ...,
+        step: builtins.int | None = ...,
+        file_version: builtins.str | None = ...,
+        graph_def: builtins.bytes | None = ...,
+        summary: tensorflow.core.framework.summary_pb2.Summary | None = ...,
+        log_message: global___LogMessage | None = ...,
+        session_log: global___SessionLog | None = ...,
+        tagged_run_metadata: global___TaggedRunMetadata | None = ...,
+        meta_graph_def: builtins.bytes | None = ...,
+        source_metadata: global___SourceMetadata | None = ...,
+    ) -> None: ...
+    def HasField(
+        self,
+        field_name: typing.Literal[
+            "file_version",
+            b"file_version",
+            "graph_def",
+            b"graph_def",
+            "log_message",
+            b"log_message",
+            "meta_graph_def",
+            b"meta_graph_def",
+            "session_log",
+            b"session_log",
+            "source_metadata",
+            b"source_metadata",
+            "summary",
+            b"summary",
+            "tagged_run_metadata",
+            b"tagged_run_metadata",
+            "what",
+            b"what",
+        ],
+    ) -> builtins.bool: ...
+    def ClearField(
+        self,
+        field_name: typing.Literal[
+            "file_version",
+            b"file_version",
+            "graph_def",
+            b"graph_def",
+            "log_message",
+            b"log_message",
+            "meta_graph_def",
+            b"meta_graph_def",
+            "session_log",
+            b"session_log",
+            "source_metadata",
+            b"source_metadata",
+            "step",
+            b"step",
+            "summary",
+            b"summary",
+            "tagged_run_metadata",
+            b"tagged_run_metadata",
+            "wall_time",
+            b"wall_time",
+            "what",
+            b"what",
+        ],
+    ) -> None: ...
+    def WhichOneof(
+        self, oneof_group: typing.Literal["what", b"what"]
+    ) -> (
+        typing.Literal[
+            "file_version", "graph_def", "summary", "log_message", "session_log", "tagged_run_metadata", "meta_graph_def"
+        ]
+        | None
+    ): ...
+
+global___Event = Event
+
+@typing.final
+class SourceMetadata(google.protobuf.message.Message):
+    """Holds the information of the source that writes the events."""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    WRITER_FIELD_NUMBER: builtins.int
+    writer: builtins.str
+    """Low level name of the summary writer, such as
+    `tensorflow.core.util.events_writer`.
+    """
+    def __init__(self, *, writer: builtins.str | None = ...) -> None: ...
+    def ClearField(self, field_name: typing.Literal["writer", b"writer"]) -> None: ...
+
+global___SourceMetadata = SourceMetadata
+
+@typing.final
+class LogMessage(google.protobuf.message.Message):
+    """Protocol buffer used for logging messages to the events file.
+
+    This was theoretically used by the defunct tensorboard_logging module, which
+    has been removed; this message is now deprecated and should not be used.
+    """
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    class _Level:
+        ValueType = typing.NewType("ValueType", builtins.int)
+        V: typing_extensions.TypeAlias = ValueType
+
+    class _LevelEnumTypeWrapper(
+        google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[LogMessage._Level.ValueType], builtins.type
+    ):
+        DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
+        UNKNOWN: LogMessage._Level.ValueType  # 0
+        DEBUGGING: LogMessage._Level.ValueType  # 10
+        """Note: The logging level 10 cannot be named DEBUG. Some software
+        projects compile their C/C++ code with -DDEBUG in debug builds. So the
+        C++ code generated from this file should not have an identifier named
+        DEBUG.
+        """
+        INFO: LogMessage._Level.ValueType  # 20
+        WARN: LogMessage._Level.ValueType  # 30
+        ERROR: LogMessage._Level.ValueType  # 40
+        FATAL: LogMessage._Level.ValueType  # 50
+
+    class Level(_Level, metaclass=_LevelEnumTypeWrapper): ...
+    UNKNOWN: LogMessage.Level.ValueType  # 0
+    DEBUGGING: LogMessage.Level.ValueType  # 10
+    """Note: The logging level 10 cannot be named DEBUG. Some software
+    projects compile their C/C++ code with -DDEBUG in debug builds. So the
+    C++ code generated from this file should not have an identifier named
+    DEBUG.
+    """
+    INFO: LogMessage.Level.ValueType  # 20
+    WARN: LogMessage.Level.ValueType  # 30
+    ERROR: LogMessage.Level.ValueType  # 40
+    FATAL: LogMessage.Level.ValueType  # 50
+
+    LEVEL_FIELD_NUMBER: builtins.int
+    MESSAGE_FIELD_NUMBER: builtins.int
+    level: global___LogMessage.Level.ValueType
+    message: builtins.str
+    def __init__(
+        self, *, level: global___LogMessage.Level.ValueType | None = ..., message: builtins.str | None = ...
+    ) -> None: ...
+    def ClearField(self, field_name: typing.Literal["level", b"level", "message", b"message"]) -> None: ...
+
+global___LogMessage = LogMessage
+
+@typing.final
+class SessionLog(google.protobuf.message.Message):
+    """Protocol buffer used for logging session state."""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    class _SessionStatus:
+        ValueType = typing.NewType("ValueType", builtins.int)
+        V: typing_extensions.TypeAlias = ValueType
+
+    class _SessionStatusEnumTypeWrapper(
+        google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[SessionLog._SessionStatus.ValueType], builtins.type
+    ):
+        DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
+        STATUS_UNSPECIFIED: SessionLog._SessionStatus.ValueType  # 0
+        START: SessionLog._SessionStatus.ValueType  # 1
+        STOP: SessionLog._SessionStatus.ValueType  # 2
+        CHECKPOINT: SessionLog._SessionStatus.ValueType  # 3
+
+    class SessionStatus(_SessionStatus, metaclass=_SessionStatusEnumTypeWrapper): ...
+    STATUS_UNSPECIFIED: SessionLog.SessionStatus.ValueType  # 0
+    START: SessionLog.SessionStatus.ValueType  # 1
+    STOP: SessionLog.SessionStatus.ValueType  # 2
+    CHECKPOINT: SessionLog.SessionStatus.ValueType  # 3
+
+    STATUS_FIELD_NUMBER: builtins.int
+    CHECKPOINT_PATH_FIELD_NUMBER: builtins.int
+    MSG_FIELD_NUMBER: builtins.int
+    status: global___SessionLog.SessionStatus.ValueType
+    checkpoint_path: builtins.str
+    """This checkpoint_path contains both the path and filename."""
+    msg: builtins.str
+    def __init__(
+        self,
+        *,
+        status: global___SessionLog.SessionStatus.ValueType | None = ...,
+        checkpoint_path: builtins.str | None = ...,
+        msg: builtins.str | None = ...,
+    ) -> None: ...
+    def ClearField(
+        self, field_name: typing.Literal["checkpoint_path", b"checkpoint_path", "msg", b"msg", "status", b"status"]
+    ) -> None: ...
+
+global___SessionLog = SessionLog
+
+@typing.final
+class TaggedRunMetadata(google.protobuf.message.Message):
+    """For logging the metadata output for a single session.run() call."""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    TAG_FIELD_NUMBER: builtins.int
+    RUN_METADATA_FIELD_NUMBER: builtins.int
+    tag: builtins.str
+    """Tag name associated with this metadata."""
+    run_metadata: builtins.bytes
+    """Byte-encoded version of the `RunMetadata` proto in order to allow lazy
+    deserialization.
+    """
+    def __init__(self, *, tag: builtins.str | None = ..., run_metadata: builtins.bytes | None = ...) -> None: ...
+    def ClearField(self, field_name: typing.Literal["run_metadata", b"run_metadata", "tag", b"tag"]) -> None: ...
+
+global___TaggedRunMetadata = TaggedRunMetadata
+
+@typing.final
+class WatchdogConfig(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    TIMEOUT_MS_FIELD_NUMBER: builtins.int
+    timeout_ms: builtins.int
+    def __init__(self, *, timeout_ms: builtins.int | None = ...) -> None: ...
+    def ClearField(self, field_name: typing.Literal["timeout_ms", b"timeout_ms"]) -> None: ...
+
+global___WatchdogConfig = WatchdogConfig
+
+@typing.final
+class RequestedExitCode(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    EXIT_CODE_FIELD_NUMBER: builtins.int
+    exit_code: builtins.int
+    def __init__(self, *, exit_code: builtins.int | None = ...) -> None: ...
+    def ClearField(self, field_name: typing.Literal["exit_code", b"exit_code"]) -> None: ...
+
+global___RequestedExitCode = RequestedExitCode
+
+@typing.final
+class WorkerHeartbeatRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    SHUTDOWN_MODE_FIELD_NUMBER: builtins.int
+    WATCHDOG_CONFIG_FIELD_NUMBER: builtins.int
+    EXIT_CODE_FIELD_NUMBER: builtins.int
+    shutdown_mode: global___WorkerShutdownMode.ValueType
+    @property
+    def watchdog_config(self) -> global___WatchdogConfig: ...
+    @property
+    def exit_code(self) -> global___RequestedExitCode: ...
+    def __init__(
+        self,
+        *,
+        shutdown_mode: global___WorkerShutdownMode.ValueType | None = ...,
+        watchdog_config: global___WatchdogConfig | None = ...,
+        exit_code: global___RequestedExitCode | None = ...,
+    ) -> None: ...
+    def HasField(
+        self, field_name: typing.Literal["exit_code", b"exit_code", "watchdog_config", b"watchdog_config"]
+    ) -> builtins.bool: ...
+    def ClearField(
+        self,
+        field_name: typing.Literal[
+            "exit_code", b"exit_code", "shutdown_mode", b"shutdown_mode", "watchdog_config", b"watchdog_config"
+        ],
+    ) -> None: ...
+
+global___WorkerHeartbeatRequest = WorkerHeartbeatRequest
+
+@typing.final
+class WorkerHeartbeatResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    HEALTH_STATUS_FIELD_NUMBER: builtins.int
+    WORKER_LOG_FIELD_NUMBER: builtins.int
+    HOSTNAME_FIELD_NUMBER: builtins.int
+    health_status: global___WorkerHealth.ValueType
+    hostname: builtins.str
+    @property
+    def worker_log(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___Event]: ...
+    def __init__(
+        self,
+        *,
+        health_status: global___WorkerHealth.ValueType | None = ...,
+        worker_log: collections.abc.Iterable[global___Event] | None = ...,
+        hostname: builtins.str | None = ...,
+    ) -> None: ...
+    def ClearField(
+        self, field_name: typing.Literal["health_status", b"health_status", "hostname", b"hostname", "worker_log", b"worker_log"]
+    ) -> None: ...
+
+global___WorkerHeartbeatResponse = WorkerHeartbeatResponse

package/dist/typeshed-fallback/stubs/tensorflow/tensorflow/core/util/memmapped_file_system_pb2.pyi

@@ -0,0 +1,65 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+Copyright 2016 The TensorFlow Authors. All Rights Reserved.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+==============================================================================
+"""
+
+import builtins
+import collections.abc
+import typing
+
+import google.protobuf.descriptor
+import google.protobuf.internal.containers
+import google.protobuf.message
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+@typing.final
+class MemmappedFileSystemDirectoryElement(google.protobuf.message.Message):
+    """A message that describes one region of memmapped file."""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    OFFSET_FIELD_NUMBER: builtins.int
+    NAME_FIELD_NUMBER: builtins.int
+    LENGTH_FIELD_NUMBER: builtins.int
+    offset: builtins.int
+    name: builtins.str
+    length: builtins.int
+    def __init__(
+        self, *, offset: builtins.int | None = ..., name: builtins.str | None = ..., length: builtins.int | None = ...
+    ) -> None: ...
+    def ClearField(self, field_name: typing.Literal["length", b"length", "name", b"name", "offset", b"offset"]) -> None: ...
+
+global___MemmappedFileSystemDirectoryElement = MemmappedFileSystemDirectoryElement
+
+@typing.final
+class MemmappedFileSystemDirectory(google.protobuf.message.Message):
+    """A directory of regions in a memmapped file."""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    ELEMENT_FIELD_NUMBER: builtins.int
+    @property
+    def element(
+        self,
+    ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___MemmappedFileSystemDirectoryElement]: ...
+    def __init__(
+        self, *, element: collections.abc.Iterable[global___MemmappedFileSystemDirectoryElement] | None = ...
+    ) -> None: ...
+    def ClearField(self, field_name: typing.Literal["element", b"element"]) -> None: ...
+
+global___MemmappedFileSystemDirectory = MemmappedFileSystemDirectory