basedpyright 1.36.2 → 1.37.0-961d59bacff1958014fe4e9ed2ef3116fe2266e2

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

@@ -0,0 +1,113 @@
+"""
+@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 SaverDef(google.protobuf.message.Message):
+    """Protocol buffer representing the configuration of a Saver."""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    class _CheckpointFormatVersion:
+        ValueType = typing.NewType("ValueType", builtins.int)
+        V: typing_extensions.TypeAlias = ValueType
+
+    class _CheckpointFormatVersionEnumTypeWrapper(
+        google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[SaverDef._CheckpointFormatVersion.ValueType], builtins.type
+    ):
+        DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
+        LEGACY: SaverDef._CheckpointFormatVersion.ValueType  # 0
+        """Internal legacy format."""
+        V1: SaverDef._CheckpointFormatVersion.ValueType  # 1
+        """Deprecated format: tf.Saver() which works with tensorflow::table::Table."""
+        V2: SaverDef._CheckpointFormatVersion.ValueType  # 2
+        """Current format: more efficient."""
+
+    class CheckpointFormatVersion(_CheckpointFormatVersion, metaclass=_CheckpointFormatVersionEnumTypeWrapper):
+        """A version number that identifies a different on-disk checkpoint format.
+        Usually, each subclass of BaseSaverBuilder works with a particular
+        version/format.  However, it is possible that the same builder may be
+        upgraded to support a newer checkpoint format in the future.
+        """
+
+    LEGACY: SaverDef.CheckpointFormatVersion.ValueType  # 0
+    """Internal legacy format."""
+    V1: SaverDef.CheckpointFormatVersion.ValueType  # 1
+    """Deprecated format: tf.Saver() which works with tensorflow::table::Table."""
+    V2: SaverDef.CheckpointFormatVersion.ValueType  # 2
+    """Current format: more efficient."""
+
+    FILENAME_TENSOR_NAME_FIELD_NUMBER: builtins.int
+    SAVE_TENSOR_NAME_FIELD_NUMBER: builtins.int
+    RESTORE_OP_NAME_FIELD_NUMBER: builtins.int
+    MAX_TO_KEEP_FIELD_NUMBER: builtins.int
+    SHARDED_FIELD_NUMBER: builtins.int
+    KEEP_CHECKPOINT_EVERY_N_HOURS_FIELD_NUMBER: builtins.int
+    VERSION_FIELD_NUMBER: builtins.int
+    filename_tensor_name: builtins.str
+    """The name of the tensor in which to specify the filename when saving or
+    restoring a model checkpoint.
+    """
+    save_tensor_name: builtins.str
+    """The operation to run when saving a model checkpoint."""
+    restore_op_name: builtins.str
+    """The operation to run when restoring a model checkpoint."""
+    max_to_keep: builtins.int
+    """Maximum number of checkpoints to keep.  If 0, no checkpoints are deleted."""
+    sharded: builtins.bool
+    """Shard the save files, one per device that has Variable nodes."""
+    keep_checkpoint_every_n_hours: builtins.float
+    """How often to keep an additional checkpoint. If not specified, only the last
+    "max_to_keep" checkpoints are kept; if specified, in addition to keeping
+    the last "max_to_keep" checkpoints, an additional checkpoint will be kept
+    for every n hours of training.
+    """
+    version: global___SaverDef.CheckpointFormatVersion.ValueType
+    def __init__(
+        self,
+        *,
+        filename_tensor_name: builtins.str | None = ...,
+        save_tensor_name: builtins.str | None = ...,
+        restore_op_name: builtins.str | None = ...,
+        max_to_keep: builtins.int | None = ...,
+        sharded: builtins.bool | None = ...,
+        keep_checkpoint_every_n_hours: builtins.float | None = ...,
+        version: global___SaverDef.CheckpointFormatVersion.ValueType | None = ...,
+    ) -> None: ...
+    def ClearField(
+        self,
+        field_name: typing.Literal[
+            "filename_tensor_name",
+            b"filename_tensor_name",
+            "keep_checkpoint_every_n_hours",
+            b"keep_checkpoint_every_n_hours",
+            "max_to_keep",
+            b"max_to_keep",
+            "restore_op_name",
+            b"restore_op_name",
+            "save_tensor_name",
+            b"save_tensor_name",
+            "sharded",
+            b"sharded",
+            "version",
+            b"version",
+        ],
+    ) -> None: ...
+
+global___SaverDef = SaverDef

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

@@ -0,0 +1,275 @@
+"""
+@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
+import tensorflow.core.protobuf.data_service_pb2
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+@typing.final
+class DispatcherConfig(google.protobuf.message.Message):
+    """Configuration for a tf.data service DispatchServer.
+    Next id: 13
+    """
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    PORT_FIELD_NUMBER: builtins.int
+    PROTOCOL_FIELD_NUMBER: builtins.int
+    WORK_DIR_FIELD_NUMBER: builtins.int
+    FAULT_TOLERANT_MODE_FIELD_NUMBER: builtins.int
+    WORKER_ADDRESSES_FIELD_NUMBER: builtins.int
+    DEPLOYMENT_MODE_FIELD_NUMBER: builtins.int
+    JOB_GC_CHECK_INTERVAL_MS_FIELD_NUMBER: builtins.int
+    JOB_GC_TIMEOUT_MS_FIELD_NUMBER: builtins.int
+    GC_DYNAMIC_SHARDING_JOBS_FIELD_NUMBER: builtins.int
+    CLIENT_TIMEOUT_MS_FIELD_NUMBER: builtins.int
+    WORKER_TIMEOUT_MS_FIELD_NUMBER: builtins.int
+    WORKER_MAX_CONCURRENT_SNAPSHOTS_FIELD_NUMBER: builtins.int
+    port: builtins.int
+    """The port for the dispatcher to bind to. A value of 0 indicates that the
+    dispatcher may bind to any available port.
+    """
+    protocol: builtins.str
+    """The protocol for the dispatcher to use when connecting to workers."""
+    work_dir: builtins.str
+    """A work directory to use for storing dispatcher state, and for recovering
+    during restarts. The empty string indicates not to use any work directory.
+    """
+    fault_tolerant_mode: builtins.bool
+    """Whether to run in fault tolerant mode, where dispatcher state is saved
+    across restarts. Requires that `work_dir` is nonempty.
+    """
+    deployment_mode: tensorflow.core.protobuf.data_service_pb2.DeploymentMode.ValueType
+    """(Optional.) tf.data service deployment mode. Supported values are "REMOTE",
+    "COLOCATED", and "HYBRID". If unspecified, it is assumed to be "REMOTE".
+    """
+    job_gc_check_interval_ms: builtins.int
+    """How often the dispatcher should scan through to delete old and unused
+    jobs. A value of 0 indicates that the decision should be left up to the
+    runtime.
+    """
+    job_gc_timeout_ms: builtins.int
+    """How long a job needs to be unused before it becomes a candidate for garbage
+    collection. A value of -1 indicates that jobs should never be garbage
+    collected. A value of 0 indicates that the decision should be left up to
+    the runtime. Note: This does not apply to dynamic sharding unless users
+    explicitly opt-in by enabling `gc_dynamic_sharding_jobs` below.
+    """
+    gc_dynamic_sharding_jobs: builtins.bool
+    """Whether dynamically sharded jobs should be eligible for garbage collection.
+    These jobs are not garbage collected by default, since if a job is garbage
+    collected and then re-created, it will revisit all data from the start. If
+    revisiting data is acceptible and you want automatic reclamation of
+    iterator memory, set `gc_dynamic_sharding_jobs` to `true`.
+    """
+    client_timeout_ms: builtins.int
+    """How long to wait before garbage-collecting a client that hasn't
+    heartbeated to the dispatcher. A value of 0 indicates that the timeout
+    should be left to the runtime.
+    """
+    worker_timeout_ms: builtins.int
+    """How long to wait for a worker to heartbeat before considering it missing.
+    A value of 0 indicates that the timeout should be left to the runtime.
+    """
+    worker_max_concurrent_snapshots: builtins.int
+    """The maximum number of snapshots that a worker can concurrently process at a
+    given point in time. This is a tradeoff between worker resource usage and
+    snapshot wall time. A value of 0 indicates that the decision should be left
+    up to the runtime.
+    """
+    @property
+    def worker_addresses(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]:
+        """(Optional.) If the job uses auto-sharding, it needs to specify a fixed list
+        of worker addresses that will register with the dispatcher. The worker
+        addresses should be in the format "host" or "host:port", where "port" is an
+        integer, named port, or %port% to match any port.
+        """
+
+    def __init__(
+        self,
+        *,
+        port: builtins.int | None = ...,
+        protocol: builtins.str | None = ...,
+        work_dir: builtins.str | None = ...,
+        fault_tolerant_mode: builtins.bool | None = ...,
+        worker_addresses: collections.abc.Iterable[builtins.str] | None = ...,
+        deployment_mode: tensorflow.core.protobuf.data_service_pb2.DeploymentMode.ValueType | None = ...,
+        job_gc_check_interval_ms: builtins.int | None = ...,
+        job_gc_timeout_ms: builtins.int | None = ...,
+        gc_dynamic_sharding_jobs: builtins.bool | None = ...,
+        client_timeout_ms: builtins.int | None = ...,
+        worker_timeout_ms: builtins.int | None = ...,
+        worker_max_concurrent_snapshots: builtins.int | None = ...,
+    ) -> None: ...
+    def ClearField(
+        self,
+        field_name: typing.Literal[
+            "client_timeout_ms",
+            b"client_timeout_ms",
+            "deployment_mode",
+            b"deployment_mode",
+            "fault_tolerant_mode",
+            b"fault_tolerant_mode",
+            "gc_dynamic_sharding_jobs",
+            b"gc_dynamic_sharding_jobs",
+            "job_gc_check_interval_ms",
+            b"job_gc_check_interval_ms",
+            "job_gc_timeout_ms",
+            b"job_gc_timeout_ms",
+            "port",
+            b"port",
+            "protocol",
+            b"protocol",
+            "work_dir",
+            b"work_dir",
+            "worker_addresses",
+            b"worker_addresses",
+            "worker_max_concurrent_snapshots",
+            b"worker_max_concurrent_snapshots",
+            "worker_timeout_ms",
+            b"worker_timeout_ms",
+        ],
+    ) -> None: ...
+
+global___DispatcherConfig = DispatcherConfig
+
+@typing.final
+class WorkerConfig(google.protobuf.message.Message):
+    """Configuration for a tf.data service WorkerServer.
+    Next id: 14
+    """
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    PORT_FIELD_NUMBER: builtins.int
+    PROTOCOL_FIELD_NUMBER: builtins.int
+    DISPATCHER_ADDRESS_FIELD_NUMBER: builtins.int
+    WORKER_ADDRESS_FIELD_NUMBER: builtins.int
+    WORKER_TAGS_FIELD_NUMBER: builtins.int
+    HEARTBEAT_INTERVAL_MS_FIELD_NUMBER: builtins.int
+    DISPATCHER_TIMEOUT_MS_FIELD_NUMBER: builtins.int
+    DATA_TRANSFER_PROTOCOL_FIELD_NUMBER: builtins.int
+    DATA_TRANSFER_PORT_FIELD_NUMBER: builtins.int
+    DATA_TRANSFER_ADDRESS_FIELD_NUMBER: builtins.int
+    CROSS_TRAINER_CACHE_SIZE_BYTES_FIELD_NUMBER: builtins.int
+    SNAPSHOT_MAX_CHUNK_SIZE_BYTES_FIELD_NUMBER: builtins.int
+    SHUTDOWN_QUIET_PERIOD_MS_FIELD_NUMBER: builtins.int
+    port: builtins.int
+    """The port for the worker to bind to. A value of 0 indicates that the
+    worker may bind to any available port.
+    """
+    protocol: builtins.str
+    """The protocol for the worker to use when connecting to the dispatcher."""
+    dispatcher_address: builtins.str
+    """The address of the dispatcher to register with."""
+    worker_address: builtins.str
+    """The address of the worker server. The substring "%port%", if specified,
+    will be replaced with the worker's bound port. This is useful when the port
+    is set to `0`.
+    """
+    heartbeat_interval_ms: builtins.int
+    """How often the worker should heartbeat to the master. A value of 0 indicates
+    that the decision should be left up to the runtime.
+    """
+    dispatcher_timeout_ms: builtins.int
+    """How long to retry requests to the dispatcher before giving up and reporting
+    an error. A value of 0 indicates that the decision should be left up to the
+    runtime.
+    """
+    data_transfer_protocol: builtins.str
+    """If set, the name of an alternative data transfer protocol for which the
+    worker starts an additional server ("data transfer server"); the trainer
+    can then get data from this server. If not set, no such server is started,
+    and the trainer can only get data from the regular worker server over
+    `protocol`.
+    """
+    data_transfer_port: builtins.int
+    """If `data_transfer_protocol` is set, the port to which the data transfer
+    server binds. If set to `0`, the server binds to any available port.
+    """
+    data_transfer_address: builtins.str
+    """If `data_transfer_protocol` is set, the address of the data transfer
+    server. The substring "%dts_port%" can be used to represent -- and is
+    replaced with -- the bound port of the data transfer server; this is useful
+    when `data_transfer_port` is set to `0`.
+    """
+    cross_trainer_cache_size_bytes: builtins.int
+    """Maximum size of the cross-trainer cache in bytes. If enabled, make sure
+    your training job provides sufficient memory resources.
+    """
+    snapshot_max_chunk_size_bytes: builtins.int
+    """The maximum size of a distributed snapshot chunk file. A value of 0
+    indicates that the decision should be left up to the runtime.
+    """
+    shutdown_quiet_period_ms: builtins.int
+    """When shutting down a worker, how long to wait for the gRPC server to
+    process the final requests. This is used to achieve clean shutdown in unit
+    tests.
+    """
+    @property
+    def worker_tags(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]:
+        """Tags attached to the worker. This allows reading from selected workers.
+        For example, by applying a "COLOCATED" tag, tf.data service is able to read
+        from the local tf.data worker if one exists, then from off-TF-host workers,
+        to avoid cross-TF-host reads.
+        """
+
+    def __init__(
+        self,
+        *,
+        port: builtins.int | None = ...,
+        protocol: builtins.str | None = ...,
+        dispatcher_address: builtins.str | None = ...,
+        worker_address: builtins.str | None = ...,
+        worker_tags: collections.abc.Iterable[builtins.str] | None = ...,
+        heartbeat_interval_ms: builtins.int | None = ...,
+        dispatcher_timeout_ms: builtins.int | None = ...,
+        data_transfer_protocol: builtins.str | None = ...,
+        data_transfer_port: builtins.int | None = ...,
+        data_transfer_address: builtins.str | None = ...,
+        cross_trainer_cache_size_bytes: builtins.int | None = ...,
+        snapshot_max_chunk_size_bytes: builtins.int | None = ...,
+        shutdown_quiet_period_ms: builtins.int | None = ...,
+    ) -> None: ...
+    def ClearField(
+        self,
+        field_name: typing.Literal[
+            "cross_trainer_cache_size_bytes",
+            b"cross_trainer_cache_size_bytes",
+            "data_transfer_address",
+            b"data_transfer_address",
+            "data_transfer_port",
+            b"data_transfer_port",
+            "data_transfer_protocol",
+            b"data_transfer_protocol",
+            "dispatcher_address",
+            b"dispatcher_address",
+            "dispatcher_timeout_ms",
+            b"dispatcher_timeout_ms",
+            "heartbeat_interval_ms",
+            b"heartbeat_interval_ms",
+            "port",
+            b"port",
+            "protocol",
+            b"protocol",
+            "shutdown_quiet_period_ms",
+            b"shutdown_quiet_period_ms",
+            "snapshot_max_chunk_size_bytes",
+            b"snapshot_max_chunk_size_bytes",
+            "worker_address",
+            b"worker_address",
+            "worker_tags",
+            b"worker_tags",
+        ],
+    ) -> None: ...
+
+global___WorkerConfig = WorkerConfig

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

@@ -0,0 +1,164 @@
+"""
+@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
+import tensorflow.core.framework.tensor_pb2
+import tensorflow.core.framework.tensor_shape_pb2
+import tensorflow.core.framework.types_pb2
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+@typing.final
+class SnapshotRecord(google.protobuf.message.Message):
+    """Each SnapshotRecord represents one batch of pre-processed input data. A batch
+    consists of a list of tensors that we encode as TensorProtos. This message
+    doesn't store the structure of the batch.
+    """
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    TENSOR_FIELD_NUMBER: builtins.int
+    @property
+    def tensor(
+        self,
+    ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[
+        tensorflow.core.framework.tensor_pb2.TensorProto
+    ]: ...
+    def __init__(
+        self, *, tensor: collections.abc.Iterable[tensorflow.core.framework.tensor_pb2.TensorProto] | None = ...
+    ) -> None: ...
+    def ClearField(self, field_name: typing.Literal["tensor", b"tensor"]) -> None: ...
+
+global___SnapshotRecord = SnapshotRecord
+
+@typing.final
+class SnapshotMetadataRecord(google.protobuf.message.Message):
+    """This stores the metadata information present in each snapshot record."""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    GRAPH_HASH_FIELD_NUMBER: builtins.int
+    RUN_ID_FIELD_NUMBER: builtins.int
+    CREATION_TIMESTAMP_FIELD_NUMBER: builtins.int
+    VERSION_FIELD_NUMBER: builtins.int
+    DTYPE_FIELD_NUMBER: builtins.int
+    NUM_ELEMENTS_FIELD_NUMBER: builtins.int
+    FINALIZED_FIELD_NUMBER: builtins.int
+    graph_hash: builtins.str
+    """Stores the fingerprint of the graph that describes the dataset that is
+    snapshotted.
+    """
+    run_id: builtins.str
+    """Run ID that this snapshot corresponds to."""
+    creation_timestamp: builtins.int
+    """Time when we started creating this snapshot."""
+    version: builtins.int
+    """Version of the snapshot data file format."""
+    num_elements: builtins.int
+    """The number of elements in the snapshot."""
+    finalized: builtins.bool
+    @property
+    def dtype(
+        self,
+    ) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[tensorflow.core.framework.types_pb2.DataType.ValueType]:
+        """A list of tensor dtype corresponding to each element of the snapshot."""
+
+    def __init__(
+        self,
+        *,
+        graph_hash: builtins.str | None = ...,
+        run_id: builtins.str | None = ...,
+        creation_timestamp: builtins.int | None = ...,
+        version: builtins.int | None = ...,
+        dtype: collections.abc.Iterable[tensorflow.core.framework.types_pb2.DataType.ValueType] | None = ...,
+        num_elements: builtins.int | None = ...,
+        finalized: builtins.bool | None = ...,
+    ) -> None: ...
+    def ClearField(
+        self,
+        field_name: typing.Literal[
+            "creation_timestamp",
+            b"creation_timestamp",
+            "dtype",
+            b"dtype",
+            "finalized",
+            b"finalized",
+            "graph_hash",
+            b"graph_hash",
+            "num_elements",
+            b"num_elements",
+            "run_id",
+            b"run_id",
+            "version",
+            b"version",
+        ],
+    ) -> None: ...
+
+global___SnapshotMetadataRecord = SnapshotMetadataRecord
+
+@typing.final
+class TensorMetadata(google.protobuf.message.Message):
+    """Metadata for a single tensor in the Snapshot Record."""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    TENSOR_SHAPE_FIELD_NUMBER: builtins.int
+    TENSOR_SIZE_BYTES_FIELD_NUMBER: builtins.int
+    tensor_size_bytes: builtins.int
+    """Number of uncompressed bytes used to store the tensor representation."""
+    @property
+    def tensor_shape(self) -> tensorflow.core.framework.tensor_shape_pb2.TensorShapeProto: ...
+    def __init__(
+        self,
+        *,
+        tensor_shape: tensorflow.core.framework.tensor_shape_pb2.TensorShapeProto | None = ...,
+        tensor_size_bytes: builtins.int | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing.Literal["tensor_shape", b"tensor_shape"]) -> builtins.bool: ...
+    def ClearField(
+        self, field_name: typing.Literal["tensor_shape", b"tensor_shape", "tensor_size_bytes", b"tensor_size_bytes"]
+    ) -> None: ...
+
+global___TensorMetadata = TensorMetadata
+
+@typing.final
+class SnapshotTensorMetadata(google.protobuf.message.Message):
+    """Metadata for all the tensors in a Snapshot Record."""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    TENSOR_METADATA_FIELD_NUMBER: builtins.int
+    @property
+    def tensor_metadata(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___TensorMetadata]: ...
+    def __init__(self, *, tensor_metadata: collections.abc.Iterable[global___TensorMetadata] | None = ...) -> None: ...
+    def ClearField(self, field_name: typing.Literal["tensor_metadata", b"tensor_metadata"]) -> None: ...
+
+global___SnapshotTensorMetadata = SnapshotTensorMetadata
+
+@typing.final
+class DistributedSnapshotMetadata(google.protobuf.message.Message):
+    """Metadata for a `tf.data.Dataset` distributed snapshot."""
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    ELEMENT_SPEC_FIELD_NUMBER: builtins.int
+    COMPRESSION_FIELD_NUMBER: builtins.int
+    element_spec: builtins.bytes
+    """The element spec of the snapshotted dataset."""
+    compression: builtins.str
+    """Whether and how to compress the snapshot.  Supported values are defined in
+    `tsl::io::compression`.  In particular, an empty string specifies not to
+    compress.
+    """
+    def __init__(self, *, element_spec: builtins.bytes | None = ..., compression: builtins.str | None = ...) -> None: ...
+    def ClearField(self, field_name: typing.Literal["compression", b"compression", "element_spec", b"element_spec"]) -> None: ...
+
+global___DistributedSnapshotMetadata = DistributedSnapshotMetadata

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

@@ -0,0 +1,13 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+Add a dummy package name. Having no package like
+core/lib/core/error_codes.proto, or having tensorflow like
+tsl/protobuf/error_codes.proto, results in name collision errors in generated
+code for some users that use JS through J2CL.
+"""
+
+import google.protobuf.descriptor
+from tensorflow.tsl.protobuf.status_pb2 import StatusProto as StatusProto
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor