diff --git a/pyproject.toml b/pyproject.toml index b3e1a5821102..0c443d8d9639 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -9,11 +9,6 @@ requires-python = ">=3.9" # Minimum version to run tests, used by uv run line-length = 130 target-version = ["py310"] skip-magic-trailing-comma = true -# Exclude protobuf files because they have long line lengths -# that can't be autofixed. Like docstrings and import aliases. -# Ideally, we could configure Black to allow longer line lengths -# for just these files, but doesn't seem possible yet. -force-exclude = ".*_pb2.pyi" [tool.ruff] line-length = 130 diff --git a/stubs/protobuf/google/protobuf/any_pb2.pyi b/stubs/protobuf/google/protobuf/any_pb2.pyi index 2edfefbfcacd..c12c7e4a8133 100644 --- a/stubs/protobuf/google/protobuf/any_pb2.pyi +++ b/stubs/protobuf/google/protobuf/any_pb2.pyi @@ -166,12 +166,7 @@ class Any(google.protobuf.message.Message, google.protobuf.internal.well_known_t """ value: builtins.bytes """Must be a valid serialized protocol buffer of the above specified type.""" - def __init__( - self, - *, - type_url: builtins.str | None = ..., - value: builtins.bytes | None = ..., - ) -> None: ... + def __init__(self, *, type_url: builtins.str | None = ..., value: builtins.bytes | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["type_url", b"type_url", "value", b"value"]) -> None: ... global___Any = Any diff --git a/stubs/protobuf/google/protobuf/api_pb2.pyi b/stubs/protobuf/google/protobuf/api_pb2.pyi index f48ce3dd2de2..71e8ca47853f 100644 --- a/stubs/protobuf/google/protobuf/api_pb2.pyi +++ b/stubs/protobuf/google/protobuf/api_pb2.pyi @@ -123,7 +123,25 @@ class Api(google.protobuf.message.Message): syntax: google.protobuf.type_pb2.Syntax.ValueType | None = ..., ) -> None: ... def HasField(self, field_name: typing.Literal["source_context", b"source_context"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["methods", b"methods", "mixins", b"mixins", "name", b"name", "options", b"options", "source_context", b"source_context", "syntax", b"syntax", "version", b"version"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "methods", + b"methods", + "mixins", + b"mixins", + "name", + b"name", + "options", + b"options", + "source_context", + b"source_context", + "syntax", + b"syntax", + "version", + b"version", + ], + ) -> None: ... global___Api = Api @@ -167,7 +185,25 @@ class Method(google.protobuf.message.Message): options: collections.abc.Iterable[google.protobuf.type_pb2.Option] | None = ..., syntax: google.protobuf.type_pb2.Syntax.ValueType | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["name", b"name", "options", b"options", "request_streaming", b"request_streaming", "request_type_url", b"request_type_url", "response_streaming", b"response_streaming", "response_type_url", b"response_type_url", "syntax", b"syntax"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "name", + b"name", + "options", + b"options", + "request_streaming", + b"request_streaming", + "request_type_url", + b"request_type_url", + "response_streaming", + b"response_streaming", + "response_type_url", + b"response_type_url", + "syntax", + b"syntax", + ], + ) -> None: ... global___Method = Method @@ -263,12 +299,7 @@ class Mixin(google.protobuf.message.Message): """If non-empty specifies a path under which inherited HTTP paths are rooted. """ - def __init__( - self, - *, - name: builtins.str | None = ..., - root: builtins.str | None = ..., - ) -> None: ... + def __init__(self, *, name: builtins.str | None = ..., root: builtins.str | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["name", b"name", "root", b"root"]) -> None: ... global___Mixin = Mixin diff --git a/stubs/protobuf/google/protobuf/compiler/plugin_pb2.pyi b/stubs/protobuf/google/protobuf/compiler/plugin_pb2.pyi index 4ba99eafb811..a6f744c26074 100644 --- a/stubs/protobuf/google/protobuf/compiler/plugin_pb2.pyi +++ b/stubs/protobuf/google/protobuf/compiler/plugin_pb2.pyi @@ -58,8 +58,12 @@ class Version(google.protobuf.message.Message): patch: builtins.int | None = ..., suffix: builtins.str | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["major", b"major", "minor", b"minor", "patch", b"patch", "suffix", b"suffix"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["major", b"major", "minor", b"minor", "patch", b"patch", "suffix", b"suffix"]) -> None: ... + def HasField( + self, field_name: typing.Literal["major", b"major", "minor", b"minor", "patch", b"patch", "suffix", b"suffix"] + ) -> builtins.bool: ... + def ClearField( + self, field_name: typing.Literal["major", b"major", "minor", b"minor", "patch", b"patch", "suffix", b"suffix"] + ) -> None: ... global___Version = Version @@ -84,7 +88,9 @@ class CodeGeneratorRequest(google.protobuf.message.Message): """ @property - def proto_file(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[google.protobuf.descriptor_pb2.FileDescriptorProto]: + def proto_file( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[google.protobuf.descriptor_pb2.FileDescriptorProto]: """FileDescriptorProtos for all files in files_to_generate and everything they import. The files will appear in topological order, so each file appears before any file that imports it. @@ -107,7 +113,9 @@ class CodeGeneratorRequest(google.protobuf.message.Message): """ @property - def source_file_descriptors(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[google.protobuf.descriptor_pb2.FileDescriptorProto]: + def source_file_descriptors( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[google.protobuf.descriptor_pb2.FileDescriptorProto]: """File descriptors with all options, including source-retention options. These descriptors are only provided for the files listed in files_to_generate. @@ -126,8 +134,24 @@ class CodeGeneratorRequest(google.protobuf.message.Message): source_file_descriptors: collections.abc.Iterable[google.protobuf.descriptor_pb2.FileDescriptorProto] | None = ..., compiler_version: global___Version | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["compiler_version", b"compiler_version", "parameter", b"parameter"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["compiler_version", b"compiler_version", "file_to_generate", b"file_to_generate", "parameter", b"parameter", "proto_file", b"proto_file", "source_file_descriptors", b"source_file_descriptors"]) -> None: ... + def HasField( + self, field_name: typing.Literal["compiler_version", b"compiler_version", "parameter", b"parameter"] + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "compiler_version", + b"compiler_version", + "file_to_generate", + b"file_to_generate", + "parameter", + b"parameter", + "proto_file", + b"proto_file", + "source_file_descriptors", + b"source_file_descriptors", + ], + ) -> None: ... global___CodeGeneratorRequest = CodeGeneratorRequest @@ -141,7 +165,9 @@ class CodeGeneratorResponse(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _FeatureEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[CodeGeneratorResponse._Feature.ValueType], builtins.type): + class _FeatureEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[CodeGeneratorResponse._Feature.ValueType], builtins.type + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor FEATURE_NONE: CodeGeneratorResponse._Feature.ValueType # 0 FEATURE_PROTO3_OPTIONAL: CodeGeneratorResponse._Feature.ValueType # 1 @@ -233,8 +259,32 @@ class CodeGeneratorResponse(google.protobuf.message.Message): content: builtins.str | None = ..., generated_code_info: google.protobuf.descriptor_pb2.GeneratedCodeInfo | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["content", b"content", "generated_code_info", b"generated_code_info", "insertion_point", b"insertion_point", "name", b"name"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["content", b"content", "generated_code_info", b"generated_code_info", "insertion_point", b"insertion_point", "name", b"name"]) -> None: ... + def HasField( + self, + field_name: typing.Literal[ + "content", + b"content", + "generated_code_info", + b"generated_code_info", + "insertion_point", + b"insertion_point", + "name", + b"name", + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "content", + b"content", + "generated_code_info", + b"generated_code_info", + "insertion_point", + b"insertion_point", + "name", + b"name", + ], + ) -> None: ... ERROR_FIELD_NUMBER: builtins.int SUPPORTED_FEATURES_FIELD_NUMBER: builtins.int @@ -268,7 +318,9 @@ class CodeGeneratorResponse(google.protobuf.message.Message): effect for plugins that have FEATURE_SUPPORTS_EDITIONS set. """ @property - def file(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___CodeGeneratorResponse.File]: ... + def file( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___CodeGeneratorResponse.File]: ... def __init__( self, *, @@ -278,7 +330,33 @@ class CodeGeneratorResponse(google.protobuf.message.Message): maximum_edition: builtins.int | None = ..., file: collections.abc.Iterable[global___CodeGeneratorResponse.File] | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["error", b"error", "maximum_edition", b"maximum_edition", "minimum_edition", b"minimum_edition", "supported_features", b"supported_features"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["error", b"error", "file", b"file", "maximum_edition", b"maximum_edition", "minimum_edition", b"minimum_edition", "supported_features", b"supported_features"]) -> None: ... + def HasField( + self, + field_name: typing.Literal[ + "error", + b"error", + "maximum_edition", + b"maximum_edition", + "minimum_edition", + b"minimum_edition", + "supported_features", + b"supported_features", + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "error", + b"error", + "file", + b"file", + "maximum_edition", + b"maximum_edition", + "minimum_edition", + b"minimum_edition", + "supported_features", + b"supported_features", + ], + ) -> None: ... global___CodeGeneratorResponse = CodeGeneratorResponse diff --git a/stubs/protobuf/google/protobuf/descriptor_pb2.pyi b/stubs/protobuf/google/protobuf/descriptor_pb2.pyi index 6ff2ce84e0b2..d3bc182e4be3 100644 --- a/stubs/protobuf/google/protobuf/descriptor_pb2.pyi +++ b/stubs/protobuf/google/protobuf/descriptor_pb2.pyi @@ -114,11 +114,7 @@ class FileDescriptorSet(google.protobuf.message.Message): FILE_FIELD_NUMBER: builtins.int @property def file(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___FileDescriptorProto]: ... - def __init__( - self, - *, - file: collections.abc.Iterable[global___FileDescriptorProto] | None = ..., - ) -> None: ... + def __init__(self, *, file: collections.abc.Iterable[global___FileDescriptorProto] | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["file", b"file"]) -> None: ... global___FileDescriptorSet = FileDescriptorSet @@ -212,8 +208,54 @@ class FileDescriptorProto(google.protobuf.message.Message): syntax: builtins.str | None = ..., edition: global___Edition.ValueType | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["edition", b"edition", "name", b"name", "options", b"options", "package", b"package", "source_code_info", b"source_code_info", "syntax", b"syntax"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["dependency", b"dependency", "edition", b"edition", "enum_type", b"enum_type", "extension", b"extension", "message_type", b"message_type", "name", b"name", "options", b"options", "package", b"package", "public_dependency", b"public_dependency", "service", b"service", "source_code_info", b"source_code_info", "syntax", b"syntax", "weak_dependency", b"weak_dependency"]) -> None: ... + def HasField( + self, + field_name: typing.Literal[ + "edition", + b"edition", + "name", + b"name", + "options", + b"options", + "package", + b"package", + "source_code_info", + b"source_code_info", + "syntax", + b"syntax", + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "dependency", + b"dependency", + "edition", + b"edition", + "enum_type", + b"enum_type", + "extension", + b"extension", + "message_type", + b"message_type", + "name", + b"name", + "options", + b"options", + "package", + b"package", + "public_dependency", + b"public_dependency", + "service", + b"service", + "source_code_info", + b"source_code_info", + "syntax", + b"syntax", + "weak_dependency", + b"weak_dependency", + ], + ) -> None: ... global___FileDescriptorProto = FileDescriptorProto @@ -243,7 +285,9 @@ class DescriptorProto(google.protobuf.message.Message): end: builtins.int | None = ..., options: global___ExtensionRangeOptions | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["end", b"end", "options", b"options", "start", b"start"]) -> builtins.bool: ... + def HasField( + self, field_name: typing.Literal["end", b"end", "options", b"options", "start", b"start"] + ) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["end", b"end", "options", b"options", "start", b"start"]) -> None: ... @typing.final @@ -261,12 +305,7 @@ class DescriptorProto(google.protobuf.message.Message): """Inclusive.""" end: builtins.int """Exclusive.""" - def __init__( - self, - *, - start: builtins.int | None = ..., - end: builtins.int | None = ..., - ) -> None: ... + def __init__(self, *, start: builtins.int | None = ..., end: builtins.int | None = ...) -> None: ... def HasField(self, field_name: typing.Literal["end", b"end", "start", b"start"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["end", b"end", "start", b"start"]) -> None: ... @@ -290,13 +329,19 @@ class DescriptorProto(google.protobuf.message.Message): @property def enum_type(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___EnumDescriptorProto]: ... @property - def extension_range(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___DescriptorProto.ExtensionRange]: ... + def extension_range( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___DescriptorProto.ExtensionRange]: ... @property - def oneof_decl(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___OneofDescriptorProto]: ... + def oneof_decl( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___OneofDescriptorProto]: ... @property def options(self) -> global___MessageOptions: ... @property - def reserved_range(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___DescriptorProto.ReservedRange]: ... + def reserved_range( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___DescriptorProto.ReservedRange]: ... @property def reserved_name(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]: """Reserved field names, which may not be used by fields in the same message. @@ -318,7 +363,31 @@ class DescriptorProto(google.protobuf.message.Message): reserved_name: collections.abc.Iterable[builtins.str] | None = ..., ) -> None: ... def HasField(self, field_name: typing.Literal["name", b"name", "options", b"options"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["enum_type", b"enum_type", "extension", b"extension", "extension_range", b"extension_range", "field", b"field", "name", b"name", "nested_type", b"nested_type", "oneof_decl", b"oneof_decl", "options", b"options", "reserved_name", b"reserved_name", "reserved_range", b"reserved_range"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "enum_type", + b"enum_type", + "extension", + b"extension", + "extension_range", + b"extension_range", + "field", + b"field", + "name", + b"name", + "nested_type", + b"nested_type", + "oneof_decl", + b"oneof_decl", + "options", + b"options", + "reserved_name", + b"reserved_name", + "reserved_range", + b"reserved_range", + ], + ) -> None: ... global___DescriptorProto = DescriptorProto @@ -330,7 +399,10 @@ class ExtensionRangeOptions(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _VerificationStateEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[ExtensionRangeOptions._VerificationState.ValueType], builtins.type): + class _VerificationStateEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[ExtensionRangeOptions._VerificationState.ValueType], + builtins.type, + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor DECLARATION: ExtensionRangeOptions._VerificationState.ValueType # 0 """All the extensions of the range must be declared.""" @@ -381,8 +453,18 @@ class ExtensionRangeOptions(google.protobuf.message.Message): reserved: builtins.bool | None = ..., repeated: builtins.bool | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["full_name", b"full_name", "number", b"number", "repeated", b"repeated", "reserved", b"reserved", "type", b"type"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["full_name", b"full_name", "number", b"number", "repeated", b"repeated", "reserved", b"reserved", "type", b"type"]) -> None: ... + def HasField( + self, + field_name: typing.Literal[ + "full_name", b"full_name", "number", b"number", "repeated", b"repeated", "reserved", b"reserved", "type", b"type" + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "full_name", b"full_name", "number", b"number", "repeated", b"repeated", "reserved", b"reserved", "type", b"type" + ], + ) -> None: ... UNINTERPRETED_OPTION_FIELD_NUMBER: builtins.int DECLARATION_FIELD_NUMBER: builtins.int @@ -394,11 +476,15 @@ class ExtensionRangeOptions(google.protobuf.message.Message): are marked as UNVERIFIED. """ @property - def uninterpreted_option(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___UninterpretedOption]: + def uninterpreted_option( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___UninterpretedOption]: """The parser stores options it doesn't recognize here. See above.""" @property - def declaration(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___ExtensionRangeOptions.Declaration]: + def declaration( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___ExtensionRangeOptions.Declaration]: """For external users: DO NOT USE. We are in the process of open sourcing extension declaration and executing internal cleanups before it can be used externally. @@ -417,7 +503,19 @@ class ExtensionRangeOptions(google.protobuf.message.Message): verification: global___ExtensionRangeOptions.VerificationState.ValueType | None = ..., ) -> None: ... def HasField(self, field_name: typing.Literal["features", b"features", "verification", b"verification"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["declaration", b"declaration", "features", b"features", "uninterpreted_option", b"uninterpreted_option", "verification", b"verification"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "declaration", + b"declaration", + "features", + b"features", + "uninterpreted_option", + b"uninterpreted_option", + "verification", + b"verification", + ], + ) -> None: ... global___ExtensionRangeOptions = ExtensionRangeOptions @@ -431,7 +529,9 @@ class FieldDescriptorProto(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _TypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[FieldDescriptorProto._Type.ValueType], builtins.type): + class _TypeEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[FieldDescriptorProto._Type.ValueType], builtins.type + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor TYPE_DOUBLE: FieldDescriptorProto._Type.ValueType # 1 """0 is reserved for errors. @@ -514,7 +614,9 @@ class FieldDescriptorProto(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _LabelEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[FieldDescriptorProto._Label.ValueType], builtins.type): + class _LabelEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[FieldDescriptorProto._Label.ValueType], builtins.type + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor LABEL_OPTIONAL: FieldDescriptorProto._Label.ValueType # 1 """0 is reserved for errors""" @@ -620,8 +722,60 @@ class FieldDescriptorProto(google.protobuf.message.Message): options: global___FieldOptions | None = ..., proto3_optional: builtins.bool | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["default_value", b"default_value", "extendee", b"extendee", "json_name", b"json_name", "label", b"label", "name", b"name", "number", b"number", "oneof_index", b"oneof_index", "options", b"options", "proto3_optional", b"proto3_optional", "type", b"type", "type_name", b"type_name"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["default_value", b"default_value", "extendee", b"extendee", "json_name", b"json_name", "label", b"label", "name", b"name", "number", b"number", "oneof_index", b"oneof_index", "options", b"options", "proto3_optional", b"proto3_optional", "type", b"type", "type_name", b"type_name"]) -> None: ... + def HasField( + self, + field_name: typing.Literal[ + "default_value", + b"default_value", + "extendee", + b"extendee", + "json_name", + b"json_name", + "label", + b"label", + "name", + b"name", + "number", + b"number", + "oneof_index", + b"oneof_index", + "options", + b"options", + "proto3_optional", + b"proto3_optional", + "type", + b"type", + "type_name", + b"type_name", + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "default_value", + b"default_value", + "extendee", + b"extendee", + "json_name", + b"json_name", + "label", + b"label", + "name", + b"name", + "number", + b"number", + "oneof_index", + b"oneof_index", + "options", + b"options", + "proto3_optional", + b"proto3_optional", + "type", + b"type", + "type_name", + b"type_name", + ], + ) -> None: ... global___FieldDescriptorProto = FieldDescriptorProto @@ -636,12 +790,7 @@ class OneofDescriptorProto(google.protobuf.message.Message): name: builtins.str @property def options(self) -> global___OneofOptions: ... - def __init__( - self, - *, - name: builtins.str | None = ..., - options: global___OneofOptions | None = ..., - ) -> None: ... + def __init__(self, *, name: builtins.str | None = ..., options: global___OneofOptions | None = ...) -> None: ... def HasField(self, field_name: typing.Literal["name", b"name", "options", b"options"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["name", b"name", "options", b"options"]) -> None: ... @@ -671,12 +820,7 @@ class EnumDescriptorProto(google.protobuf.message.Message): """Inclusive.""" end: builtins.int """Inclusive.""" - def __init__( - self, - *, - start: builtins.int | None = ..., - end: builtins.int | None = ..., - ) -> None: ... + def __init__(self, *, start: builtins.int | None = ..., end: builtins.int | None = ...) -> None: ... def HasField(self, field_name: typing.Literal["end", b"end", "start", b"start"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["end", b"end", "start", b"start"]) -> None: ... @@ -691,7 +835,9 @@ class EnumDescriptorProto(google.protobuf.message.Message): @property def options(self) -> global___EnumOptions: ... @property - def reserved_range(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___EnumDescriptorProto.EnumReservedRange]: + def reserved_range( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___EnumDescriptorProto.EnumReservedRange]: """Range of reserved numeric values. Reserved numeric values may not be used by enum values in the same enum declaration. Reserved ranges may not overlap. @@ -713,7 +859,21 @@ class EnumDescriptorProto(google.protobuf.message.Message): reserved_name: collections.abc.Iterable[builtins.str] | None = ..., ) -> None: ... def HasField(self, field_name: typing.Literal["name", b"name", "options", b"options"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["name", b"name", "options", b"options", "reserved_name", b"reserved_name", "reserved_range", b"reserved_range", "value", b"value"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "name", + b"name", + "options", + b"options", + "reserved_name", + b"reserved_name", + "reserved_range", + b"reserved_range", + "value", + b"value", + ], + ) -> None: ... global___EnumDescriptorProto = EnumDescriptorProto @@ -737,7 +897,9 @@ class EnumValueDescriptorProto(google.protobuf.message.Message): number: builtins.int | None = ..., options: global___EnumValueOptions | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["name", b"name", "number", b"number", "options", b"options"]) -> builtins.bool: ... + def HasField( + self, field_name: typing.Literal["name", b"name", "number", b"number", "options", b"options"] + ) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["name", b"name", "number", b"number", "options", b"options"]) -> None: ... global___EnumValueDescriptorProto = EnumValueDescriptorProto @@ -802,8 +964,40 @@ class MethodDescriptorProto(google.protobuf.message.Message): client_streaming: builtins.bool | None = ..., server_streaming: builtins.bool | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["client_streaming", b"client_streaming", "input_type", b"input_type", "name", b"name", "options", b"options", "output_type", b"output_type", "server_streaming", b"server_streaming"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["client_streaming", b"client_streaming", "input_type", b"input_type", "name", b"name", "options", b"options", "output_type", b"output_type", "server_streaming", b"server_streaming"]) -> None: ... + def HasField( + self, + field_name: typing.Literal[ + "client_streaming", + b"client_streaming", + "input_type", + b"input_type", + "name", + b"name", + "options", + b"options", + "output_type", + b"output_type", + "server_streaming", + b"server_streaming", + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "client_streaming", + b"client_streaming", + "input_type", + b"input_type", + "name", + b"name", + "options", + b"options", + "output_type", + b"output_type", + "server_streaming", + b"server_streaming", + ], + ) -> None: ... global___MethodDescriptorProto = MethodDescriptorProto @@ -845,7 +1039,9 @@ class FileOptions(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _OptimizeModeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[FileOptions._OptimizeMode.ValueType], builtins.type): + class _OptimizeModeEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[FileOptions._OptimizeMode.ValueType], builtins.type + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor SPEED: FileOptions._OptimizeMode.ValueType # 1 """Generate complete code for parsing, serialization,""" @@ -996,7 +1192,9 @@ class FileOptions(google.protobuf.message.Message): """ @property - def uninterpreted_option(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___UninterpretedOption]: + def uninterpreted_option( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___UninterpretedOption]: """The parser stores options it doesn't recognize here. See the documentation for the "Options" section above. """ @@ -1026,8 +1224,98 @@ class FileOptions(google.protobuf.message.Message): features: global___FeatureSet | None = ..., uninterpreted_option: collections.abc.Iterable[global___UninterpretedOption] | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["cc_enable_arenas", b"cc_enable_arenas", "cc_generic_services", b"cc_generic_services", "csharp_namespace", b"csharp_namespace", "deprecated", b"deprecated", "features", b"features", "go_package", b"go_package", "java_generate_equals_and_hash", b"java_generate_equals_and_hash", "java_generic_services", b"java_generic_services", "java_multiple_files", b"java_multiple_files", "java_outer_classname", b"java_outer_classname", "java_package", b"java_package", "java_string_check_utf8", b"java_string_check_utf8", "objc_class_prefix", b"objc_class_prefix", "optimize_for", b"optimize_for", "php_class_prefix", b"php_class_prefix", "php_metadata_namespace", b"php_metadata_namespace", "php_namespace", b"php_namespace", "py_generic_services", b"py_generic_services", "ruby_package", b"ruby_package", "swift_prefix", b"swift_prefix"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["cc_enable_arenas", b"cc_enable_arenas", "cc_generic_services", b"cc_generic_services", "csharp_namespace", b"csharp_namespace", "deprecated", b"deprecated", "features", b"features", "go_package", b"go_package", "java_generate_equals_and_hash", b"java_generate_equals_and_hash", "java_generic_services", b"java_generic_services", "java_multiple_files", b"java_multiple_files", "java_outer_classname", b"java_outer_classname", "java_package", b"java_package", "java_string_check_utf8", b"java_string_check_utf8", "objc_class_prefix", b"objc_class_prefix", "optimize_for", b"optimize_for", "php_class_prefix", b"php_class_prefix", "php_metadata_namespace", b"php_metadata_namespace", "php_namespace", b"php_namespace", "py_generic_services", b"py_generic_services", "ruby_package", b"ruby_package", "swift_prefix", b"swift_prefix", "uninterpreted_option", b"uninterpreted_option"]) -> None: ... + def HasField( + self, + field_name: typing.Literal[ + "cc_enable_arenas", + b"cc_enable_arenas", + "cc_generic_services", + b"cc_generic_services", + "csharp_namespace", + b"csharp_namespace", + "deprecated", + b"deprecated", + "features", + b"features", + "go_package", + b"go_package", + "java_generate_equals_and_hash", + b"java_generate_equals_and_hash", + "java_generic_services", + b"java_generic_services", + "java_multiple_files", + b"java_multiple_files", + "java_outer_classname", + b"java_outer_classname", + "java_package", + b"java_package", + "java_string_check_utf8", + b"java_string_check_utf8", + "objc_class_prefix", + b"objc_class_prefix", + "optimize_for", + b"optimize_for", + "php_class_prefix", + b"php_class_prefix", + "php_metadata_namespace", + b"php_metadata_namespace", + "php_namespace", + b"php_namespace", + "py_generic_services", + b"py_generic_services", + "ruby_package", + b"ruby_package", + "swift_prefix", + b"swift_prefix", + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "cc_enable_arenas", + b"cc_enable_arenas", + "cc_generic_services", + b"cc_generic_services", + "csharp_namespace", + b"csharp_namespace", + "deprecated", + b"deprecated", + "features", + b"features", + "go_package", + b"go_package", + "java_generate_equals_and_hash", + b"java_generate_equals_and_hash", + "java_generic_services", + b"java_generic_services", + "java_multiple_files", + b"java_multiple_files", + "java_outer_classname", + b"java_outer_classname", + "java_package", + b"java_package", + "java_string_check_utf8", + b"java_string_check_utf8", + "objc_class_prefix", + b"objc_class_prefix", + "optimize_for", + b"optimize_for", + "php_class_prefix", + b"php_class_prefix", + "php_metadata_namespace", + b"php_metadata_namespace", + "php_namespace", + b"php_namespace", + "py_generic_services", + b"py_generic_services", + "ruby_package", + b"ruby_package", + "swift_prefix", + b"swift_prefix", + "uninterpreted_option", + b"uninterpreted_option", + ], + ) -> None: ... global___FileOptions = FileOptions @@ -1117,7 +1405,9 @@ class MessageOptions(google.protobuf.message.Message): """ @property - def uninterpreted_option(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___UninterpretedOption]: + def uninterpreted_option( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___UninterpretedOption]: """The parser stores options it doesn't recognize here. See above.""" def __init__( @@ -1131,8 +1421,42 @@ class MessageOptions(google.protobuf.message.Message): features: global___FeatureSet | None = ..., uninterpreted_option: collections.abc.Iterable[global___UninterpretedOption] | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["deprecated", b"deprecated", "deprecated_legacy_json_field_conflicts", b"deprecated_legacy_json_field_conflicts", "features", b"features", "map_entry", b"map_entry", "message_set_wire_format", b"message_set_wire_format", "no_standard_descriptor_accessor", b"no_standard_descriptor_accessor"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["deprecated", b"deprecated", "deprecated_legacy_json_field_conflicts", b"deprecated_legacy_json_field_conflicts", "features", b"features", "map_entry", b"map_entry", "message_set_wire_format", b"message_set_wire_format", "no_standard_descriptor_accessor", b"no_standard_descriptor_accessor", "uninterpreted_option", b"uninterpreted_option"]) -> None: ... + def HasField( + self, + field_name: typing.Literal[ + "deprecated", + b"deprecated", + "deprecated_legacy_json_field_conflicts", + b"deprecated_legacy_json_field_conflicts", + "features", + b"features", + "map_entry", + b"map_entry", + "message_set_wire_format", + b"message_set_wire_format", + "no_standard_descriptor_accessor", + b"no_standard_descriptor_accessor", + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "deprecated", + b"deprecated", + "deprecated_legacy_json_field_conflicts", + b"deprecated_legacy_json_field_conflicts", + "features", + b"features", + "map_entry", + b"map_entry", + "message_set_wire_format", + b"message_set_wire_format", + "no_standard_descriptor_accessor", + b"no_standard_descriptor_accessor", + "uninterpreted_option", + b"uninterpreted_option", + ], + ) -> None: ... global___MessageOptions = MessageOptions @@ -1144,7 +1468,9 @@ class FieldOptions(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _CTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[FieldOptions._CType.ValueType], builtins.type): + class _CTypeEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[FieldOptions._CType.ValueType], builtins.type + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor STRING: FieldOptions._CType.ValueType # 0 """Default mode.""" @@ -1175,7 +1501,9 @@ class FieldOptions(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _JSTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[FieldOptions._JSType.ValueType], builtins.type): + class _JSTypeEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[FieldOptions._JSType.ValueType], builtins.type + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor JS_NORMAL: FieldOptions._JSType.ValueType # 0 """Use the default type.""" @@ -1196,7 +1524,9 @@ class FieldOptions(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _OptionRetentionEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[FieldOptions._OptionRetention.ValueType], builtins.type): + class _OptionRetentionEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[FieldOptions._OptionRetention.ValueType], builtins.type + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor RETENTION_UNKNOWN: FieldOptions._OptionRetention.ValueType # 0 RETENTION_RUNTIME: FieldOptions._OptionRetention.ValueType # 1 @@ -1213,7 +1543,9 @@ class FieldOptions(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _OptionTargetTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[FieldOptions._OptionTargetType.ValueType], builtins.type): + class _OptionTargetTypeEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[FieldOptions._OptionTargetType.ValueType], builtins.type + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor TARGET_TYPE_UNKNOWN: FieldOptions._OptionTargetType.ValueType # 0 TARGET_TYPE_FILE: FieldOptions._OptionTargetType.ValueType # 1 @@ -1252,12 +1584,7 @@ class FieldOptions(google.protobuf.message.Message): edition: global___Edition.ValueType value: builtins.str """Textproto value.""" - def __init__( - self, - *, - edition: global___Edition.ValueType | None = ..., - value: builtins.str | None = ..., - ) -> None: ... + def __init__(self, *, edition: global___Edition.ValueType | None = ..., value: builtins.str | None = ...) -> None: ... def HasField(self, field_name: typing.Literal["edition", b"edition", "value", b"value"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["edition", b"edition", "value", b"value"]) -> None: ... @@ -1297,8 +1624,32 @@ class FieldOptions(google.protobuf.message.Message): deprecation_warning: builtins.str | None = ..., edition_removed: global___Edition.ValueType | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["deprecation_warning", b"deprecation_warning", "edition_deprecated", b"edition_deprecated", "edition_introduced", b"edition_introduced", "edition_removed", b"edition_removed"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["deprecation_warning", b"deprecation_warning", "edition_deprecated", b"edition_deprecated", "edition_introduced", b"edition_introduced", "edition_removed", b"edition_removed"]) -> None: ... + def HasField( + self, + field_name: typing.Literal[ + "deprecation_warning", + b"deprecation_warning", + "edition_deprecated", + b"edition_deprecated", + "edition_introduced", + b"edition_introduced", + "edition_removed", + b"edition_removed", + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "deprecation_warning", + b"deprecation_warning", + "edition_deprecated", + b"edition_deprecated", + "edition_introduced", + b"edition_introduced", + "edition_removed", + b"edition_removed", + ], + ) -> None: ... CTYPE_FIELD_NUMBER: builtins.int PACKED_FIELD_NUMBER: builtins.int @@ -1388,9 +1739,13 @@ class FieldOptions(google.protobuf.message.Message): """ retention: global___FieldOptions.OptionRetention.ValueType @property - def targets(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[global___FieldOptions.OptionTargetType.ValueType]: ... + def targets( + self, + ) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[global___FieldOptions.OptionTargetType.ValueType]: ... @property - def edition_defaults(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___FieldOptions.EditionDefault]: ... + def edition_defaults( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___FieldOptions.EditionDefault]: ... @property def features(self) -> global___FeatureSet: """Any features defined in the specific edition. @@ -1402,7 +1757,9 @@ class FieldOptions(google.protobuf.message.Message): @property def feature_support(self) -> global___FieldOptions.FeatureSupport: ... @property - def uninterpreted_option(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___UninterpretedOption]: + def uninterpreted_option( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___UninterpretedOption]: """The parser stores options it doesn't recognize here. See above.""" def __init__( @@ -1423,8 +1780,66 @@ class FieldOptions(google.protobuf.message.Message): feature_support: global___FieldOptions.FeatureSupport | None = ..., uninterpreted_option: collections.abc.Iterable[global___UninterpretedOption] | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["ctype", b"ctype", "debug_redact", b"debug_redact", "deprecated", b"deprecated", "feature_support", b"feature_support", "features", b"features", "jstype", b"jstype", "lazy", b"lazy", "packed", b"packed", "retention", b"retention", "unverified_lazy", b"unverified_lazy", "weak", b"weak"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["ctype", b"ctype", "debug_redact", b"debug_redact", "deprecated", b"deprecated", "edition_defaults", b"edition_defaults", "feature_support", b"feature_support", "features", b"features", "jstype", b"jstype", "lazy", b"lazy", "packed", b"packed", "retention", b"retention", "targets", b"targets", "uninterpreted_option", b"uninterpreted_option", "unverified_lazy", b"unverified_lazy", "weak", b"weak"]) -> None: ... + def HasField( + self, + field_name: typing.Literal[ + "ctype", + b"ctype", + "debug_redact", + b"debug_redact", + "deprecated", + b"deprecated", + "feature_support", + b"feature_support", + "features", + b"features", + "jstype", + b"jstype", + "lazy", + b"lazy", + "packed", + b"packed", + "retention", + b"retention", + "unverified_lazy", + b"unverified_lazy", + "weak", + b"weak", + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "ctype", + b"ctype", + "debug_redact", + b"debug_redact", + "deprecated", + b"deprecated", + "edition_defaults", + b"edition_defaults", + "feature_support", + b"feature_support", + "features", + b"features", + "jstype", + b"jstype", + "lazy", + b"lazy", + "packed", + b"packed", + "retention", + b"retention", + "targets", + b"targets", + "uninterpreted_option", + b"uninterpreted_option", + "unverified_lazy", + b"unverified_lazy", + "weak", + b"weak", + ], + ) -> None: ... global___FieldOptions = FieldOptions @@ -1443,7 +1858,9 @@ class OneofOptions(google.protobuf.message.Message): """ @property - def uninterpreted_option(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___UninterpretedOption]: + def uninterpreted_option( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___UninterpretedOption]: """The parser stores options it doesn't recognize here. See above.""" def __init__( @@ -1453,7 +1870,9 @@ class OneofOptions(google.protobuf.message.Message): uninterpreted_option: collections.abc.Iterable[global___UninterpretedOption] | None = ..., ) -> None: ... def HasField(self, field_name: typing.Literal["features", b"features"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["features", b"features", "uninterpreted_option", b"uninterpreted_option"]) -> None: ... + def ClearField( + self, field_name: typing.Literal["features", b"features", "uninterpreted_option", b"uninterpreted_option"] + ) -> None: ... global___OneofOptions = OneofOptions @@ -1493,7 +1912,9 @@ class EnumOptions(google.protobuf.message.Message): """ @property - def uninterpreted_option(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___UninterpretedOption]: + def uninterpreted_option( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___UninterpretedOption]: """The parser stores options it doesn't recognize here. See above.""" def __init__( @@ -1505,8 +1926,34 @@ class EnumOptions(google.protobuf.message.Message): features: global___FeatureSet | None = ..., uninterpreted_option: collections.abc.Iterable[global___UninterpretedOption] | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["allow_alias", b"allow_alias", "deprecated", b"deprecated", "deprecated_legacy_json_field_conflicts", b"deprecated_legacy_json_field_conflicts", "features", b"features"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["allow_alias", b"allow_alias", "deprecated", b"deprecated", "deprecated_legacy_json_field_conflicts", b"deprecated_legacy_json_field_conflicts", "features", b"features", "uninterpreted_option", b"uninterpreted_option"]) -> None: ... + def HasField( + self, + field_name: typing.Literal[ + "allow_alias", + b"allow_alias", + "deprecated", + b"deprecated", + "deprecated_legacy_json_field_conflicts", + b"deprecated_legacy_json_field_conflicts", + "features", + b"features", + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "allow_alias", + b"allow_alias", + "deprecated", + b"deprecated", + "deprecated_legacy_json_field_conflicts", + b"deprecated_legacy_json_field_conflicts", + "features", + b"features", + "uninterpreted_option", + b"uninterpreted_option", + ], + ) -> None: ... global___EnumOptions = EnumOptions @@ -1543,7 +1990,9 @@ class EnumValueOptions(google.protobuf.message.Message): """Information about the support window of a feature value.""" @property - def uninterpreted_option(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___UninterpretedOption]: + def uninterpreted_option( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___UninterpretedOption]: """The parser stores options it doesn't recognize here. See above.""" def __init__( @@ -1555,8 +2004,34 @@ class EnumValueOptions(google.protobuf.message.Message): feature_support: global___FieldOptions.FeatureSupport | None = ..., uninterpreted_option: collections.abc.Iterable[global___UninterpretedOption] | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["debug_redact", b"debug_redact", "deprecated", b"deprecated", "feature_support", b"feature_support", "features", b"features"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["debug_redact", b"debug_redact", "deprecated", b"deprecated", "feature_support", b"feature_support", "features", b"features", "uninterpreted_option", b"uninterpreted_option"]) -> None: ... + def HasField( + self, + field_name: typing.Literal[ + "debug_redact", + b"debug_redact", + "deprecated", + b"deprecated", + "feature_support", + b"feature_support", + "features", + b"features", + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "debug_redact", + b"debug_redact", + "deprecated", + b"deprecated", + "feature_support", + b"feature_support", + "features", + b"features", + "uninterpreted_option", + b"uninterpreted_option", + ], + ) -> None: ... global___EnumValueOptions = EnumValueOptions @@ -1587,7 +2062,9 @@ class ServiceOptions(google.protobuf.message.Message): """ @property - def uninterpreted_option(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___UninterpretedOption]: + def uninterpreted_option( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___UninterpretedOption]: """The parser stores options it doesn't recognize here. See above.""" def __init__( @@ -1598,7 +2075,12 @@ class ServiceOptions(google.protobuf.message.Message): uninterpreted_option: collections.abc.Iterable[global___UninterpretedOption] | None = ..., ) -> None: ... def HasField(self, field_name: typing.Literal["deprecated", b"deprecated", "features", b"features"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["deprecated", b"deprecated", "features", b"features", "uninterpreted_option", b"uninterpreted_option"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "deprecated", b"deprecated", "features", b"features", "uninterpreted_option", b"uninterpreted_option" + ], + ) -> None: ... global___ServiceOptions = ServiceOptions @@ -1610,7 +2092,9 @@ class MethodOptions(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _IdempotencyLevelEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[MethodOptions._IdempotencyLevel.ValueType], builtins.type): + class _IdempotencyLevelEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[MethodOptions._IdempotencyLevel.ValueType], builtins.type + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor IDEMPOTENCY_UNKNOWN: MethodOptions._IdempotencyLevel.ValueType # 0 NO_SIDE_EFFECTS: MethodOptions._IdempotencyLevel.ValueType # 1 @@ -1655,7 +2139,9 @@ class MethodOptions(google.protobuf.message.Message): """ @property - def uninterpreted_option(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___UninterpretedOption]: + def uninterpreted_option( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___UninterpretedOption]: """The parser stores options it doesn't recognize here. See above.""" def __init__( @@ -1666,8 +2152,25 @@ class MethodOptions(google.protobuf.message.Message): features: global___FeatureSet | None = ..., uninterpreted_option: collections.abc.Iterable[global___UninterpretedOption] | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["deprecated", b"deprecated", "features", b"features", "idempotency_level", b"idempotency_level"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["deprecated", b"deprecated", "features", b"features", "idempotency_level", b"idempotency_level", "uninterpreted_option", b"uninterpreted_option"]) -> None: ... + def HasField( + self, + field_name: typing.Literal[ + "deprecated", b"deprecated", "features", b"features", "idempotency_level", b"idempotency_level" + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "deprecated", + b"deprecated", + "features", + b"features", + "idempotency_level", + b"idempotency_level", + "uninterpreted_option", + b"uninterpreted_option", + ], + ) -> None: ... global___MethodOptions = MethodOptions @@ -1698,13 +2201,10 @@ class UninterpretedOption(google.protobuf.message.Message): IS_EXTENSION_FIELD_NUMBER: builtins.int name_part: builtins.str is_extension: builtins.bool - def __init__( - self, - *, - name_part: builtins.str | None = ..., - is_extension: builtins.bool | None = ..., - ) -> None: ... - def HasField(self, field_name: typing.Literal["is_extension", b"is_extension", "name_part", b"name_part"]) -> builtins.bool: ... + def __init__(self, *, name_part: builtins.str | None = ..., is_extension: builtins.bool | None = ...) -> None: ... + def HasField( + self, field_name: typing.Literal["is_extension", b"is_extension", "name_part", b"name_part"] + ) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["is_extension", b"is_extension", "name_part", b"name_part"]) -> None: ... NAME_FIELD_NUMBER: builtins.int @@ -1724,7 +2224,9 @@ class UninterpretedOption(google.protobuf.message.Message): string_value: builtins.bytes aggregate_value: builtins.str @property - def name(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___UninterpretedOption.NamePart]: ... + def name( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___UninterpretedOption.NamePart]: ... def __init__( self, *, @@ -1736,8 +2238,42 @@ class UninterpretedOption(google.protobuf.message.Message): string_value: builtins.bytes | None = ..., aggregate_value: builtins.str | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["aggregate_value", b"aggregate_value", "double_value", b"double_value", "identifier_value", b"identifier_value", "negative_int_value", b"negative_int_value", "positive_int_value", b"positive_int_value", "string_value", b"string_value"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["aggregate_value", b"aggregate_value", "double_value", b"double_value", "identifier_value", b"identifier_value", "name", b"name", "negative_int_value", b"negative_int_value", "positive_int_value", b"positive_int_value", "string_value", b"string_value"]) -> None: ... + def HasField( + self, + field_name: typing.Literal[ + "aggregate_value", + b"aggregate_value", + "double_value", + b"double_value", + "identifier_value", + b"identifier_value", + "negative_int_value", + b"negative_int_value", + "positive_int_value", + b"positive_int_value", + "string_value", + b"string_value", + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "aggregate_value", + b"aggregate_value", + "double_value", + b"double_value", + "identifier_value", + b"identifier_value", + "name", + b"name", + "negative_int_value", + b"negative_int_value", + "positive_int_value", + b"positive_int_value", + "string_value", + b"string_value", + ], + ) -> None: ... global___UninterpretedOption = UninterpretedOption @@ -1760,7 +2296,9 @@ class FeatureSet(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _FieldPresenceEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[FeatureSet._FieldPresence.ValueType], builtins.type): + class _FieldPresenceEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[FeatureSet._FieldPresence.ValueType], builtins.type + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor FIELD_PRESENCE_UNKNOWN: FeatureSet._FieldPresence.ValueType # 0 EXPLICIT: FeatureSet._FieldPresence.ValueType # 1 @@ -1777,7 +2315,9 @@ class FeatureSet(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _EnumTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[FeatureSet._EnumType.ValueType], builtins.type): + class _EnumTypeEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[FeatureSet._EnumType.ValueType], builtins.type + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor ENUM_TYPE_UNKNOWN: FeatureSet._EnumType.ValueType # 0 OPEN: FeatureSet._EnumType.ValueType # 1 @@ -1792,7 +2332,9 @@ class FeatureSet(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _RepeatedFieldEncodingEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[FeatureSet._RepeatedFieldEncoding.ValueType], builtins.type): + class _RepeatedFieldEncodingEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[FeatureSet._RepeatedFieldEncoding.ValueType], builtins.type + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor REPEATED_FIELD_ENCODING_UNKNOWN: FeatureSet._RepeatedFieldEncoding.ValueType # 0 PACKED: FeatureSet._RepeatedFieldEncoding.ValueType # 1 @@ -1807,7 +2349,9 @@ class FeatureSet(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _Utf8ValidationEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[FeatureSet._Utf8Validation.ValueType], builtins.type): + class _Utf8ValidationEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[FeatureSet._Utf8Validation.ValueType], builtins.type + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor UTF8_VALIDATION_UNKNOWN: FeatureSet._Utf8Validation.ValueType # 0 VERIFY: FeatureSet._Utf8Validation.ValueType # 2 @@ -1822,7 +2366,9 @@ class FeatureSet(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _MessageEncodingEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[FeatureSet._MessageEncoding.ValueType], builtins.type): + class _MessageEncodingEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[FeatureSet._MessageEncoding.ValueType], builtins.type + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor MESSAGE_ENCODING_UNKNOWN: FeatureSet._MessageEncoding.ValueType # 0 LENGTH_PREFIXED: FeatureSet._MessageEncoding.ValueType # 1 @@ -1837,7 +2383,9 @@ class FeatureSet(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _JsonFormatEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[FeatureSet._JsonFormat.ValueType], builtins.type): + class _JsonFormatEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[FeatureSet._JsonFormat.ValueType], builtins.type + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor JSON_FORMAT_UNKNOWN: FeatureSet._JsonFormat.ValueType # 0 ALLOW: FeatureSet._JsonFormat.ValueType # 1 @@ -1852,7 +2400,9 @@ class FeatureSet(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _EnforceNamingStyleEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[FeatureSet._EnforceNamingStyle.ValueType], builtins.type): + class _EnforceNamingStyleEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[FeatureSet._EnforceNamingStyle.ValueType], builtins.type + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor ENFORCE_NAMING_STYLE_UNKNOWN: FeatureSet._EnforceNamingStyle.ValueType # 0 STYLE2024: FeatureSet._EnforceNamingStyle.ValueType # 1 @@ -1888,8 +2438,44 @@ class FeatureSet(google.protobuf.message.Message): json_format: global___FeatureSet.JsonFormat.ValueType | None = ..., enforce_naming_style: global___FeatureSet.EnforceNamingStyle.ValueType | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["enforce_naming_style", b"enforce_naming_style", "enum_type", b"enum_type", "field_presence", b"field_presence", "json_format", b"json_format", "message_encoding", b"message_encoding", "repeated_field_encoding", b"repeated_field_encoding", "utf8_validation", b"utf8_validation"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["enforce_naming_style", b"enforce_naming_style", "enum_type", b"enum_type", "field_presence", b"field_presence", "json_format", b"json_format", "message_encoding", b"message_encoding", "repeated_field_encoding", b"repeated_field_encoding", "utf8_validation", b"utf8_validation"]) -> None: ... + def HasField( + self, + field_name: typing.Literal[ + "enforce_naming_style", + b"enforce_naming_style", + "enum_type", + b"enum_type", + "field_presence", + b"field_presence", + "json_format", + b"json_format", + "message_encoding", + b"message_encoding", + "repeated_field_encoding", + b"repeated_field_encoding", + "utf8_validation", + b"utf8_validation", + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "enforce_naming_style", + b"enforce_naming_style", + "enum_type", + b"enum_type", + "field_presence", + b"field_presence", + "json_format", + b"json_format", + "message_encoding", + b"message_encoding", + "repeated_field_encoding", + b"repeated_field_encoding", + "utf8_validation", + b"utf8_validation", + ], + ) -> None: ... global___FeatureSet = FeatureSet @@ -1932,8 +2518,18 @@ class FeatureSetDefaults(google.protobuf.message.Message): overridable_features: global___FeatureSet | None = ..., fixed_features: global___FeatureSet | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["edition", b"edition", "fixed_features", b"fixed_features", "overridable_features", b"overridable_features"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["edition", b"edition", "fixed_features", b"fixed_features", "overridable_features", b"overridable_features"]) -> None: ... + def HasField( + self, + field_name: typing.Literal[ + "edition", b"edition", "fixed_features", b"fixed_features", "overridable_features", b"overridable_features" + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "edition", b"edition", "fixed_features", b"fixed_features", "overridable_features", b"overridable_features" + ], + ) -> None: ... DEFAULTS_FIELD_NUMBER: builtins.int MINIMUM_EDITION_FIELD_NUMBER: builtins.int @@ -1947,7 +2543,11 @@ class FeatureSetDefaults(google.protobuf.message.Message): after this will not have reliable defaults. """ @property - def defaults(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___FeatureSetDefaults.FeatureSetEditionDefault]: ... + def defaults( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[ + global___FeatureSetDefaults.FeatureSetEditionDefault + ]: ... def __init__( self, *, @@ -1955,8 +2555,15 @@ class FeatureSetDefaults(google.protobuf.message.Message): minimum_edition: global___Edition.ValueType | None = ..., maximum_edition: global___Edition.ValueType | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["maximum_edition", b"maximum_edition", "minimum_edition", b"minimum_edition"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["defaults", b"defaults", "maximum_edition", b"maximum_edition", "minimum_edition", b"minimum_edition"]) -> None: ... + def HasField( + self, field_name: typing.Literal["maximum_edition", b"maximum_edition", "minimum_edition", b"minimum_edition"] + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "defaults", b"defaults", "maximum_edition", b"maximum_edition", "minimum_edition", b"minimum_edition" + ], + ) -> None: ... global___FeatureSetDefaults = FeatureSetDefaults @@ -2077,8 +2684,24 @@ class SourceCodeInfo(google.protobuf.message.Message): trailing_comments: builtins.str | None = ..., leading_detached_comments: collections.abc.Iterable[builtins.str] | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["leading_comments", b"leading_comments", "trailing_comments", b"trailing_comments"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["leading_comments", b"leading_comments", "leading_detached_comments", b"leading_detached_comments", "path", b"path", "span", b"span", "trailing_comments", b"trailing_comments"]) -> None: ... + def HasField( + self, field_name: typing.Literal["leading_comments", b"leading_comments", "trailing_comments", b"trailing_comments"] + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "leading_comments", + b"leading_comments", + "leading_detached_comments", + b"leading_detached_comments", + "path", + b"path", + "span", + b"span", + "trailing_comments", + b"trailing_comments", + ], + ) -> None: ... LOCATION_FIELD_NUMBER: builtins.int @property @@ -2128,11 +2751,7 @@ class SourceCodeInfo(google.protobuf.message.Message): be recorded in the future. """ - def __init__( - self, - *, - location: collections.abc.Iterable[global___SourceCodeInfo.Location] | None = ..., - ) -> None: ... + def __init__(self, *, location: collections.abc.Iterable[global___SourceCodeInfo.Location] | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["location", b"location"]) -> None: ... global___SourceCodeInfo = SourceCodeInfo @@ -2154,7 +2773,10 @@ class GeneratedCodeInfo(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _SemanticEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[GeneratedCodeInfo.Annotation._Semantic.ValueType], builtins.type): + class _SemanticEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[GeneratedCodeInfo.Annotation._Semantic.ValueType], + builtins.type, + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor NONE: GeneratedCodeInfo.Annotation._Semantic.ValueType # 0 """There is no effect or the effect is indescribable.""" @@ -2207,21 +2829,27 @@ class GeneratedCodeInfo(google.protobuf.message.Message): end: builtins.int | None = ..., semantic: global___GeneratedCodeInfo.Annotation.Semantic.ValueType | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["begin", b"begin", "end", b"end", "semantic", b"semantic", "source_file", b"source_file"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["begin", b"begin", "end", b"end", "path", b"path", "semantic", b"semantic", "source_file", b"source_file"]) -> None: ... + def HasField( + self, + field_name: typing.Literal["begin", b"begin", "end", b"end", "semantic", b"semantic", "source_file", b"source_file"], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "begin", b"begin", "end", b"end", "path", b"path", "semantic", b"semantic", "source_file", b"source_file" + ], + ) -> None: ... ANNOTATION_FIELD_NUMBER: builtins.int @property - def annotation(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___GeneratedCodeInfo.Annotation]: + def annotation( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___GeneratedCodeInfo.Annotation]: """An Annotation connects some span of text in generated code to an element of its generating .proto file. """ - def __init__( - self, - *, - annotation: collections.abc.Iterable[global___GeneratedCodeInfo.Annotation] | None = ..., - ) -> None: ... + def __init__(self, *, annotation: collections.abc.Iterable[global___GeneratedCodeInfo.Annotation] | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["annotation", b"annotation"]) -> None: ... global___GeneratedCodeInfo = GeneratedCodeInfo diff --git a/stubs/protobuf/google/protobuf/duration_pb2.pyi b/stubs/protobuf/google/protobuf/duration_pb2.pyi index 4beb927266cb..2cb846889f4d 100644 --- a/stubs/protobuf/google/protobuf/duration_pb2.pyi +++ b/stubs/protobuf/google/protobuf/duration_pb2.pyi @@ -120,12 +120,7 @@ class Duration(google.protobuf.message.Message, google.protobuf.internal.well_kn of the same sign as the `seconds` field. Must be from -999,999,999 to +999,999,999 inclusive. """ - def __init__( - self, - *, - seconds: builtins.int | None = ..., - nanos: builtins.int | None = ..., - ) -> None: ... + def __init__(self, *, seconds: builtins.int | None = ..., nanos: builtins.int | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["nanos", b"nanos", "seconds", b"seconds"]) -> None: ... global___Duration = Duration diff --git a/stubs/protobuf/google/protobuf/empty_pb2.pyi b/stubs/protobuf/google/protobuf/empty_pb2.pyi index 593917f63474..317979279540 100644 --- a/stubs/protobuf/google/protobuf/empty_pb2.pyi +++ b/stubs/protobuf/google/protobuf/empty_pb2.pyi @@ -52,8 +52,6 @@ class Empty(google.protobuf.message.Message): DESCRIPTOR: google.protobuf.descriptor.Descriptor - def __init__( - self, - ) -> None: ... + def __init__(self) -> None: ... global___Empty = Empty diff --git a/stubs/protobuf/google/protobuf/field_mask_pb2.pyi b/stubs/protobuf/google/protobuf/field_mask_pb2.pyi index 1cc4929f00b5..82dea7a11b56 100644 --- a/stubs/protobuf/google/protobuf/field_mask_pb2.pyi +++ b/stubs/protobuf/google/protobuf/field_mask_pb2.pyi @@ -253,11 +253,7 @@ class FieldMask(google.protobuf.message.Message, google.protobuf.internal.well_k def paths(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]: """The set of field mask paths.""" - def __init__( - self, - *, - paths: collections.abc.Iterable[builtins.str] | None = ..., - ) -> None: ... + def __init__(self, *, paths: collections.abc.Iterable[builtins.str] | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["paths", b"paths"]) -> None: ... global___FieldMask = FieldMask diff --git a/stubs/protobuf/google/protobuf/source_context_pb2.pyi b/stubs/protobuf/google/protobuf/source_context_pb2.pyi index a2d35937211d..9d74fb3d5926 100644 --- a/stubs/protobuf/google/protobuf/source_context_pb2.pyi +++ b/stubs/protobuf/google/protobuf/source_context_pb2.pyi @@ -53,11 +53,7 @@ class SourceContext(google.protobuf.message.Message): """The path-qualified name of the .proto file that contained the associated protobuf element. For example: `"google/protobuf/source_context.proto"`. """ - def __init__( - self, - *, - file_name: builtins.str | None = ..., - ) -> None: ... + def __init__(self, *, file_name: builtins.str | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["file_name", b"file_name"]) -> None: ... global___SourceContext = SourceContext diff --git a/stubs/protobuf/google/protobuf/struct_pb2.pyi b/stubs/protobuf/google/protobuf/struct_pb2.pyi index 684f3c9d4e08..8e63dabfc2f1 100644 --- a/stubs/protobuf/google/protobuf/struct_pb2.pyi +++ b/stubs/protobuf/google/protobuf/struct_pb2.pyi @@ -93,12 +93,7 @@ class Struct(google.protobuf.message.Message, google.protobuf.internal.well_know key: builtins.str @property def value(self) -> global___Value: ... - def __init__( - self, - *, - key: builtins.str | None = ..., - value: global___Value | None = ..., - ) -> None: ... + def __init__(self, *, key: builtins.str | None = ..., value: global___Value | None = ...) -> None: ... def HasField(self, field_name: typing.Literal["value", b"value"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["key", b"key", "value", b"value"]) -> None: ... @@ -107,11 +102,7 @@ class Struct(google.protobuf.message.Message, google.protobuf.internal.well_know def fields(self) -> google.protobuf.internal.containers.MessageMap[builtins.str, global___Value]: """Unordered map of dynamically typed values.""" - def __init__( - self, - *, - fields: collections.abc.Mapping[builtins.str, global___Value] | None = ..., - ) -> None: ... + def __init__(self, *, fields: collections.abc.Mapping[builtins.str, global___Value] | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["fields", b"fields"]) -> None: ... global___Struct = Struct @@ -160,9 +151,47 @@ class Value(google.protobuf.message.Message): struct_value: global___Struct | None = ..., list_value: global___ListValue | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["bool_value", b"bool_value", "kind", b"kind", "list_value", b"list_value", "null_value", b"null_value", "number_value", b"number_value", "string_value", b"string_value", "struct_value", b"struct_value"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["bool_value", b"bool_value", "kind", b"kind", "list_value", b"list_value", "null_value", b"null_value", "number_value", b"number_value", "string_value", b"string_value", "struct_value", b"struct_value"]) -> None: ... - def WhichOneof(self, oneof_group: typing.Literal["kind", b"kind"]) -> typing.Literal["null_value", "number_value", "string_value", "bool_value", "struct_value", "list_value"] | None: ... + def HasField( + self, + field_name: typing.Literal[ + "bool_value", + b"bool_value", + "kind", + b"kind", + "list_value", + b"list_value", + "null_value", + b"null_value", + "number_value", + b"number_value", + "string_value", + b"string_value", + "struct_value", + b"struct_value", + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "bool_value", + b"bool_value", + "kind", + b"kind", + "list_value", + b"list_value", + "null_value", + b"null_value", + "number_value", + b"number_value", + "string_value", + b"string_value", + "struct_value", + b"struct_value", + ], + ) -> None: ... + def WhichOneof( + self, oneof_group: typing.Literal["kind", b"kind"] + ) -> typing.Literal["null_value", "number_value", "string_value", "bool_value", "struct_value", "list_value"] | None: ... global___Value = Value @@ -180,11 +209,7 @@ class ListValue(google.protobuf.message.Message, google.protobuf.internal.well_k def values(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___Value]: """Repeated field of dynamically typed values.""" - def __init__( - self, - *, - values: collections.abc.Iterable[global___Value] | None = ..., - ) -> None: ... + def __init__(self, *, values: collections.abc.Iterable[global___Value] | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["values", b"values"]) -> None: ... global___ListValue = ListValue diff --git a/stubs/protobuf/google/protobuf/timestamp_pb2.pyi b/stubs/protobuf/google/protobuf/timestamp_pb2.pyi index 017a881de8b9..52daff363ddf 100644 --- a/stubs/protobuf/google/protobuf/timestamp_pb2.pyi +++ b/stubs/protobuf/google/protobuf/timestamp_pb2.pyi @@ -149,12 +149,7 @@ class Timestamp(google.protobuf.message.Message, google.protobuf.internal.well_k that count forward in time. Must be from 0 to 999,999,999 inclusive. """ - def __init__( - self, - *, - seconds: builtins.int | None = ..., - nanos: builtins.int | None = ..., - ) -> None: ... + def __init__(self, *, seconds: builtins.int | None = ..., nanos: builtins.int | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["nanos", b"nanos", "seconds", b"seconds"]) -> None: ... global___Timestamp = Timestamp diff --git a/stubs/protobuf/google/protobuf/type_pb2.pyi b/stubs/protobuf/google/protobuf/type_pb2.pyi index 78e6b5b355a2..4d3fc1cb6577 100644 --- a/stubs/protobuf/google/protobuf/type_pb2.pyi +++ b/stubs/protobuf/google/protobuf/type_pb2.pyi @@ -122,7 +122,25 @@ class Type(google.protobuf.message.Message): edition: builtins.str | None = ..., ) -> None: ... def HasField(self, field_name: typing.Literal["source_context", b"source_context"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["edition", b"edition", "fields", b"fields", "name", b"name", "oneofs", b"oneofs", "options", b"options", "source_context", b"source_context", "syntax", b"syntax"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "edition", + b"edition", + "fields", + b"fields", + "name", + b"name", + "oneofs", + b"oneofs", + "options", + b"options", + "source_context", + b"source_context", + "syntax", + b"syntax", + ], + ) -> None: ... global___Type = Type @@ -223,7 +241,9 @@ class Field(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _CardinalityEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[Field._Cardinality.ValueType], builtins.type): + class _CardinalityEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[Field._Cardinality.ValueType], builtins.type + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor CARDINALITY_UNKNOWN: Field._Cardinality.ValueType # 0 """For fields with unknown cardinality.""" @@ -296,7 +316,31 @@ class Field(google.protobuf.message.Message): json_name: builtins.str | None = ..., default_value: builtins.str | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["cardinality", b"cardinality", "default_value", b"default_value", "json_name", b"json_name", "kind", b"kind", "name", b"name", "number", b"number", "oneof_index", b"oneof_index", "options", b"options", "packed", b"packed", "type_url", b"type_url"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "cardinality", + b"cardinality", + "default_value", + b"default_value", + "json_name", + b"json_name", + "kind", + b"kind", + "name", + b"name", + "number", + b"number", + "oneof_index", + b"oneof_index", + "options", + b"options", + "packed", + b"packed", + "type_url", + b"type_url", + ], + ) -> None: ... global___Field = Field @@ -341,7 +385,23 @@ class Enum(google.protobuf.message.Message): edition: builtins.str | None = ..., ) -> None: ... def HasField(self, field_name: typing.Literal["source_context", b"source_context"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["edition", b"edition", "enumvalue", b"enumvalue", "name", b"name", "options", b"options", "source_context", b"source_context", "syntax", b"syntax"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "edition", + b"edition", + "enumvalue", + b"enumvalue", + "name", + b"name", + "options", + b"options", + "source_context", + b"source_context", + "syntax", + b"syntax", + ], + ) -> None: ... global___Enum = Enum @@ -397,12 +457,7 @@ class Option(google.protobuf.message.Message): value using the google.protobuf.Int32Value type. """ - def __init__( - self, - *, - name: builtins.str | None = ..., - value: google.protobuf.any_pb2.Any | None = ..., - ) -> None: ... + def __init__(self, *, name: builtins.str | None = ..., value: google.protobuf.any_pb2.Any | None = ...) -> None: ... def HasField(self, field_name: typing.Literal["value", b"value"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["name", b"name", "value", b"value"]) -> None: ... diff --git a/stubs/protobuf/google/protobuf/wrappers_pb2.pyi b/stubs/protobuf/google/protobuf/wrappers_pb2.pyi index 24aae899790d..b3d875bccf67 100644 --- a/stubs/protobuf/google/protobuf/wrappers_pb2.pyi +++ b/stubs/protobuf/google/protobuf/wrappers_pb2.pyi @@ -72,11 +72,7 @@ class DoubleValue(google.protobuf.message.Message): VALUE_FIELD_NUMBER: builtins.int value: builtins.float """The double value.""" - def __init__( - self, - *, - value: builtins.float | None = ..., - ) -> None: ... + def __init__(self, *, value: builtins.float | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["value", b"value"]) -> None: ... global___DoubleValue = DoubleValue @@ -96,11 +92,7 @@ class FloatValue(google.protobuf.message.Message): VALUE_FIELD_NUMBER: builtins.int value: builtins.float """The float value.""" - def __init__( - self, - *, - value: builtins.float | None = ..., - ) -> None: ... + def __init__(self, *, value: builtins.float | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["value", b"value"]) -> None: ... global___FloatValue = FloatValue @@ -120,11 +112,7 @@ class Int64Value(google.protobuf.message.Message): VALUE_FIELD_NUMBER: builtins.int value: builtins.int """The int64 value.""" - def __init__( - self, - *, - value: builtins.int | None = ..., - ) -> None: ... + def __init__(self, *, value: builtins.int | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["value", b"value"]) -> None: ... global___Int64Value = Int64Value @@ -144,11 +132,7 @@ class UInt64Value(google.protobuf.message.Message): VALUE_FIELD_NUMBER: builtins.int value: builtins.int """The uint64 value.""" - def __init__( - self, - *, - value: builtins.int | None = ..., - ) -> None: ... + def __init__(self, *, value: builtins.int | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["value", b"value"]) -> None: ... global___UInt64Value = UInt64Value @@ -168,11 +152,7 @@ class Int32Value(google.protobuf.message.Message): VALUE_FIELD_NUMBER: builtins.int value: builtins.int """The int32 value.""" - def __init__( - self, - *, - value: builtins.int | None = ..., - ) -> None: ... + def __init__(self, *, value: builtins.int | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["value", b"value"]) -> None: ... global___Int32Value = Int32Value @@ -192,11 +172,7 @@ class UInt32Value(google.protobuf.message.Message): VALUE_FIELD_NUMBER: builtins.int value: builtins.int """The uint32 value.""" - def __init__( - self, - *, - value: builtins.int | None = ..., - ) -> None: ... + def __init__(self, *, value: builtins.int | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["value", b"value"]) -> None: ... global___UInt32Value = UInt32Value @@ -216,11 +192,7 @@ class BoolValue(google.protobuf.message.Message): VALUE_FIELD_NUMBER: builtins.int value: builtins.bool """The bool value.""" - def __init__( - self, - *, - value: builtins.bool | None = ..., - ) -> None: ... + def __init__(self, *, value: builtins.bool | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["value", b"value"]) -> None: ... global___BoolValue = BoolValue @@ -240,11 +212,7 @@ class StringValue(google.protobuf.message.Message): VALUE_FIELD_NUMBER: builtins.int value: builtins.str """The string value.""" - def __init__( - self, - *, - value: builtins.str | None = ..., - ) -> None: ... + def __init__(self, *, value: builtins.str | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["value", b"value"]) -> None: ... global___StringValue = StringValue @@ -264,11 +232,7 @@ class BytesValue(google.protobuf.message.Message): VALUE_FIELD_NUMBER: builtins.int value: builtins.bytes """The bytes value.""" - def __init__( - self, - *, - value: builtins.bytes | None = ..., - ) -> None: ... + def __init__(self, *, value: builtins.bytes | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["value", b"value"]) -> None: ... global___BytesValue = BytesValue diff --git a/stubs/s2clientprotocol/s2clientprotocol/common_pb2.pyi b/stubs/s2clientprotocol/s2clientprotocol/common_pb2.pyi index 298ef9bf1c76..cfa0bec86506 100644 --- a/stubs/s2clientprotocol/s2clientprotocol/common_pb2.pyi +++ b/stubs/s2clientprotocol/s2clientprotocol/common_pb2.pyi @@ -47,14 +47,13 @@ class AvailableAbility(google.protobuf.message.Message): REQUIRES_POINT_FIELD_NUMBER: builtins.int ability_id: builtins.int requires_point: builtins.bool - def __init__( - self, - *, - ability_id: builtins.int | None = ..., - requires_point: builtins.bool | None = ..., + def __init__(self, *, ability_id: builtins.int | None = ..., requires_point: builtins.bool | None = ...) -> None: ... + def HasField( + self, field_name: typing.Literal["ability_id", b"ability_id", "requires_point", b"requires_point"] + ) -> builtins.bool: ... + def ClearField( + self, field_name: typing.Literal["ability_id", b"ability_id", "requires_point", b"requires_point"] ) -> None: ... - def HasField(self, field_name: typing.Literal["ability_id", b"ability_id", "requires_point", b"requires_point"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["ability_id", b"ability_id", "requires_point", b"requires_point"]) -> None: ... global___AvailableAbility = AvailableAbility @@ -74,14 +73,14 @@ class ImageData(google.protobuf.message.Message): """Dimension in pixels.""" def __init__( - self, - *, - bits_per_pixel: builtins.int | None = ..., - size: global___Size2DI | None = ..., - data: builtins.bytes | None = ..., + self, *, bits_per_pixel: builtins.int | None = ..., size: global___Size2DI | None = ..., data: builtins.bytes | None = ... + ) -> None: ... + def HasField( + self, field_name: typing.Literal["bits_per_pixel", b"bits_per_pixel", "data", b"data", "size", b"size"] + ) -> builtins.bool: ... + def ClearField( + self, field_name: typing.Literal["bits_per_pixel", b"bits_per_pixel", "data", b"data", "size", b"size"] ) -> None: ... - def HasField(self, field_name: typing.Literal["bits_per_pixel", b"bits_per_pixel", "data", b"data", "size", b"size"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["bits_per_pixel", b"bits_per_pixel", "data", b"data", "size", b"size"]) -> None: ... global___ImageData = ImageData @@ -97,12 +96,7 @@ class PointI(google.protobuf.message.Message): Y_FIELD_NUMBER: builtins.int x: builtins.int y: builtins.int - def __init__( - self, - *, - x: builtins.int | None = ..., - y: builtins.int | None = ..., - ) -> None: ... + def __init__(self, *, x: builtins.int | None = ..., y: builtins.int | None = ...) -> None: ... def HasField(self, field_name: typing.Literal["x", b"x", "y", b"y"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["x", b"x", "y", b"y"]) -> None: ... @@ -120,12 +114,7 @@ class RectangleI(google.protobuf.message.Message): def p0(self) -> global___PointI: ... @property def p1(self) -> global___PointI: ... - def __init__( - self, - *, - p0: global___PointI | None = ..., - p1: global___PointI | None = ..., - ) -> None: ... + def __init__(self, *, p0: global___PointI | None = ..., p1: global___PointI | None = ...) -> None: ... def HasField(self, field_name: typing.Literal["p0", b"p0", "p1", b"p1"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["p0", b"p0", "p1", b"p1"]) -> None: ... @@ -143,12 +132,7 @@ class Point2D(google.protobuf.message.Message): Y_FIELD_NUMBER: builtins.int x: builtins.float y: builtins.float - def __init__( - self, - *, - x: builtins.float | None = ..., - y: builtins.float | None = ..., - ) -> None: ... + def __init__(self, *, x: builtins.float | None = ..., y: builtins.float | None = ...) -> None: ... def HasField(self, field_name: typing.Literal["x", b"x", "y", b"y"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["x", b"x", "y", b"y"]) -> None: ... @@ -169,11 +153,7 @@ class Point(google.protobuf.message.Message): y: builtins.float z: builtins.float def __init__( - self, - *, - x: builtins.float | None = ..., - y: builtins.float | None = ..., - z: builtins.float | None = ..., + self, *, x: builtins.float | None = ..., y: builtins.float | None = ..., z: builtins.float | None = ... ) -> None: ... def HasField(self, field_name: typing.Literal["x", b"x", "y", b"y", "z", b"z"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["x", b"x", "y", b"y", "z", b"z"]) -> None: ... @@ -190,12 +170,7 @@ class Size2DI(google.protobuf.message.Message): Y_FIELD_NUMBER: builtins.int x: builtins.int y: builtins.int - def __init__( - self, - *, - x: builtins.int | None = ..., - y: builtins.int | None = ..., - ) -> None: ... + def __init__(self, *, x: builtins.int | None = ..., y: builtins.int | None = ...) -> None: ... def HasField(self, field_name: typing.Literal["x", b"x", "y", b"y"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["x", b"x", "y", b"y"]) -> None: ... diff --git a/stubs/s2clientprotocol/s2clientprotocol/data_pb2.pyi b/stubs/s2clientprotocol/s2clientprotocol/data_pb2.pyi index 6c33927e724e..cd68bb65df02 100644 --- a/stubs/s2clientprotocol/s2clientprotocol/data_pb2.pyi +++ b/stubs/s2clientprotocol/s2clientprotocol/data_pb2.pyi @@ -68,7 +68,9 @@ class AbilityData(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _TargetEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[AbilityData._Target.ValueType], builtins.type): + class _TargetEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[AbilityData._Target.ValueType], builtins.type + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor Point: AbilityData._Target.ValueType # 2 """Requires a target position.""" @@ -153,8 +155,76 @@ class AbilityData(google.protobuf.message.Message): is_instant_placement: builtins.bool | None = ..., cast_range: builtins.float | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["ability_id", b"ability_id", "allow_autocast", b"allow_autocast", "allow_minimap", b"allow_minimap", "available", b"available", "button_name", b"button_name", "cast_range", b"cast_range", "footprint_radius", b"footprint_radius", "friendly_name", b"friendly_name", "hotkey", b"hotkey", "is_building", b"is_building", "is_instant_placement", b"is_instant_placement", "link_index", b"link_index", "link_name", b"link_name", "remaps_to_ability_id", b"remaps_to_ability_id", "target", b"target"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["ability_id", b"ability_id", "allow_autocast", b"allow_autocast", "allow_minimap", b"allow_minimap", "available", b"available", "button_name", b"button_name", "cast_range", b"cast_range", "footprint_radius", b"footprint_radius", "friendly_name", b"friendly_name", "hotkey", b"hotkey", "is_building", b"is_building", "is_instant_placement", b"is_instant_placement", "link_index", b"link_index", "link_name", b"link_name", "remaps_to_ability_id", b"remaps_to_ability_id", "target", b"target"]) -> None: ... + def HasField( + self, + field_name: typing.Literal[ + "ability_id", + b"ability_id", + "allow_autocast", + b"allow_autocast", + "allow_minimap", + b"allow_minimap", + "available", + b"available", + "button_name", + b"button_name", + "cast_range", + b"cast_range", + "footprint_radius", + b"footprint_radius", + "friendly_name", + b"friendly_name", + "hotkey", + b"hotkey", + "is_building", + b"is_building", + "is_instant_placement", + b"is_instant_placement", + "link_index", + b"link_index", + "link_name", + b"link_name", + "remaps_to_ability_id", + b"remaps_to_ability_id", + "target", + b"target", + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "ability_id", + b"ability_id", + "allow_autocast", + b"allow_autocast", + "allow_minimap", + b"allow_minimap", + "available", + b"available", + "button_name", + b"button_name", + "cast_range", + b"cast_range", + "footprint_radius", + b"footprint_radius", + "friendly_name", + b"friendly_name", + "hotkey", + b"hotkey", + "is_building", + b"is_building", + "is_instant_placement", + b"is_instant_placement", + "link_index", + b"link_index", + "link_name", + b"link_name", + "remaps_to_ability_id", + b"remaps_to_ability_id", + "target", + b"target", + ], + ) -> None: ... global___AbilityData = AbilityData @@ -166,12 +236,7 @@ class DamageBonus(google.protobuf.message.Message): BONUS_FIELD_NUMBER: builtins.int attribute: global___Attribute.ValueType bonus: builtins.float - def __init__( - self, - *, - attribute: global___Attribute.ValueType | None = ..., - bonus: builtins.float | None = ..., - ) -> None: ... + def __init__(self, *, attribute: global___Attribute.ValueType | None = ..., bonus: builtins.float | None = ...) -> None: ... def HasField(self, field_name: typing.Literal["attribute", b"attribute", "bonus", b"bonus"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["attribute", b"attribute", "bonus", b"bonus"]) -> None: ... @@ -185,7 +250,9 @@ class Weapon(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _TargetTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[Weapon._TargetType.ValueType], builtins.type): + class _TargetTypeEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[Weapon._TargetType.ValueType], builtins.type + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor Ground: Weapon._TargetType.ValueType # 1 Air: Weapon._TargetType.ValueType # 2 @@ -221,8 +288,29 @@ class Weapon(google.protobuf.message.Message): range: builtins.float | None = ..., speed: builtins.float | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["attacks", b"attacks", "damage", b"damage", "range", b"range", "speed", b"speed", "type", b"type"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["attacks", b"attacks", "damage", b"damage", "damage_bonus", b"damage_bonus", "range", b"range", "speed", b"speed", "type", b"type"]) -> None: ... + def HasField( + self, + field_name: typing.Literal[ + "attacks", b"attacks", "damage", b"damage", "range", b"range", "speed", b"speed", "type", b"type" + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "attacks", + b"attacks", + "damage", + b"damage", + "damage_bonus", + b"damage_bonus", + "range", + b"range", + "speed", + b"speed", + "type", + b"type", + ], + ) -> None: ... global___Weapon = Weapon @@ -316,8 +404,98 @@ class UnitTypeData(google.protobuf.message.Message): armor: builtins.float | None = ..., weapons: collections.abc.Iterable[global___Weapon] | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["ability_id", b"ability_id", "armor", b"armor", "available", b"available", "build_time", b"build_time", "cargo_size", b"cargo_size", "food_provided", b"food_provided", "food_required", b"food_required", "has_minerals", b"has_minerals", "has_vespene", b"has_vespene", "mineral_cost", b"mineral_cost", "movement_speed", b"movement_speed", "name", b"name", "race", b"race", "require_attached", b"require_attached", "sight_range", b"sight_range", "tech_requirement", b"tech_requirement", "unit_alias", b"unit_alias", "unit_id", b"unit_id", "vespene_cost", b"vespene_cost"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["ability_id", b"ability_id", "armor", b"armor", "attributes", b"attributes", "available", b"available", "build_time", b"build_time", "cargo_size", b"cargo_size", "food_provided", b"food_provided", "food_required", b"food_required", "has_minerals", b"has_minerals", "has_vespene", b"has_vespene", "mineral_cost", b"mineral_cost", "movement_speed", b"movement_speed", "name", b"name", "race", b"race", "require_attached", b"require_attached", "sight_range", b"sight_range", "tech_alias", b"tech_alias", "tech_requirement", b"tech_requirement", "unit_alias", b"unit_alias", "unit_id", b"unit_id", "vespene_cost", b"vespene_cost", "weapons", b"weapons"]) -> None: ... + def HasField( + self, + field_name: typing.Literal[ + "ability_id", + b"ability_id", + "armor", + b"armor", + "available", + b"available", + "build_time", + b"build_time", + "cargo_size", + b"cargo_size", + "food_provided", + b"food_provided", + "food_required", + b"food_required", + "has_minerals", + b"has_minerals", + "has_vespene", + b"has_vespene", + "mineral_cost", + b"mineral_cost", + "movement_speed", + b"movement_speed", + "name", + b"name", + "race", + b"race", + "require_attached", + b"require_attached", + "sight_range", + b"sight_range", + "tech_requirement", + b"tech_requirement", + "unit_alias", + b"unit_alias", + "unit_id", + b"unit_id", + "vespene_cost", + b"vespene_cost", + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "ability_id", + b"ability_id", + "armor", + b"armor", + "attributes", + b"attributes", + "available", + b"available", + "build_time", + b"build_time", + "cargo_size", + b"cargo_size", + "food_provided", + b"food_provided", + "food_required", + b"food_required", + "has_minerals", + b"has_minerals", + "has_vespene", + b"has_vespene", + "mineral_cost", + b"mineral_cost", + "movement_speed", + b"movement_speed", + "name", + b"name", + "race", + b"race", + "require_attached", + b"require_attached", + "sight_range", + b"sight_range", + "tech_alias", + b"tech_alias", + "tech_requirement", + b"tech_requirement", + "unit_alias", + b"unit_alias", + "unit_id", + b"unit_id", + "vespene_cost", + b"vespene_cost", + "weapons", + b"weapons", + ], + ) -> None: ... global___UnitTypeData = UnitTypeData @@ -348,8 +526,40 @@ class UpgradeData(google.protobuf.message.Message): research_time: builtins.float | None = ..., ability_id: builtins.int | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["ability_id", b"ability_id", "mineral_cost", b"mineral_cost", "name", b"name", "research_time", b"research_time", "upgrade_id", b"upgrade_id", "vespene_cost", b"vespene_cost"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["ability_id", b"ability_id", "mineral_cost", b"mineral_cost", "name", b"name", "research_time", b"research_time", "upgrade_id", b"upgrade_id", "vespene_cost", b"vespene_cost"]) -> None: ... + def HasField( + self, + field_name: typing.Literal[ + "ability_id", + b"ability_id", + "mineral_cost", + b"mineral_cost", + "name", + b"name", + "research_time", + b"research_time", + "upgrade_id", + b"upgrade_id", + "vespene_cost", + b"vespene_cost", + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "ability_id", + b"ability_id", + "mineral_cost", + b"mineral_cost", + "name", + b"name", + "research_time", + b"research_time", + "upgrade_id", + b"upgrade_id", + "vespene_cost", + b"vespene_cost", + ], + ) -> None: ... global___UpgradeData = UpgradeData @@ -362,12 +572,7 @@ class BuffData(google.protobuf.message.Message): buff_id: builtins.int """Stable ID.""" name: builtins.str - def __init__( - self, - *, - buff_id: builtins.int | None = ..., - name: builtins.str | None = ..., - ) -> None: ... + def __init__(self, *, buff_id: builtins.int | None = ..., name: builtins.str | None = ...) -> None: ... def HasField(self, field_name: typing.Literal["buff_id", b"buff_id", "name", b"name"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["buff_id", b"buff_id", "name", b"name"]) -> None: ... @@ -394,7 +599,17 @@ class EffectData(google.protobuf.message.Message): friendly_name: builtins.str | None = ..., radius: builtins.float | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["effect_id", b"effect_id", "friendly_name", b"friendly_name", "name", b"name", "radius", b"radius"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["effect_id", b"effect_id", "friendly_name", b"friendly_name", "name", b"name", "radius", b"radius"]) -> None: ... + def HasField( + self, + field_name: typing.Literal[ + "effect_id", b"effect_id", "friendly_name", b"friendly_name", "name", b"name", "radius", b"radius" + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "effect_id", b"effect_id", "friendly_name", b"friendly_name", "name", b"name", "radius", b"radius" + ], + ) -> None: ... global___EffectData = EffectData diff --git a/stubs/s2clientprotocol/s2clientprotocol/debug_pb2.pyi b/stubs/s2clientprotocol/s2clientprotocol/debug_pb2.pyi index 658d5bd20590..eba2ae972064 100644 --- a/stubs/s2clientprotocol/s2clientprotocol/debug_pb2.pyi +++ b/stubs/s2clientprotocol/s2clientprotocol/debug_pb2.pyi @@ -25,7 +25,9 @@ class _DebugGameState: ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType -class _DebugGameStateEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_DebugGameState.ValueType], builtins.type): +class _DebugGameStateEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_DebugGameState.ValueType], builtins.type +): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor show_map: _DebugGameState.ValueType # 1 control_enemy: _DebugGameState.ValueType # 2 @@ -99,9 +101,57 @@ class DebugCommand(google.protobuf.message.Message): end_game: global___DebugEndGame | None = ..., unit_value: global___DebugSetUnitValue | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["command", b"command", "create_unit", b"create_unit", "draw", b"draw", "end_game", b"end_game", "game_state", b"game_state", "kill_unit", b"kill_unit", "score", b"score", "test_process", b"test_process", "unit_value", b"unit_value"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["command", b"command", "create_unit", b"create_unit", "draw", b"draw", "end_game", b"end_game", "game_state", b"game_state", "kill_unit", b"kill_unit", "score", b"score", "test_process", b"test_process", "unit_value", b"unit_value"]) -> None: ... - def WhichOneof(self, oneof_group: typing.Literal["command", b"command"]) -> typing.Literal["draw", "game_state", "create_unit", "kill_unit", "test_process", "score", "end_game", "unit_value"] | None: ... + def HasField( + self, + field_name: typing.Literal[ + "command", + b"command", + "create_unit", + b"create_unit", + "draw", + b"draw", + "end_game", + b"end_game", + "game_state", + b"game_state", + "kill_unit", + b"kill_unit", + "score", + b"score", + "test_process", + b"test_process", + "unit_value", + b"unit_value", + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "command", + b"command", + "create_unit", + b"create_unit", + "draw", + b"draw", + "end_game", + b"end_game", + "game_state", + b"game_state", + "kill_unit", + b"kill_unit", + "score", + b"score", + "test_process", + b"test_process", + "unit_value", + b"unit_value", + ], + ) -> None: ... + def WhichOneof( + self, oneof_group: typing.Literal["command", b"command"] + ) -> ( + typing.Literal["draw", "game_state", "create_unit", "kill_unit", "test_process", "score", "end_game", "unit_value"] | None + ): ... global___DebugCommand = DebugCommand @@ -129,7 +179,9 @@ class DebugDraw(google.protobuf.message.Message): boxes: collections.abc.Iterable[global___DebugBox] | None = ..., spheres: collections.abc.Iterable[global___DebugSphere] | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["boxes", b"boxes", "lines", b"lines", "spheres", b"spheres", "text", b"text"]) -> None: ... + def ClearField( + self, field_name: typing.Literal["boxes", b"boxes", "lines", b"lines", "spheres", b"spheres", "text", b"text"] + ) -> None: ... global___DebugDraw = DebugDraw @@ -144,10 +196,7 @@ class Line(google.protobuf.message.Message): @property def p1(self) -> s2clientprotocol.common_pb2.Point: ... def __init__( - self, - *, - p0: s2clientprotocol.common_pb2.Point | None = ..., - p1: s2clientprotocol.common_pb2.Point | None = ..., + self, *, p0: s2clientprotocol.common_pb2.Point | None = ..., p1: s2clientprotocol.common_pb2.Point | None = ... ) -> None: ... def HasField(self, field_name: typing.Literal["p0", b"p0", "p1", b"p1"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["p0", b"p0", "p1", b"p1"]) -> None: ... @@ -164,13 +213,7 @@ class Color(google.protobuf.message.Message): r: builtins.int g: builtins.int b: builtins.int - def __init__( - self, - *, - r: builtins.int | None = ..., - g: builtins.int | None = ..., - b: builtins.int | None = ..., - ) -> None: ... + def __init__(self, *, r: builtins.int | None = ..., g: builtins.int | None = ..., b: builtins.int | None = ...) -> None: ... def HasField(self, field_name: typing.Literal["b", b"b", "g", b"g", "r", b"r"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["b", b"b", "g", b"g", "r", b"r"]) -> None: ... @@ -210,8 +253,18 @@ class DebugText(google.protobuf.message.Message): world_pos: s2clientprotocol.common_pb2.Point | None = ..., size: builtins.int | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["color", b"color", "size", b"size", "text", b"text", "virtual_pos", b"virtual_pos", "world_pos", b"world_pos"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["color", b"color", "size", b"size", "text", b"text", "virtual_pos", b"virtual_pos", "world_pos", b"world_pos"]) -> None: ... + def HasField( + self, + field_name: typing.Literal[ + "color", b"color", "size", b"size", "text", b"text", "virtual_pos", b"virtual_pos", "world_pos", b"world_pos" + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "color", b"color", "size", b"size", "text", b"text", "virtual_pos", b"virtual_pos", "world_pos", b"world_pos" + ], + ) -> None: ... global___DebugText = DebugText @@ -229,12 +282,7 @@ class DebugLine(google.protobuf.message.Message): def line(self) -> global___Line: """World space line.""" - def __init__( - self, - *, - color: global___Color | None = ..., - line: global___Line | None = ..., - ) -> None: ... + def __init__(self, *, color: global___Color | None = ..., line: global___Line | None = ...) -> None: ... def HasField(self, field_name: typing.Literal["color", b"color", "line", b"line"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["color", b"color", "line", b"line"]) -> None: ... @@ -314,8 +362,12 @@ class DebugCreateUnit(google.protobuf.message.Message): pos: s2clientprotocol.common_pb2.Point2D | None = ..., quantity: builtins.int | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["owner", b"owner", "pos", b"pos", "quantity", b"quantity", "unit_type", b"unit_type"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["owner", b"owner", "pos", b"pos", "quantity", b"quantity", "unit_type", b"unit_type"]) -> None: ... + def HasField( + self, field_name: typing.Literal["owner", b"owner", "pos", b"pos", "quantity", b"quantity", "unit_type", b"unit_type"] + ) -> builtins.bool: ... + def ClearField( + self, field_name: typing.Literal["owner", b"owner", "pos", b"pos", "quantity", b"quantity", "unit_type", b"unit_type"] + ) -> None: ... global___DebugCreateUnit = DebugCreateUnit @@ -326,11 +378,7 @@ class DebugKillUnit(google.protobuf.message.Message): TAG_FIELD_NUMBER: builtins.int @property def tag(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.int]: ... - def __init__( - self, - *, - tag: collections.abc.Iterable[builtins.int] | None = ..., - ) -> None: ... + def __init__(self, *, tag: collections.abc.Iterable[builtins.int] | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["tag", b"tag"]) -> None: ... global___DebugKillUnit = DebugKillUnit @@ -343,7 +391,9 @@ class DebugTestProcess(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _TestEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[DebugTestProcess._Test.ValueType], builtins.type): + class _TestEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[DebugTestProcess._Test.ValueType], builtins.type + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor hang: DebugTestProcess._Test.ValueType # 1 crash: DebugTestProcess._Test.ValueType # 2 @@ -359,10 +409,7 @@ class DebugTestProcess(google.protobuf.message.Message): test: global___DebugTestProcess.Test.ValueType delay_ms: builtins.int def __init__( - self, - *, - test: global___DebugTestProcess.Test.ValueType | None = ..., - delay_ms: builtins.int | None = ..., + self, *, test: global___DebugTestProcess.Test.ValueType | None = ..., delay_ms: builtins.int | None = ... ) -> None: ... def HasField(self, field_name: typing.Literal["delay_ms", b"delay_ms", "test", b"test"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["delay_ms", b"delay_ms", "test", b"test"]) -> None: ... @@ -375,11 +422,7 @@ class DebugSetScore(google.protobuf.message.Message): SCORE_FIELD_NUMBER: builtins.int score: builtins.float - def __init__( - self, - *, - score: builtins.float | None = ..., - ) -> None: ... + def __init__(self, *, score: builtins.float | None = ...) -> None: ... def HasField(self, field_name: typing.Literal["score", b"score"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["score", b"score"]) -> None: ... @@ -393,7 +436,9 @@ class DebugEndGame(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _EndResultEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[DebugEndGame._EndResult.ValueType], builtins.type): + class _EndResultEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[DebugEndGame._EndResult.ValueType], builtins.type + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor Surrender: DebugEndGame._EndResult.ValueType # 1 """Default if nothing is set. The current player admits defeat.""" @@ -406,11 +451,7 @@ class DebugEndGame(google.protobuf.message.Message): END_RESULT_FIELD_NUMBER: builtins.int end_result: global___DebugEndGame.EndResult.ValueType - def __init__( - self, - *, - end_result: global___DebugEndGame.EndResult.ValueType | None = ..., - ) -> None: ... + def __init__(self, *, end_result: global___DebugEndGame.EndResult.ValueType | None = ...) -> None: ... def HasField(self, field_name: typing.Literal["end_result", b"end_result"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["end_result", b"end_result"]) -> None: ... @@ -424,7 +465,9 @@ class DebugSetUnitValue(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _UnitValueEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[DebugSetUnitValue._UnitValue.ValueType], builtins.type): + class _UnitValueEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[DebugSetUnitValue._UnitValue.ValueType], builtins.type + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor Energy: DebugSetUnitValue._UnitValue.ValueType # 1 Life: DebugSetUnitValue._UnitValue.ValueType # 2 @@ -448,7 +491,11 @@ class DebugSetUnitValue(google.protobuf.message.Message): value: builtins.float | None = ..., unit_tag: builtins.int | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["unit_tag", b"unit_tag", "unit_value", b"unit_value", "value", b"value"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["unit_tag", b"unit_tag", "unit_value", b"unit_value", "value", b"value"]) -> None: ... + def HasField( + self, field_name: typing.Literal["unit_tag", b"unit_tag", "unit_value", b"unit_value", "value", b"value"] + ) -> builtins.bool: ... + def ClearField( + self, field_name: typing.Literal["unit_tag", b"unit_tag", "unit_value", b"unit_value", "value", b"value"] + ) -> None: ... global___DebugSetUnitValue = DebugSetUnitValue diff --git a/stubs/s2clientprotocol/s2clientprotocol/error_pb2.pyi b/stubs/s2clientprotocol/s2clientprotocol/error_pb2.pyi index c21b03cbc1b8..5c14aaba748c 100644 --- a/stubs/s2clientprotocol/s2clientprotocol/error_pb2.pyi +++ b/stubs/s2clientprotocol/s2clientprotocol/error_pb2.pyi @@ -21,7 +21,9 @@ class _ActionResult: ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType -class _ActionResultEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_ActionResult.ValueType], builtins.type): +class _ActionResultEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_ActionResult.ValueType], builtins.type +): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor Success: _ActionResult.ValueType # 1 NotSupported: _ActionResult.ValueType # 2 diff --git a/stubs/s2clientprotocol/s2clientprotocol/query_pb2.pyi b/stubs/s2clientprotocol/s2clientprotocol/query_pb2.pyi index ea772e2f6242..2df098c7eb4a 100644 --- a/stubs/s2clientprotocol/s2clientprotocol/query_pb2.pyi +++ b/stubs/s2clientprotocol/s2clientprotocol/query_pb2.pyi @@ -28,9 +28,13 @@ class RequestQuery(google.protobuf.message.Message): @property def pathing(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___RequestQueryPathing]: ... @property - def abilities(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___RequestQueryAvailableAbilities]: ... + def abilities( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___RequestQueryAvailableAbilities]: ... @property - def placements(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___RequestQueryBuildingPlacement]: ... + def placements( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___RequestQueryBuildingPlacement]: ... def __init__( self, *, @@ -39,8 +43,22 @@ class RequestQuery(google.protobuf.message.Message): placements: collections.abc.Iterable[global___RequestQueryBuildingPlacement] | None = ..., ignore_resource_requirements: builtins.bool | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["ignore_resource_requirements", b"ignore_resource_requirements"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["abilities", b"abilities", "ignore_resource_requirements", b"ignore_resource_requirements", "pathing", b"pathing", "placements", b"placements"]) -> None: ... + def HasField( + self, field_name: typing.Literal["ignore_resource_requirements", b"ignore_resource_requirements"] + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "abilities", + b"abilities", + "ignore_resource_requirements", + b"ignore_resource_requirements", + "pathing", + b"pathing", + "placements", + b"placements", + ], + ) -> None: ... global___RequestQuery = RequestQuery @@ -54,9 +72,13 @@ class ResponseQuery(google.protobuf.message.Message): @property def pathing(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___ResponseQueryPathing]: ... @property - def abilities(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___ResponseQueryAvailableAbilities]: ... + def abilities( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___ResponseQueryAvailableAbilities]: ... @property - def placements(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___ResponseQueryBuildingPlacement]: ... + def placements( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___ResponseQueryBuildingPlacement]: ... def __init__( self, *, @@ -64,7 +86,9 @@ class ResponseQuery(google.protobuf.message.Message): abilities: collections.abc.Iterable[global___ResponseQueryAvailableAbilities] | None = ..., placements: collections.abc.Iterable[global___ResponseQueryBuildingPlacement] | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["abilities", b"abilities", "pathing", b"pathing", "placements", b"placements"]) -> None: ... + def ClearField( + self, field_name: typing.Literal["abilities", b"abilities", "pathing", b"pathing", "placements", b"placements"] + ) -> None: ... global___ResponseQuery = ResponseQuery @@ -89,8 +113,14 @@ class RequestQueryPathing(google.protobuf.message.Message): unit_tag: builtins.int | None = ..., end_pos: s2clientprotocol.common_pb2.Point2D | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["end_pos", b"end_pos", "start", b"start", "start_pos", b"start_pos", "unit_tag", b"unit_tag"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["end_pos", b"end_pos", "start", b"start", "start_pos", b"start_pos", "unit_tag", b"unit_tag"]) -> None: ... + def HasField( + self, + field_name: typing.Literal["end_pos", b"end_pos", "start", b"start", "start_pos", b"start_pos", "unit_tag", b"unit_tag"], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal["end_pos", b"end_pos", "start", b"start", "start_pos", b"start_pos", "unit_tag", b"unit_tag"], + ) -> None: ... def WhichOneof(self, oneof_group: typing.Literal["start", b"start"]) -> typing.Literal["start_pos", "unit_tag"] | None: ... global___RequestQueryPathing = RequestQueryPathing @@ -102,11 +132,7 @@ class ResponseQueryPathing(google.protobuf.message.Message): DISTANCE_FIELD_NUMBER: builtins.int distance: builtins.float """0 if no path exists""" - def __init__( - self, - *, - distance: builtins.float | None = ..., - ) -> None: ... + def __init__(self, *, distance: builtins.float | None = ...) -> None: ... def HasField(self, field_name: typing.Literal["distance", b"distance"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["distance", b"distance"]) -> None: ... @@ -120,11 +146,7 @@ class RequestQueryAvailableAbilities(google.protobuf.message.Message): UNIT_TAG_FIELD_NUMBER: builtins.int unit_tag: builtins.int - def __init__( - self, - *, - unit_tag: builtins.int | None = ..., - ) -> None: ... + def __init__(self, *, unit_tag: builtins.int | None = ...) -> None: ... def HasField(self, field_name: typing.Literal["unit_tag", b"unit_tag"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["unit_tag", b"unit_tag"]) -> None: ... @@ -140,7 +162,9 @@ class ResponseQueryAvailableAbilities(google.protobuf.message.Message): unit_tag: builtins.int unit_type_id: builtins.int @property - def abilities(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[s2clientprotocol.common_pb2.AvailableAbility]: ... + def abilities( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[s2clientprotocol.common_pb2.AvailableAbility]: ... def __init__( self, *, @@ -149,7 +173,9 @@ class ResponseQueryAvailableAbilities(google.protobuf.message.Message): unit_type_id: builtins.int | None = ..., ) -> None: ... def HasField(self, field_name: typing.Literal["unit_tag", b"unit_tag", "unit_type_id", b"unit_type_id"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["abilities", b"abilities", "unit_tag", b"unit_tag", "unit_type_id", b"unit_type_id"]) -> None: ... + def ClearField( + self, field_name: typing.Literal["abilities", b"abilities", "unit_tag", b"unit_tag", "unit_type_id", b"unit_type_id"] + ) -> None: ... global___ResponseQueryAvailableAbilities = ResponseQueryAvailableAbilities @@ -174,8 +200,18 @@ class RequestQueryBuildingPlacement(google.protobuf.message.Message): target_pos: s2clientprotocol.common_pb2.Point2D | None = ..., placing_unit_tag: builtins.int | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["ability_id", b"ability_id", "placing_unit_tag", b"placing_unit_tag", "target_pos", b"target_pos"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["ability_id", b"ability_id", "placing_unit_tag", b"placing_unit_tag", "target_pos", b"target_pos"]) -> None: ... + def HasField( + self, + field_name: typing.Literal[ + "ability_id", b"ability_id", "placing_unit_tag", b"placing_unit_tag", "target_pos", b"target_pos" + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "ability_id", b"ability_id", "placing_unit_tag", b"placing_unit_tag", "target_pos", b"target_pos" + ], + ) -> None: ... global___RequestQueryBuildingPlacement = RequestQueryBuildingPlacement @@ -185,11 +221,7 @@ class ResponseQueryBuildingPlacement(google.protobuf.message.Message): RESULT_FIELD_NUMBER: builtins.int result: s2clientprotocol.error_pb2.ActionResult.ValueType - def __init__( - self, - *, - result: s2clientprotocol.error_pb2.ActionResult.ValueType | None = ..., - ) -> None: ... + def __init__(self, *, result: s2clientprotocol.error_pb2.ActionResult.ValueType | None = ...) -> None: ... def HasField(self, field_name: typing.Literal["result", b"result"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["result", b"result"]) -> None: ... diff --git a/stubs/s2clientprotocol/s2clientprotocol/raw_pb2.pyi b/stubs/s2clientprotocol/s2clientprotocol/raw_pb2.pyi index 35a08c2337ea..a50167f1e087 100644 --- a/stubs/s2clientprotocol/s2clientprotocol/raw_pb2.pyi +++ b/stubs/s2clientprotocol/s2clientprotocol/raw_pb2.pyi @@ -25,7 +25,9 @@ class _DisplayType: ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType -class _DisplayTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_DisplayType.ValueType], builtins.type): +class _DisplayTypeEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_DisplayType.ValueType], builtins.type +): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor Visible: _DisplayType.ValueType # 1 """Fully visible""" @@ -71,7 +73,9 @@ class _CloakState: ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType -class _CloakStateEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_CloakState.ValueType], builtins.type): +class _CloakStateEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_CloakState.ValueType], builtins.type +): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor CloakedUnknown: _CloakState.ValueType # 0 """Under the fog, so unknown whether it's cloaked or not.""" @@ -125,7 +129,9 @@ class StartRaw(google.protobuf.message.Message): """The playable cells.""" @property - def start_locations(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[s2clientprotocol.common_pb2.Point2D]: + def start_locations( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[s2clientprotocol.common_pb2.Point2D]: """Possible start locations for players.""" def __init__( @@ -138,8 +144,38 @@ class StartRaw(google.protobuf.message.Message): playable_area: s2clientprotocol.common_pb2.RectangleI | None = ..., start_locations: collections.abc.Iterable[s2clientprotocol.common_pb2.Point2D] | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["map_size", b"map_size", "pathing_grid", b"pathing_grid", "placement_grid", b"placement_grid", "playable_area", b"playable_area", "terrain_height", b"terrain_height"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["map_size", b"map_size", "pathing_grid", b"pathing_grid", "placement_grid", b"placement_grid", "playable_area", b"playable_area", "start_locations", b"start_locations", "terrain_height", b"terrain_height"]) -> None: ... + def HasField( + self, + field_name: typing.Literal[ + "map_size", + b"map_size", + "pathing_grid", + b"pathing_grid", + "placement_grid", + b"placement_grid", + "playable_area", + b"playable_area", + "terrain_height", + b"terrain_height", + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "map_size", + b"map_size", + "pathing_grid", + b"pathing_grid", + "placement_grid", + b"placement_grid", + "playable_area", + b"playable_area", + "start_locations", + b"start_locations", + "terrain_height", + b"terrain_height", + ], + ) -> None: ... global___StartRaw = StartRaw @@ -181,8 +217,26 @@ class ObservationRaw(google.protobuf.message.Message): effects: collections.abc.Iterable[global___Effect] | None = ..., radar: collections.abc.Iterable[global___RadarRing] | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["event", b"event", "map_state", b"map_state", "player", b"player"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["effects", b"effects", "event", b"event", "map_state", b"map_state", "player", b"player", "radar", b"radar", "units", b"units"]) -> None: ... + def HasField( + self, field_name: typing.Literal["event", b"event", "map_state", b"map_state", "player", b"player"] + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "effects", + b"effects", + "event", + b"event", + "map_state", + b"map_state", + "player", + b"player", + "radar", + b"radar", + "units", + b"units", + ], + ) -> None: ... global___ObservationRaw = ObservationRaw @@ -195,12 +249,7 @@ class RadarRing(google.protobuf.message.Message): radius: builtins.float @property def pos(self) -> s2clientprotocol.common_pb2.Point: ... - def __init__( - self, - *, - pos: s2clientprotocol.common_pb2.Point | None = ..., - radius: builtins.float | None = ..., - ) -> None: ... + def __init__(self, *, pos: s2clientprotocol.common_pb2.Point | None = ..., radius: builtins.float | None = ...) -> None: ... def HasField(self, field_name: typing.Literal["pos", b"pos", "radius", b"radius"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["pos", b"pos", "radius", b"radius"]) -> None: ... @@ -252,7 +301,9 @@ class PlayerRaw(google.protobuf.message.Message): upgrade_ids: collections.abc.Iterable[builtins.int] | None = ..., ) -> None: ... def HasField(self, field_name: typing.Literal["camera", b"camera"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["camera", b"camera", "power_sources", b"power_sources", "upgrade_ids", b"upgrade_ids"]) -> None: ... + def ClearField( + self, field_name: typing.Literal["camera", b"camera", "power_sources", b"power_sources", "upgrade_ids", b"upgrade_ids"] + ) -> None: ... global___PlayerRaw = PlayerRaw @@ -278,9 +329,39 @@ class UnitOrder(google.protobuf.message.Message): target_unit_tag: builtins.int | None = ..., progress: builtins.float | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["ability_id", b"ability_id", "progress", b"progress", "target", b"target", "target_unit_tag", b"target_unit_tag", "target_world_space_pos", b"target_world_space_pos"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["ability_id", b"ability_id", "progress", b"progress", "target", b"target", "target_unit_tag", b"target_unit_tag", "target_world_space_pos", b"target_world_space_pos"]) -> None: ... - def WhichOneof(self, oneof_group: typing.Literal["target", b"target"]) -> typing.Literal["target_world_space_pos", "target_unit_tag"] | None: ... + def HasField( + self, + field_name: typing.Literal[ + "ability_id", + b"ability_id", + "progress", + b"progress", + "target", + b"target", + "target_unit_tag", + b"target_unit_tag", + "target_world_space_pos", + b"target_world_space_pos", + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "ability_id", + b"ability_id", + "progress", + b"progress", + "target", + b"target", + "target_unit_tag", + b"target_unit_tag", + "target_world_space_pos", + b"target_world_space_pos", + ], + ) -> None: ... + def WhichOneof( + self, oneof_group: typing.Literal["target", b"target"] + ) -> typing.Literal["target_world_space_pos", "target_unit_tag"] | None: ... global___UnitOrder = UnitOrder @@ -316,8 +397,48 @@ class PassengerUnit(google.protobuf.message.Message): energy_max: builtins.float | None = ..., unit_type: builtins.int | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["energy", b"energy", "energy_max", b"energy_max", "health", b"health", "health_max", b"health_max", "shield", b"shield", "shield_max", b"shield_max", "tag", b"tag", "unit_type", b"unit_type"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["energy", b"energy", "energy_max", b"energy_max", "health", b"health", "health_max", b"health_max", "shield", b"shield", "shield_max", b"shield_max", "tag", b"tag", "unit_type", b"unit_type"]) -> None: ... + def HasField( + self, + field_name: typing.Literal[ + "energy", + b"energy", + "energy_max", + b"energy_max", + "health", + b"health", + "health_max", + b"health_max", + "shield", + b"shield", + "shield_max", + b"shield_max", + "tag", + b"tag", + "unit_type", + b"unit_type", + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "energy", + b"energy", + "energy_max", + b"energy_max", + "health", + b"health", + "health_max", + b"health_max", + "shield", + b"shield", + "shield_max", + b"shield_max", + "tag", + b"tag", + "unit_type", + b"unit_type", + ], + ) -> None: ... global___PassengerUnit = PassengerUnit @@ -333,12 +454,7 @@ class RallyTarget(google.protobuf.message.Message): def point(self) -> s2clientprotocol.common_pb2.Point: """Will always be filled.""" - def __init__( - self, - *, - point: s2clientprotocol.common_pb2.Point | None = ..., - tag: builtins.int | None = ..., - ) -> None: ... + def __init__(self, *, point: s2clientprotocol.common_pb2.Point | None = ..., tag: builtins.int | None = ...) -> None: ... def HasField(self, field_name: typing.Literal["point", b"point", "tag", b"tag"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["point", b"point", "tag", b"tag"]) -> None: ... @@ -501,8 +617,184 @@ class Unit(google.protobuf.message.Message): buff_duration_max: builtins.int | None = ..., rally_targets: collections.abc.Iterable[global___RallyTarget] | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["add_on_tag", b"add_on_tag", "alliance", b"alliance", "armor_upgrade_level", b"armor_upgrade_level", "assigned_harvesters", b"assigned_harvesters", "attack_upgrade_level", b"attack_upgrade_level", "buff_duration_max", b"buff_duration_max", "buff_duration_remain", b"buff_duration_remain", "build_progress", b"build_progress", "cargo_space_max", b"cargo_space_max", "cargo_space_taken", b"cargo_space_taken", "cloak", b"cloak", "detect_range", b"detect_range", "display_type", b"display_type", "energy", b"energy", "energy_max", b"energy_max", "engaged_target_tag", b"engaged_target_tag", "facing", b"facing", "health", b"health", "health_max", b"health_max", "ideal_harvesters", b"ideal_harvesters", "is_active", b"is_active", "is_blip", b"is_blip", "is_burrowed", b"is_burrowed", "is_flying", b"is_flying", "is_hallucination", b"is_hallucination", "is_on_screen", b"is_on_screen", "is_powered", b"is_powered", "is_selected", b"is_selected", "mineral_contents", b"mineral_contents", "owner", b"owner", "pos", b"pos", "radar_range", b"radar_range", "radius", b"radius", "shield", b"shield", "shield_max", b"shield_max", "shield_upgrade_level", b"shield_upgrade_level", "tag", b"tag", "unit_type", b"unit_type", "vespene_contents", b"vespene_contents", "weapon_cooldown", b"weapon_cooldown"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["add_on_tag", b"add_on_tag", "alliance", b"alliance", "armor_upgrade_level", b"armor_upgrade_level", "assigned_harvesters", b"assigned_harvesters", "attack_upgrade_level", b"attack_upgrade_level", "buff_duration_max", b"buff_duration_max", "buff_duration_remain", b"buff_duration_remain", "buff_ids", b"buff_ids", "build_progress", b"build_progress", "cargo_space_max", b"cargo_space_max", "cargo_space_taken", b"cargo_space_taken", "cloak", b"cloak", "detect_range", b"detect_range", "display_type", b"display_type", "energy", b"energy", "energy_max", b"energy_max", "engaged_target_tag", b"engaged_target_tag", "facing", b"facing", "health", b"health", "health_max", b"health_max", "ideal_harvesters", b"ideal_harvesters", "is_active", b"is_active", "is_blip", b"is_blip", "is_burrowed", b"is_burrowed", "is_flying", b"is_flying", "is_hallucination", b"is_hallucination", "is_on_screen", b"is_on_screen", "is_powered", b"is_powered", "is_selected", b"is_selected", "mineral_contents", b"mineral_contents", "orders", b"orders", "owner", b"owner", "passengers", b"passengers", "pos", b"pos", "radar_range", b"radar_range", "radius", b"radius", "rally_targets", b"rally_targets", "shield", b"shield", "shield_max", b"shield_max", "shield_upgrade_level", b"shield_upgrade_level", "tag", b"tag", "unit_type", b"unit_type", "vespene_contents", b"vespene_contents", "weapon_cooldown", b"weapon_cooldown"]) -> None: ... + def HasField( + self, + field_name: typing.Literal[ + "add_on_tag", + b"add_on_tag", + "alliance", + b"alliance", + "armor_upgrade_level", + b"armor_upgrade_level", + "assigned_harvesters", + b"assigned_harvesters", + "attack_upgrade_level", + b"attack_upgrade_level", + "buff_duration_max", + b"buff_duration_max", + "buff_duration_remain", + b"buff_duration_remain", + "build_progress", + b"build_progress", + "cargo_space_max", + b"cargo_space_max", + "cargo_space_taken", + b"cargo_space_taken", + "cloak", + b"cloak", + "detect_range", + b"detect_range", + "display_type", + b"display_type", + "energy", + b"energy", + "energy_max", + b"energy_max", + "engaged_target_tag", + b"engaged_target_tag", + "facing", + b"facing", + "health", + b"health", + "health_max", + b"health_max", + "ideal_harvesters", + b"ideal_harvesters", + "is_active", + b"is_active", + "is_blip", + b"is_blip", + "is_burrowed", + b"is_burrowed", + "is_flying", + b"is_flying", + "is_hallucination", + b"is_hallucination", + "is_on_screen", + b"is_on_screen", + "is_powered", + b"is_powered", + "is_selected", + b"is_selected", + "mineral_contents", + b"mineral_contents", + "owner", + b"owner", + "pos", + b"pos", + "radar_range", + b"radar_range", + "radius", + b"radius", + "shield", + b"shield", + "shield_max", + b"shield_max", + "shield_upgrade_level", + b"shield_upgrade_level", + "tag", + b"tag", + "unit_type", + b"unit_type", + "vespene_contents", + b"vespene_contents", + "weapon_cooldown", + b"weapon_cooldown", + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "add_on_tag", + b"add_on_tag", + "alliance", + b"alliance", + "armor_upgrade_level", + b"armor_upgrade_level", + "assigned_harvesters", + b"assigned_harvesters", + "attack_upgrade_level", + b"attack_upgrade_level", + "buff_duration_max", + b"buff_duration_max", + "buff_duration_remain", + b"buff_duration_remain", + "buff_ids", + b"buff_ids", + "build_progress", + b"build_progress", + "cargo_space_max", + b"cargo_space_max", + "cargo_space_taken", + b"cargo_space_taken", + "cloak", + b"cloak", + "detect_range", + b"detect_range", + "display_type", + b"display_type", + "energy", + b"energy", + "energy_max", + b"energy_max", + "engaged_target_tag", + b"engaged_target_tag", + "facing", + b"facing", + "health", + b"health", + "health_max", + b"health_max", + "ideal_harvesters", + b"ideal_harvesters", + "is_active", + b"is_active", + "is_blip", + b"is_blip", + "is_burrowed", + b"is_burrowed", + "is_flying", + b"is_flying", + "is_hallucination", + b"is_hallucination", + "is_on_screen", + b"is_on_screen", + "is_powered", + b"is_powered", + "is_selected", + b"is_selected", + "mineral_contents", + b"mineral_contents", + "orders", + b"orders", + "owner", + b"owner", + "passengers", + b"passengers", + "pos", + b"pos", + "radar_range", + b"radar_range", + "radius", + b"radius", + "rally_targets", + b"rally_targets", + "shield", + b"shield", + "shield_max", + b"shield_max", + "shield_upgrade_level", + b"shield_upgrade_level", + "tag", + b"tag", + "unit_type", + b"unit_type", + "vespene_contents", + b"vespene_contents", + "weapon_cooldown", + b"weapon_cooldown", + ], + ) -> None: ... global___Unit = Unit @@ -538,11 +830,7 @@ class Event(google.protobuf.message.Message): DEAD_UNITS_FIELD_NUMBER: builtins.int @property def dead_units(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.int]: ... - def __init__( - self, - *, - dead_units: collections.abc.Iterable[builtins.int] | None = ..., - ) -> None: ... + def __init__(self, *, dead_units: collections.abc.Iterable[builtins.int] | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["dead_units", b"dead_units"]) -> None: ... global___Event = Event @@ -573,8 +861,16 @@ class Effect(google.protobuf.message.Message): owner: builtins.int | None = ..., radius: builtins.float | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["alliance", b"alliance", "effect_id", b"effect_id", "owner", b"owner", "radius", b"radius"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["alliance", b"alliance", "effect_id", b"effect_id", "owner", b"owner", "pos", b"pos", "radius", b"radius"]) -> None: ... + def HasField( + self, + field_name: typing.Literal["alliance", b"alliance", "effect_id", b"effect_id", "owner", b"owner", "radius", b"radius"], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "alliance", b"alliance", "effect_id", b"effect_id", "owner", b"owner", "pos", b"pos", "radius", b"radius" + ], + ) -> None: ... global___Effect = Effect @@ -602,9 +898,35 @@ class ActionRaw(google.protobuf.message.Message): camera_move: global___ActionRawCameraMove | None = ..., toggle_autocast: global___ActionRawToggleAutocast | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["action", b"action", "camera_move", b"camera_move", "toggle_autocast", b"toggle_autocast", "unit_command", b"unit_command"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["action", b"action", "camera_move", b"camera_move", "toggle_autocast", b"toggle_autocast", "unit_command", b"unit_command"]) -> None: ... - def WhichOneof(self, oneof_group: typing.Literal["action", b"action"]) -> typing.Literal["unit_command", "camera_move", "toggle_autocast"] | None: ... + def HasField( + self, + field_name: typing.Literal[ + "action", + b"action", + "camera_move", + b"camera_move", + "toggle_autocast", + b"toggle_autocast", + "unit_command", + b"unit_command", + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "action", + b"action", + "camera_move", + b"camera_move", + "toggle_autocast", + b"toggle_autocast", + "unit_command", + b"unit_command", + ], + ) -> None: ... + def WhichOneof( + self, oneof_group: typing.Literal["action", b"action"] + ) -> typing.Literal["unit_command", "camera_move", "toggle_autocast"] | None: ... global___ActionRaw = ActionRaw @@ -633,9 +955,41 @@ class ActionRawUnitCommand(google.protobuf.message.Message): unit_tags: collections.abc.Iterable[builtins.int] | None = ..., queue_command: builtins.bool | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["ability_id", b"ability_id", "queue_command", b"queue_command", "target", b"target", "target_unit_tag", b"target_unit_tag", "target_world_space_pos", b"target_world_space_pos"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["ability_id", b"ability_id", "queue_command", b"queue_command", "target", b"target", "target_unit_tag", b"target_unit_tag", "target_world_space_pos", b"target_world_space_pos", "unit_tags", b"unit_tags"]) -> None: ... - def WhichOneof(self, oneof_group: typing.Literal["target", b"target"]) -> typing.Literal["target_world_space_pos", "target_unit_tag"] | None: ... + def HasField( + self, + field_name: typing.Literal[ + "ability_id", + b"ability_id", + "queue_command", + b"queue_command", + "target", + b"target", + "target_unit_tag", + b"target_unit_tag", + "target_world_space_pos", + b"target_world_space_pos", + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "ability_id", + b"ability_id", + "queue_command", + b"queue_command", + "target", + b"target", + "target_unit_tag", + b"target_unit_tag", + "target_world_space_pos", + b"target_world_space_pos", + "unit_tags", + b"unit_tags", + ], + ) -> None: ... + def WhichOneof( + self, oneof_group: typing.Literal["target", b"target"] + ) -> typing.Literal["target_world_space_pos", "target_unit_tag"] | None: ... global___ActionRawUnitCommand = ActionRawUnitCommand @@ -646,11 +1000,7 @@ class ActionRawCameraMove(google.protobuf.message.Message): CENTER_WORLD_SPACE_FIELD_NUMBER: builtins.int @property def center_world_space(self) -> s2clientprotocol.common_pb2.Point: ... - def __init__( - self, - *, - center_world_space: s2clientprotocol.common_pb2.Point | None = ..., - ) -> None: ... + def __init__(self, *, center_world_space: s2clientprotocol.common_pb2.Point | None = ...) -> None: ... def HasField(self, field_name: typing.Literal["center_world_space", b"center_world_space"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["center_world_space", b"center_world_space"]) -> None: ... @@ -666,10 +1016,7 @@ class ActionRawToggleAutocast(google.protobuf.message.Message): @property def unit_tags(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.int]: ... def __init__( - self, - *, - ability_id: builtins.int | None = ..., - unit_tags: collections.abc.Iterable[builtins.int] | None = ..., + self, *, ability_id: builtins.int | None = ..., unit_tags: collections.abc.Iterable[builtins.int] | None = ... ) -> None: ... def HasField(self, field_name: typing.Literal["ability_id", b"ability_id"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["ability_id", b"ability_id", "unit_tags", b"unit_tags"]) -> None: ... diff --git a/stubs/s2clientprotocol/s2clientprotocol/sc2api_pb2.pyi b/stubs/s2clientprotocol/s2clientprotocol/sc2api_pb2.pyi index b979697237f5..fbf53b67a613 100644 --- a/stubs/s2clientprotocol/s2clientprotocol/sc2api_pb2.pyi +++ b/stubs/s2clientprotocol/s2clientprotocol/sc2api_pb2.pyi @@ -72,7 +72,9 @@ class _Difficulty: ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType -class _DifficultyEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_Difficulty.ValueType], builtins.type): +class _DifficultyEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_Difficulty.ValueType], builtins.type +): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor VeryEasy: _Difficulty.ValueType # 1 Easy: _Difficulty.ValueType # 2 @@ -106,7 +108,9 @@ class _PlayerType: ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType -class _PlayerTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_PlayerType.ValueType], builtins.type): +class _PlayerTypeEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_PlayerType.ValueType], builtins.type +): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor Participant: _PlayerType.ValueType # 1 Computer: _PlayerType.ValueType # 2 @@ -371,9 +375,141 @@ class Request(google.protobuf.message.Message): debug: global___RequestDebug | None = ..., id: builtins.int | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["action", b"action", "available_maps", b"available_maps", "create_game", b"create_game", "data", b"data", "debug", b"debug", "game_info", b"game_info", "id", b"id", "join_game", b"join_game", "leave_game", b"leave_game", "map_command", b"map_command", "obs_action", b"obs_action", "observation", b"observation", "ping", b"ping", "query", b"query", "quick_load", b"quick_load", "quick_save", b"quick_save", "quit", b"quit", "replay_info", b"replay_info", "request", b"request", "restart_game", b"restart_game", "save_map", b"save_map", "save_replay", b"save_replay", "start_replay", b"start_replay", "step", b"step"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["action", b"action", "available_maps", b"available_maps", "create_game", b"create_game", "data", b"data", "debug", b"debug", "game_info", b"game_info", "id", b"id", "join_game", b"join_game", "leave_game", b"leave_game", "map_command", b"map_command", "obs_action", b"obs_action", "observation", b"observation", "ping", b"ping", "query", b"query", "quick_load", b"quick_load", "quick_save", b"quick_save", "quit", b"quit", "replay_info", b"replay_info", "request", b"request", "restart_game", b"restart_game", "save_map", b"save_map", "save_replay", b"save_replay", "start_replay", b"start_replay", "step", b"step"]) -> None: ... - def WhichOneof(self, oneof_group: typing.Literal["request", b"request"]) -> typing.Literal["create_game", "join_game", "restart_game", "start_replay", "leave_game", "quick_save", "quick_load", "quit", "game_info", "observation", "action", "obs_action", "step", "data", "query", "save_replay", "map_command", "replay_info", "available_maps", "save_map", "ping", "debug"] | None: ... + def HasField( + self, + field_name: typing.Literal[ + "action", + b"action", + "available_maps", + b"available_maps", + "create_game", + b"create_game", + "data", + b"data", + "debug", + b"debug", + "game_info", + b"game_info", + "id", + b"id", + "join_game", + b"join_game", + "leave_game", + b"leave_game", + "map_command", + b"map_command", + "obs_action", + b"obs_action", + "observation", + b"observation", + "ping", + b"ping", + "query", + b"query", + "quick_load", + b"quick_load", + "quick_save", + b"quick_save", + "quit", + b"quit", + "replay_info", + b"replay_info", + "request", + b"request", + "restart_game", + b"restart_game", + "save_map", + b"save_map", + "save_replay", + b"save_replay", + "start_replay", + b"start_replay", + "step", + b"step", + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "action", + b"action", + "available_maps", + b"available_maps", + "create_game", + b"create_game", + "data", + b"data", + "debug", + b"debug", + "game_info", + b"game_info", + "id", + b"id", + "join_game", + b"join_game", + "leave_game", + b"leave_game", + "map_command", + b"map_command", + "obs_action", + b"obs_action", + "observation", + b"observation", + "ping", + b"ping", + "query", + b"query", + "quick_load", + b"quick_load", + "quick_save", + b"quick_save", + "quit", + b"quit", + "replay_info", + b"replay_info", + "request", + b"request", + "restart_game", + b"restart_game", + "save_map", + b"save_map", + "save_replay", + b"save_replay", + "start_replay", + b"start_replay", + "step", + b"step", + ], + ) -> None: ... + def WhichOneof( + self, oneof_group: typing.Literal["request", b"request"] + ) -> ( + typing.Literal[ + "create_game", + "join_game", + "restart_game", + "start_replay", + "leave_game", + "quick_save", + "quick_load", + "quit", + "game_info", + "observation", + "action", + "obs_action", + "step", + "data", + "query", + "save_replay", + "map_command", + "replay_info", + "available_maps", + "save_map", + "ping", + "debug", + ] + | None + ): ... global___Request = Request @@ -488,9 +624,147 @@ class Response(google.protobuf.message.Message): error: collections.abc.Iterable[builtins.str] | None = ..., status: global___Status.ValueType | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["action", b"action", "available_maps", b"available_maps", "create_game", b"create_game", "data", b"data", "debug", b"debug", "game_info", b"game_info", "id", b"id", "join_game", b"join_game", "leave_game", b"leave_game", "map_command", b"map_command", "obs_action", b"obs_action", "observation", b"observation", "ping", b"ping", "query", b"query", "quick_load", b"quick_load", "quick_save", b"quick_save", "quit", b"quit", "replay_info", b"replay_info", "response", b"response", "restart_game", b"restart_game", "save_map", b"save_map", "save_replay", b"save_replay", "start_replay", b"start_replay", "status", b"status", "step", b"step"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["action", b"action", "available_maps", b"available_maps", "create_game", b"create_game", "data", b"data", "debug", b"debug", "error", b"error", "game_info", b"game_info", "id", b"id", "join_game", b"join_game", "leave_game", b"leave_game", "map_command", b"map_command", "obs_action", b"obs_action", "observation", b"observation", "ping", b"ping", "query", b"query", "quick_load", b"quick_load", "quick_save", b"quick_save", "quit", b"quit", "replay_info", b"replay_info", "response", b"response", "restart_game", b"restart_game", "save_map", b"save_map", "save_replay", b"save_replay", "start_replay", b"start_replay", "status", b"status", "step", b"step"]) -> None: ... - def WhichOneof(self, oneof_group: typing.Literal["response", b"response"]) -> typing.Literal["create_game", "join_game", "restart_game", "start_replay", "leave_game", "quick_save", "quick_load", "quit", "game_info", "observation", "action", "obs_action", "step", "data", "query", "save_replay", "replay_info", "available_maps", "save_map", "map_command", "ping", "debug"] | None: ... + def HasField( + self, + field_name: typing.Literal[ + "action", + b"action", + "available_maps", + b"available_maps", + "create_game", + b"create_game", + "data", + b"data", + "debug", + b"debug", + "game_info", + b"game_info", + "id", + b"id", + "join_game", + b"join_game", + "leave_game", + b"leave_game", + "map_command", + b"map_command", + "obs_action", + b"obs_action", + "observation", + b"observation", + "ping", + b"ping", + "query", + b"query", + "quick_load", + b"quick_load", + "quick_save", + b"quick_save", + "quit", + b"quit", + "replay_info", + b"replay_info", + "response", + b"response", + "restart_game", + b"restart_game", + "save_map", + b"save_map", + "save_replay", + b"save_replay", + "start_replay", + b"start_replay", + "status", + b"status", + "step", + b"step", + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "action", + b"action", + "available_maps", + b"available_maps", + "create_game", + b"create_game", + "data", + b"data", + "debug", + b"debug", + "error", + b"error", + "game_info", + b"game_info", + "id", + b"id", + "join_game", + b"join_game", + "leave_game", + b"leave_game", + "map_command", + b"map_command", + "obs_action", + b"obs_action", + "observation", + b"observation", + "ping", + b"ping", + "query", + b"query", + "quick_load", + b"quick_load", + "quick_save", + b"quick_save", + "quit", + b"quit", + "replay_info", + b"replay_info", + "response", + b"response", + "restart_game", + b"restart_game", + "save_map", + b"save_map", + "save_replay", + b"save_replay", + "start_replay", + b"start_replay", + "status", + b"status", + "step", + b"step", + ], + ) -> None: ... + def WhichOneof( + self, oneof_group: typing.Literal["response", b"response"] + ) -> ( + typing.Literal[ + "create_game", + "join_game", + "restart_game", + "start_replay", + "leave_game", + "quick_save", + "quick_load", + "quit", + "game_info", + "observation", + "action", + "obs_action", + "step", + "data", + "query", + "save_replay", + "replay_info", + "available_maps", + "save_map", + "map_command", + "ping", + "debug", + ] + | None + ): ... global___Response = Response @@ -532,9 +806,45 @@ class RequestCreateGame(google.protobuf.message.Message): random_seed: builtins.int | None = ..., realtime: builtins.bool | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["Map", b"Map", "battlenet_map_name", b"battlenet_map_name", "disable_fog", b"disable_fog", "local_map", b"local_map", "random_seed", b"random_seed", "realtime", b"realtime"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["Map", b"Map", "battlenet_map_name", b"battlenet_map_name", "disable_fog", b"disable_fog", "local_map", b"local_map", "player_setup", b"player_setup", "random_seed", b"random_seed", "realtime", b"realtime"]) -> None: ... - def WhichOneof(self, oneof_group: typing.Literal["Map", b"Map"]) -> typing.Literal["local_map", "battlenet_map_name"] | None: ... + def HasField( + self, + field_name: typing.Literal[ + "Map", + b"Map", + "battlenet_map_name", + b"battlenet_map_name", + "disable_fog", + b"disable_fog", + "local_map", + b"local_map", + "random_seed", + b"random_seed", + "realtime", + b"realtime", + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "Map", + b"Map", + "battlenet_map_name", + b"battlenet_map_name", + "disable_fog", + b"disable_fog", + "local_map", + b"local_map", + "player_setup", + b"player_setup", + "random_seed", + b"random_seed", + "realtime", + b"realtime", + ], + ) -> None: ... + def WhichOneof( + self, oneof_group: typing.Literal["Map", b"Map"] + ) -> typing.Literal["local_map", "battlenet_map_name"] | None: ... global___RequestCreateGame = RequestCreateGame @@ -550,12 +860,7 @@ class LocalMap(google.protobuf.message.Message): into the replay. (260 character max) """ map_data: builtins.bytes - def __init__( - self, - *, - map_path: builtins.str | None = ..., - map_data: builtins.bytes | None = ..., - ) -> None: ... + def __init__(self, *, map_path: builtins.str | None = ..., map_data: builtins.bytes | None = ...) -> None: ... def HasField(self, field_name: typing.Literal["map_data", b"map_data", "map_path", b"map_path"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["map_data", b"map_data", "map_path", b"map_path"]) -> None: ... @@ -569,7 +874,9 @@ class ResponseCreateGame(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _ErrorEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[ResponseCreateGame._Error.ValueType], builtins.type): + class _ErrorEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[ResponseCreateGame._Error.ValueType], builtins.type + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor MissingMap: ResponseCreateGame._Error.ValueType # 1 InvalidMapPath: ResponseCreateGame._Error.ValueType # 2 @@ -597,10 +904,7 @@ class ResponseCreateGame(google.protobuf.message.Message): error: global___ResponseCreateGame.Error.ValueType error_details: builtins.str def __init__( - self, - *, - error: global___ResponseCreateGame.Error.ValueType | None = ..., - error_details: builtins.str | None = ..., + self, *, error: global___ResponseCreateGame.Error.ValueType | None = ..., error_details: builtins.str | None = ... ) -> None: ... def HasField(self, field_name: typing.Literal["error", b"error", "error_details", b"error_details"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["error", b"error", "error_details", b"error_details"]) -> None: ... @@ -660,9 +964,53 @@ class RequestJoinGame(google.protobuf.message.Message): player_name: builtins.str | None = ..., host_ip: builtins.str | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["host_ip", b"host_ip", "observed_player_id", b"observed_player_id", "options", b"options", "participation", b"participation", "player_name", b"player_name", "race", b"race", "server_ports", b"server_ports", "shared_port", b"shared_port"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["client_ports", b"client_ports", "host_ip", b"host_ip", "observed_player_id", b"observed_player_id", "options", b"options", "participation", b"participation", "player_name", b"player_name", "race", b"race", "server_ports", b"server_ports", "shared_port", b"shared_port"]) -> None: ... - def WhichOneof(self, oneof_group: typing.Literal["participation", b"participation"]) -> typing.Literal["race", "observed_player_id"] | None: ... + def HasField( + self, + field_name: typing.Literal[ + "host_ip", + b"host_ip", + "observed_player_id", + b"observed_player_id", + "options", + b"options", + "participation", + b"participation", + "player_name", + b"player_name", + "race", + b"race", + "server_ports", + b"server_ports", + "shared_port", + b"shared_port", + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "client_ports", + b"client_ports", + "host_ip", + b"host_ip", + "observed_player_id", + b"observed_player_id", + "options", + b"options", + "participation", + b"participation", + "player_name", + b"player_name", + "race", + b"race", + "server_ports", + b"server_ports", + "shared_port", + b"shared_port", + ], + ) -> None: ... + def WhichOneof( + self, oneof_group: typing.Literal["participation", b"participation"] + ) -> typing.Literal["race", "observed_player_id"] | None: ... global___RequestJoinGame = RequestJoinGame @@ -675,12 +1023,7 @@ class PortSet(google.protobuf.message.Message): game_port: builtins.int """Game right now needs two internal ports to establish a multiplay game on the local host.""" base_port: builtins.int - def __init__( - self, - *, - game_port: builtins.int | None = ..., - base_port: builtins.int | None = ..., - ) -> None: ... + def __init__(self, *, game_port: builtins.int | None = ..., base_port: builtins.int | None = ...) -> None: ... def HasField(self, field_name: typing.Literal["base_port", b"base_port", "game_port", b"game_port"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["base_port", b"base_port", "game_port", b"game_port"]) -> None: ... @@ -694,7 +1037,9 @@ class ResponseJoinGame(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _ErrorEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[ResponseJoinGame._Error.ValueType], builtins.type): + class _ErrorEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[ResponseJoinGame._Error.ValueType], builtins.type + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor MissingParticipation: ResponseJoinGame._Error.ValueType # 1 InvalidObservedPlayerId: ResponseJoinGame._Error.ValueType # 2 @@ -744,8 +1089,12 @@ class ResponseJoinGame(google.protobuf.message.Message): error: global___ResponseJoinGame.Error.ValueType | None = ..., error_details: builtins.str | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["error", b"error", "error_details", b"error_details", "player_id", b"player_id"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["error", b"error", "error_details", b"error_details", "player_id", b"player_id"]) -> None: ... + def HasField( + self, field_name: typing.Literal["error", b"error", "error_details", b"error_details", "player_id", b"player_id"] + ) -> builtins.bool: ... + def ClearField( + self, field_name: typing.Literal["error", b"error", "error_details", b"error_details", "player_id", b"player_id"] + ) -> None: ... global___ResponseJoinGame = ResponseJoinGame @@ -755,9 +1104,7 @@ class RequestRestartGame(google.protobuf.message.Message): DESCRIPTOR: google.protobuf.descriptor.Descriptor - def __init__( - self, - ) -> None: ... + def __init__(self) -> None: ... global___RequestRestartGame = RequestRestartGame @@ -771,7 +1118,9 @@ class ResponseRestartGame(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _ErrorEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[ResponseRestartGame._Error.ValueType], builtins.type): + class _ErrorEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[ResponseRestartGame._Error.ValueType], builtins.type + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor LaunchError: ResponseRestartGame._Error.ValueType # 1 @@ -792,8 +1141,14 @@ class ResponseRestartGame(google.protobuf.message.Message): error_details: builtins.str | None = ..., need_hard_reset: builtins.bool | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["error", b"error", "error_details", b"error_details", "need_hard_reset", b"need_hard_reset"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["error", b"error", "error_details", b"error_details", "need_hard_reset", b"need_hard_reset"]) -> None: ... + def HasField( + self, + field_name: typing.Literal["error", b"error", "error_details", b"error_details", "need_hard_reset", b"need_hard_reset"], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal["error", b"error", "error_details", b"error_details", "need_hard_reset", b"need_hard_reset"], + ) -> None: ... global___ResponseRestartGame = ResponseRestartGame @@ -834,9 +1189,55 @@ class RequestStartReplay(google.protobuf.message.Message): realtime: builtins.bool | None = ..., record_replay: builtins.bool | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["disable_fog", b"disable_fog", "map_data", b"map_data", "observed_player_id", b"observed_player_id", "options", b"options", "realtime", b"realtime", "record_replay", b"record_replay", "replay", b"replay", "replay_data", b"replay_data", "replay_path", b"replay_path"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["disable_fog", b"disable_fog", "map_data", b"map_data", "observed_player_id", b"observed_player_id", "options", b"options", "realtime", b"realtime", "record_replay", b"record_replay", "replay", b"replay", "replay_data", b"replay_data", "replay_path", b"replay_path"]) -> None: ... - def WhichOneof(self, oneof_group: typing.Literal["replay", b"replay"]) -> typing.Literal["replay_path", "replay_data"] | None: ... + def HasField( + self, + field_name: typing.Literal[ + "disable_fog", + b"disable_fog", + "map_data", + b"map_data", + "observed_player_id", + b"observed_player_id", + "options", + b"options", + "realtime", + b"realtime", + "record_replay", + b"record_replay", + "replay", + b"replay", + "replay_data", + b"replay_data", + "replay_path", + b"replay_path", + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "disable_fog", + b"disable_fog", + "map_data", + b"map_data", + "observed_player_id", + b"observed_player_id", + "options", + b"options", + "realtime", + b"realtime", + "record_replay", + b"record_replay", + "replay", + b"replay", + "replay_data", + b"replay_data", + "replay_path", + b"replay_path", + ], + ) -> None: ... + def WhichOneof( + self, oneof_group: typing.Literal["replay", b"replay"] + ) -> typing.Literal["replay_path", "replay_data"] | None: ... global___RequestStartReplay = RequestStartReplay @@ -848,7 +1249,9 @@ class ResponseStartReplay(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _ErrorEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[ResponseStartReplay._Error.ValueType], builtins.type): + class _ErrorEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[ResponseStartReplay._Error.ValueType], builtins.type + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor MissingReplay: ResponseStartReplay._Error.ValueType # 1 InvalidReplayPath: ResponseStartReplay._Error.ValueType # 2 @@ -872,10 +1275,7 @@ class ResponseStartReplay(google.protobuf.message.Message): error: global___ResponseStartReplay.Error.ValueType error_details: builtins.str def __init__( - self, - *, - error: global___ResponseStartReplay.Error.ValueType | None = ..., - error_details: builtins.str | None = ..., + self, *, error: global___ResponseStartReplay.Error.ValueType | None = ..., error_details: builtins.str | None = ... ) -> None: ... def HasField(self, field_name: typing.Literal["error", b"error", "error_details", b"error_details"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["error", b"error", "error_details", b"error_details"]) -> None: ... @@ -890,11 +1290,7 @@ class RequestMapCommand(google.protobuf.message.Message): TRIGGER_CMD_FIELD_NUMBER: builtins.int trigger_cmd: builtins.str - def __init__( - self, - *, - trigger_cmd: builtins.str | None = ..., - ) -> None: ... + def __init__(self, *, trigger_cmd: builtins.str | None = ...) -> None: ... def HasField(self, field_name: typing.Literal["trigger_cmd", b"trigger_cmd"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["trigger_cmd", b"trigger_cmd"]) -> None: ... @@ -908,7 +1304,9 @@ class ResponseMapCommand(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _ErrorEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[ResponseMapCommand._Error.ValueType], builtins.type): + class _ErrorEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[ResponseMapCommand._Error.ValueType], builtins.type + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor NoTriggerError: ResponseMapCommand._Error.ValueType # 1 @@ -920,10 +1318,7 @@ class ResponseMapCommand(google.protobuf.message.Message): error: global___ResponseMapCommand.Error.ValueType error_details: builtins.str def __init__( - self, - *, - error: global___ResponseMapCommand.Error.ValueType | None = ..., - error_details: builtins.str | None = ..., + self, *, error: global___ResponseMapCommand.Error.ValueType | None = ..., error_details: builtins.str | None = ... ) -> None: ... def HasField(self, field_name: typing.Literal["error", b"error", "error_details", b"error_details"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["error", b"error", "error_details", b"error_details"]) -> None: ... @@ -936,9 +1331,7 @@ class RequestLeaveGame(google.protobuf.message.Message): DESCRIPTOR: google.protobuf.descriptor.Descriptor - def __init__( - self, - ) -> None: ... + def __init__(self) -> None: ... global___RequestLeaveGame = RequestLeaveGame @@ -946,9 +1339,7 @@ global___RequestLeaveGame = RequestLeaveGame class ResponseLeaveGame(google.protobuf.message.Message): DESCRIPTOR: google.protobuf.descriptor.Descriptor - def __init__( - self, - ) -> None: ... + def __init__(self) -> None: ... global___ResponseLeaveGame = ResponseLeaveGame @@ -958,9 +1349,7 @@ class RequestQuickSave(google.protobuf.message.Message): DESCRIPTOR: google.protobuf.descriptor.Descriptor - def __init__( - self, - ) -> None: ... + def __init__(self) -> None: ... global___RequestQuickSave = RequestQuickSave @@ -970,9 +1359,7 @@ class ResponseQuickSave(google.protobuf.message.Message): DESCRIPTOR: google.protobuf.descriptor.Descriptor - def __init__( - self, - ) -> None: ... + def __init__(self) -> None: ... global___ResponseQuickSave = ResponseQuickSave @@ -982,9 +1369,7 @@ class RequestQuickLoad(google.protobuf.message.Message): DESCRIPTOR: google.protobuf.descriptor.Descriptor - def __init__( - self, - ) -> None: ... + def __init__(self) -> None: ... global___RequestQuickLoad = RequestQuickLoad @@ -994,9 +1379,7 @@ class ResponseQuickLoad(google.protobuf.message.Message): DESCRIPTOR: google.protobuf.descriptor.Descriptor - def __init__( - self, - ) -> None: ... + def __init__(self) -> None: ... global___ResponseQuickLoad = ResponseQuickLoad @@ -1006,9 +1389,7 @@ class RequestQuit(google.protobuf.message.Message): DESCRIPTOR: google.protobuf.descriptor.Descriptor - def __init__( - self, - ) -> None: ... + def __init__(self) -> None: ... global___RequestQuit = RequestQuit @@ -1016,9 +1397,7 @@ global___RequestQuit = RequestQuit class ResponseQuit(google.protobuf.message.Message): DESCRIPTOR: google.protobuf.descriptor.Descriptor - def __init__( - self, - ) -> None: ... + def __init__(self) -> None: ... global___ResponseQuit = ResponseQuit @@ -1028,9 +1407,7 @@ class RequestGameInfo(google.protobuf.message.Message): DESCRIPTOR: google.protobuf.descriptor.Descriptor - def __init__( - self, - ) -> None: ... + def __init__(self) -> None: ... global___RequestGameInfo = RequestGameInfo @@ -1066,8 +1443,29 @@ class ResponseGameInfo(google.protobuf.message.Message): start_raw: s2clientprotocol.raw_pb2.StartRaw | None = ..., options: global___InterfaceOptions | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["local_map_path", b"local_map_path", "map_name", b"map_name", "options", b"options", "start_raw", b"start_raw"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["local_map_path", b"local_map_path", "map_name", b"map_name", "mod_names", b"mod_names", "options", b"options", "player_info", b"player_info", "start_raw", b"start_raw"]) -> None: ... + def HasField( + self, + field_name: typing.Literal[ + "local_map_path", b"local_map_path", "map_name", b"map_name", "options", b"options", "start_raw", b"start_raw" + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "local_map_path", + b"local_map_path", + "map_name", + b"map_name", + "mod_names", + b"mod_names", + "options", + b"options", + "player_info", + b"player_info", + "start_raw", + b"start_raw", + ], + ) -> None: ... global___ResponseGameInfo = ResponseGameInfo @@ -1082,12 +1480,7 @@ class RequestObservation(google.protobuf.message.Message): disable_fog: builtins.bool game_loop: builtins.int """In realtime the request will only return once the simulation game loop has reached this value. When not realtime this value is ignored.""" - def __init__( - self, - *, - disable_fog: builtins.bool | None = ..., - game_loop: builtins.int | None = ..., - ) -> None: ... + def __init__(self, *, disable_fog: builtins.bool | None = ..., game_loop: builtins.int | None = ...) -> None: ... def HasField(self, field_name: typing.Literal["disable_fog", b"disable_fog", "game_loop", b"game_loop"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["disable_fog", b"disable_fog", "game_loop", b"game_loop"]) -> None: ... @@ -1128,7 +1521,21 @@ class ResponseObservation(google.protobuf.message.Message): chat: collections.abc.Iterable[global___ChatReceived] | None = ..., ) -> None: ... def HasField(self, field_name: typing.Literal["observation", b"observation"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["action_errors", b"action_errors", "actions", b"actions", "chat", b"chat", "observation", b"observation", "player_result", b"player_result"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "action_errors", + b"action_errors", + "actions", + b"actions", + "chat", + b"chat", + "observation", + b"observation", + "player_result", + b"player_result", + ], + ) -> None: ... global___ResponseObservation = ResponseObservation @@ -1140,12 +1547,7 @@ class ChatReceived(google.protobuf.message.Message): MESSAGE_FIELD_NUMBER: builtins.int player_id: builtins.int message: builtins.str - def __init__( - self, - *, - player_id: builtins.int | None = ..., - message: builtins.str | None = ..., - ) -> None: ... + def __init__(self, *, player_id: builtins.int | None = ..., message: builtins.str | None = ...) -> None: ... def HasField(self, field_name: typing.Literal["message", b"message", "player_id", b"player_id"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["message", b"message", "player_id", b"player_id"]) -> None: ... @@ -1160,11 +1562,7 @@ class RequestAction(google.protobuf.message.Message): ACTIONS_FIELD_NUMBER: builtins.int @property def actions(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___Action]: ... - def __init__( - self, - *, - actions: collections.abc.Iterable[global___Action] | None = ..., - ) -> None: ... + def __init__(self, *, actions: collections.abc.Iterable[global___Action] | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["actions", b"actions"]) -> None: ... global___RequestAction = RequestAction @@ -1175,11 +1573,11 @@ class ResponseAction(google.protobuf.message.Message): RESULT_FIELD_NUMBER: builtins.int @property - def result(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[s2clientprotocol.error_pb2.ActionResult.ValueType]: ... - def __init__( + def result( self, - *, - result: collections.abc.Iterable[s2clientprotocol.error_pb2.ActionResult.ValueType] | None = ..., + ) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[s2clientprotocol.error_pb2.ActionResult.ValueType]: ... + def __init__( + self, *, result: collections.abc.Iterable[s2clientprotocol.error_pb2.ActionResult.ValueType] | None = ... ) -> None: ... def ClearField(self, field_name: typing.Literal["result", b"result"]) -> None: ... @@ -1194,11 +1592,7 @@ class RequestObserverAction(google.protobuf.message.Message): ACTIONS_FIELD_NUMBER: builtins.int @property def actions(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___ObserverAction]: ... - def __init__( - self, - *, - actions: collections.abc.Iterable[global___ObserverAction] | None = ..., - ) -> None: ... + def __init__(self, *, actions: collections.abc.Iterable[global___ObserverAction] | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["actions", b"actions"]) -> None: ... global___RequestObserverAction = RequestObserverAction @@ -1207,9 +1601,7 @@ global___RequestObserverAction = RequestObserverAction class ResponseObserverAction(google.protobuf.message.Message): DESCRIPTOR: google.protobuf.descriptor.Descriptor - def __init__( - self, - ) -> None: ... + def __init__(self) -> None: ... global___ResponseObserverAction = ResponseObserverAction @@ -1222,11 +1614,7 @@ class RequestStep(google.protobuf.message.Message): COUNT_FIELD_NUMBER: builtins.int count: builtins.int """Number of game loops to simulate for the next frame.""" - def __init__( - self, - *, - count: builtins.int | None = ..., - ) -> None: ... + def __init__(self, *, count: builtins.int | None = ...) -> None: ... def HasField(self, field_name: typing.Literal["count", b"count"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["count", b"count"]) -> None: ... @@ -1243,11 +1631,7 @@ class ResponseStep(google.protobuf.message.Message): representable as a positive fixed point number. When we reach the "end of time", permanently pause the game and end the game for all. """ - def __init__( - self, - *, - simulation_loop: builtins.int | None = ..., - ) -> None: ... + def __init__(self, *, simulation_loop: builtins.int | None = ...) -> None: ... def HasField(self, field_name: typing.Literal["simulation_loop", b"simulation_loop"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["simulation_loop", b"simulation_loop"]) -> None: ... @@ -1278,8 +1662,36 @@ class RequestData(google.protobuf.message.Message): buff_id: builtins.bool | None = ..., effect_id: builtins.bool | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["ability_id", b"ability_id", "buff_id", b"buff_id", "effect_id", b"effect_id", "unit_type_id", b"unit_type_id", "upgrade_id", b"upgrade_id"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["ability_id", b"ability_id", "buff_id", b"buff_id", "effect_id", b"effect_id", "unit_type_id", b"unit_type_id", "upgrade_id", b"upgrade_id"]) -> None: ... + def HasField( + self, + field_name: typing.Literal[ + "ability_id", + b"ability_id", + "buff_id", + b"buff_id", + "effect_id", + b"effect_id", + "unit_type_id", + b"unit_type_id", + "upgrade_id", + b"upgrade_id", + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "ability_id", + b"ability_id", + "buff_id", + b"buff_id", + "effect_id", + b"effect_id", + "unit_type_id", + b"unit_type_id", + "upgrade_id", + b"upgrade_id", + ], + ) -> None: ... global___RequestData = RequestData @@ -1293,15 +1705,25 @@ class ResponseData(google.protobuf.message.Message): BUFFS_FIELD_NUMBER: builtins.int EFFECTS_FIELD_NUMBER: builtins.int @property - def abilities(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[s2clientprotocol.data_pb2.AbilityData]: ... + def abilities( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[s2clientprotocol.data_pb2.AbilityData]: ... @property - def units(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[s2clientprotocol.data_pb2.UnitTypeData]: ... + def units( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[s2clientprotocol.data_pb2.UnitTypeData]: ... @property - def upgrades(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[s2clientprotocol.data_pb2.UpgradeData]: ... + def upgrades( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[s2clientprotocol.data_pb2.UpgradeData]: ... @property - def buffs(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[s2clientprotocol.data_pb2.BuffData]: ... + def buffs( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[s2clientprotocol.data_pb2.BuffData]: ... @property - def effects(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[s2clientprotocol.data_pb2.EffectData]: ... + def effects( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[s2clientprotocol.data_pb2.EffectData]: ... def __init__( self, *, @@ -1311,7 +1733,12 @@ class ResponseData(google.protobuf.message.Message): buffs: collections.abc.Iterable[s2clientprotocol.data_pb2.BuffData] | None = ..., effects: collections.abc.Iterable[s2clientprotocol.data_pb2.EffectData] | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["abilities", b"abilities", "buffs", b"buffs", "effects", b"effects", "units", b"units", "upgrades", b"upgrades"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "abilities", b"abilities", "buffs", b"buffs", "effects", b"effects", "units", b"units", "upgrades", b"upgrades" + ], + ) -> None: ... global___ResponseData = ResponseData @@ -1321,9 +1748,7 @@ class RequestSaveReplay(google.protobuf.message.Message): DESCRIPTOR: google.protobuf.descriptor.Descriptor - def __init__( - self, - ) -> None: ... + def __init__(self) -> None: ... global___RequestSaveReplay = RequestSaveReplay @@ -1333,11 +1758,7 @@ class ResponseSaveReplay(google.protobuf.message.Message): DATA_FIELD_NUMBER: builtins.int data: builtins.bytes - def __init__( - self, - *, - data: builtins.bytes | None = ..., - ) -> None: ... + def __init__(self, *, data: builtins.bytes | None = ...) -> None: ... def HasField(self, field_name: typing.Literal["data", b"data"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["data", b"data"]) -> None: ... @@ -1364,9 +1785,21 @@ class RequestReplayInfo(google.protobuf.message.Message): replay_data: builtins.bytes | None = ..., download_data: builtins.bool | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["download_data", b"download_data", "replay", b"replay", "replay_data", b"replay_data", "replay_path", b"replay_path"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["download_data", b"download_data", "replay", b"replay", "replay_data", b"replay_data", "replay_path", b"replay_path"]) -> None: ... - def WhichOneof(self, oneof_group: typing.Literal["replay", b"replay"]) -> typing.Literal["replay_path", "replay_data"] | None: ... + def HasField( + self, + field_name: typing.Literal[ + "download_data", b"download_data", "replay", b"replay", "replay_data", b"replay_data", "replay_path", b"replay_path" + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "download_data", b"download_data", "replay", b"replay", "replay_data", b"replay_data", "replay_path", b"replay_path" + ], + ) -> None: ... + def WhichOneof( + self, oneof_group: typing.Literal["replay", b"replay"] + ) -> typing.Literal["replay_path", "replay_data"] | None: ... global___RequestReplayInfo = RequestReplayInfo @@ -1392,8 +1825,32 @@ class PlayerInfoExtra(google.protobuf.message.Message): player_mmr: builtins.int | None = ..., player_apm: builtins.int | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["player_apm", b"player_apm", "player_info", b"player_info", "player_mmr", b"player_mmr", "player_result", b"player_result"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["player_apm", b"player_apm", "player_info", b"player_info", "player_mmr", b"player_mmr", "player_result", b"player_result"]) -> None: ... + def HasField( + self, + field_name: typing.Literal[ + "player_apm", + b"player_apm", + "player_info", + b"player_info", + "player_mmr", + b"player_mmr", + "player_result", + b"player_result", + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "player_apm", + b"player_apm", + "player_info", + b"player_info", + "player_mmr", + b"player_mmr", + "player_result", + b"player_result", + ], + ) -> None: ... global___PlayerInfoExtra = PlayerInfoExtra @@ -1405,7 +1862,9 @@ class ResponseReplayInfo(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _ErrorEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[ResponseReplayInfo._Error.ValueType], builtins.type): + class _ErrorEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[ResponseReplayInfo._Error.ValueType], builtins.type + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor MissingReplay: ResponseReplayInfo._Error.ValueType # 1 InvalidReplayPath: ResponseReplayInfo._Error.ValueType # 2 @@ -1458,8 +1917,58 @@ class ResponseReplayInfo(google.protobuf.message.Message): error: global___ResponseReplayInfo.Error.ValueType | None = ..., error_details: builtins.str | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["base_build", b"base_build", "data_build", b"data_build", "data_version", b"data_version", "error", b"error", "error_details", b"error_details", "game_duration_loops", b"game_duration_loops", "game_duration_seconds", b"game_duration_seconds", "game_version", b"game_version", "local_map_path", b"local_map_path", "map_name", b"map_name"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["base_build", b"base_build", "data_build", b"data_build", "data_version", b"data_version", "error", b"error", "error_details", b"error_details", "game_duration_loops", b"game_duration_loops", "game_duration_seconds", b"game_duration_seconds", "game_version", b"game_version", "local_map_path", b"local_map_path", "map_name", b"map_name", "player_info", b"player_info"]) -> None: ... + def HasField( + self, + field_name: typing.Literal[ + "base_build", + b"base_build", + "data_build", + b"data_build", + "data_version", + b"data_version", + "error", + b"error", + "error_details", + b"error_details", + "game_duration_loops", + b"game_duration_loops", + "game_duration_seconds", + b"game_duration_seconds", + "game_version", + b"game_version", + "local_map_path", + b"local_map_path", + "map_name", + b"map_name", + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "base_build", + b"base_build", + "data_build", + b"data_build", + "data_version", + b"data_version", + "error", + b"error", + "error_details", + b"error_details", + "game_duration_loops", + b"game_duration_loops", + "game_duration_seconds", + b"game_duration_seconds", + "game_version", + b"game_version", + "local_map_path", + b"local_map_path", + "map_name", + b"map_name", + "player_info", + b"player_info", + ], + ) -> None: ... global___ResponseReplayInfo = ResponseReplayInfo @@ -1469,9 +1978,7 @@ class RequestAvailableMaps(google.protobuf.message.Message): DESCRIPTOR: google.protobuf.descriptor.Descriptor - def __init__( - self, - ) -> None: ... + def __init__(self) -> None: ... global___RequestAvailableMaps = RequestAvailableMaps @@ -1500,7 +2007,9 @@ class ResponseAvailableMaps(google.protobuf.message.Message): local_map_paths: collections.abc.Iterable[builtins.str] | None = ..., battlenet_map_names: collections.abc.Iterable[builtins.str] | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["battlenet_map_names", b"battlenet_map_names", "local_map_paths", b"local_map_paths"]) -> None: ... + def ClearField( + self, field_name: typing.Literal["battlenet_map_names", b"battlenet_map_names", "local_map_paths", b"local_map_paths"] + ) -> None: ... global___ResponseAvailableMaps = ResponseAvailableMaps @@ -1518,12 +2027,7 @@ class RequestSaveMap(google.protobuf.message.Message): """Path the game process will write to, relative to the temp directory. (260 character max)""" map_data: builtins.bytes """Binary map data of a .SC2Map.""" - def __init__( - self, - *, - map_path: builtins.str | None = ..., - map_data: builtins.bytes | None = ..., - ) -> None: ... + def __init__(self, *, map_path: builtins.str | None = ..., map_data: builtins.bytes | None = ...) -> None: ... def HasField(self, field_name: typing.Literal["map_data", b"map_data", "map_path", b"map_path"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["map_data", b"map_data", "map_path", b"map_path"]) -> None: ... @@ -1537,7 +2041,9 @@ class ResponseSaveMap(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _ErrorEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[ResponseSaveMap._Error.ValueType], builtins.type): + class _ErrorEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[ResponseSaveMap._Error.ValueType], builtins.type + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor InvalidMapData: ResponseSaveMap._Error.ValueType # 1 @@ -1546,11 +2052,7 @@ class ResponseSaveMap(google.protobuf.message.Message): ERROR_FIELD_NUMBER: builtins.int error: global___ResponseSaveMap.Error.ValueType - def __init__( - self, - *, - error: global___ResponseSaveMap.Error.ValueType | None = ..., - ) -> None: ... + def __init__(self, *, error: global___ResponseSaveMap.Error.ValueType | None = ...) -> None: ... def HasField(self, field_name: typing.Literal["error", b"error"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["error", b"error"]) -> None: ... @@ -1562,9 +2064,7 @@ class RequestPing(google.protobuf.message.Message): DESCRIPTOR: google.protobuf.descriptor.Descriptor - def __init__( - self, - ) -> None: ... + def __init__(self) -> None: ... global___RequestPing = RequestPing @@ -1588,8 +2088,32 @@ class ResponsePing(google.protobuf.message.Message): data_build: builtins.int | None = ..., base_build: builtins.int | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["base_build", b"base_build", "data_build", b"data_build", "data_version", b"data_version", "game_version", b"game_version"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["base_build", b"base_build", "data_build", b"data_build", "data_version", b"data_version", "game_version", b"game_version"]) -> None: ... + def HasField( + self, + field_name: typing.Literal[ + "base_build", + b"base_build", + "data_build", + b"data_build", + "data_version", + b"data_version", + "game_version", + b"game_version", + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "base_build", + b"base_build", + "data_build", + b"data_build", + "data_version", + b"data_version", + "game_version", + b"game_version", + ], + ) -> None: ... global___ResponsePing = ResponsePing @@ -1601,12 +2125,10 @@ class RequestDebug(google.protobuf.message.Message): DEBUG_FIELD_NUMBER: builtins.int @property - def debug(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[s2clientprotocol.debug_pb2.DebugCommand]: ... - def __init__( + def debug( self, - *, - debug: collections.abc.Iterable[s2clientprotocol.debug_pb2.DebugCommand] | None = ..., - ) -> None: ... + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[s2clientprotocol.debug_pb2.DebugCommand]: ... + def __init__(self, *, debug: collections.abc.Iterable[s2clientprotocol.debug_pb2.DebugCommand] | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["debug", b"debug"]) -> None: ... global___RequestDebug = RequestDebug @@ -1615,9 +2137,7 @@ global___RequestDebug = RequestDebug class ResponseDebug(google.protobuf.message.Message): DESCRIPTOR: google.protobuf.descriptor.Descriptor - def __init__( - self, - ) -> None: ... + def __init__(self) -> None: ... global___ResponseDebug = ResponseDebug @@ -1645,8 +2165,18 @@ class PlayerSetup(google.protobuf.message.Message): player_name: builtins.str | None = ..., ai_build: global___AIBuild.ValueType | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["ai_build", b"ai_build", "difficulty", b"difficulty", "player_name", b"player_name", "race", b"race", "type", b"type"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["ai_build", b"ai_build", "difficulty", b"difficulty", "player_name", b"player_name", "race", b"race", "type", b"type"]) -> None: ... + def HasField( + self, + field_name: typing.Literal[ + "ai_build", b"ai_build", "difficulty", b"difficulty", "player_name", b"player_name", "race", b"race", "type", b"type" + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "ai_build", b"ai_build", "difficulty", b"difficulty", "player_name", b"player_name", "race", b"race", "type", b"type" + ], + ) -> None: ... global___PlayerSetup = PlayerSetup @@ -1680,8 +2210,36 @@ class SpatialCameraSetup(google.protobuf.message.Message): crop_to_playable_area: builtins.bool | None = ..., allow_cheating_layers: builtins.bool | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["allow_cheating_layers", b"allow_cheating_layers", "crop_to_playable_area", b"crop_to_playable_area", "minimap_resolution", b"minimap_resolution", "resolution", b"resolution", "width", b"width"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["allow_cheating_layers", b"allow_cheating_layers", "crop_to_playable_area", b"crop_to_playable_area", "minimap_resolution", b"minimap_resolution", "resolution", b"resolution", "width", b"width"]) -> None: ... + def HasField( + self, + field_name: typing.Literal[ + "allow_cheating_layers", + b"allow_cheating_layers", + "crop_to_playable_area", + b"crop_to_playable_area", + "minimap_resolution", + b"minimap_resolution", + "resolution", + b"resolution", + "width", + b"width", + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "allow_cheating_layers", + b"allow_cheating_layers", + "crop_to_playable_area", + b"crop_to_playable_area", + "minimap_resolution", + b"minimap_resolution", + "resolution", + b"resolution", + "width", + b"width", + ], + ) -> None: ... global___SpatialCameraSetup = SpatialCameraSetup @@ -1739,8 +2297,52 @@ class InterfaceOptions(google.protobuf.message.Message): raw_affects_selection: builtins.bool | None = ..., raw_crop_to_playable_area: builtins.bool | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["feature_layer", b"feature_layer", "raw", b"raw", "raw_affects_selection", b"raw_affects_selection", "raw_crop_to_playable_area", b"raw_crop_to_playable_area", "render", b"render", "score", b"score", "show_burrowed_shadows", b"show_burrowed_shadows", "show_cloaked", b"show_cloaked", "show_placeholders", b"show_placeholders"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["feature_layer", b"feature_layer", "raw", b"raw", "raw_affects_selection", b"raw_affects_selection", "raw_crop_to_playable_area", b"raw_crop_to_playable_area", "render", b"render", "score", b"score", "show_burrowed_shadows", b"show_burrowed_shadows", "show_cloaked", b"show_cloaked", "show_placeholders", b"show_placeholders"]) -> None: ... + def HasField( + self, + field_name: typing.Literal[ + "feature_layer", + b"feature_layer", + "raw", + b"raw", + "raw_affects_selection", + b"raw_affects_selection", + "raw_crop_to_playable_area", + b"raw_crop_to_playable_area", + "render", + b"render", + "score", + b"score", + "show_burrowed_shadows", + b"show_burrowed_shadows", + "show_cloaked", + b"show_cloaked", + "show_placeholders", + b"show_placeholders", + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "feature_layer", + b"feature_layer", + "raw", + b"raw", + "raw_affects_selection", + b"raw_affects_selection", + "raw_crop_to_playable_area", + b"raw_crop_to_playable_area", + "render", + b"render", + "score", + b"score", + "show_burrowed_shadows", + b"show_burrowed_shadows", + "show_cloaked", + b"show_cloaked", + "show_placeholders", + b"show_placeholders", + ], + ) -> None: ... global___InterfaceOptions = InterfaceOptions @@ -1777,8 +2379,44 @@ class PlayerInfo(google.protobuf.message.Message): ai_build: global___AIBuild.ValueType | None = ..., player_name: builtins.str | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["ai_build", b"ai_build", "difficulty", b"difficulty", "player_id", b"player_id", "player_name", b"player_name", "race_actual", b"race_actual", "race_requested", b"race_requested", "type", b"type"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["ai_build", b"ai_build", "difficulty", b"difficulty", "player_id", b"player_id", "player_name", b"player_name", "race_actual", b"race_actual", "race_requested", b"race_requested", "type", b"type"]) -> None: ... + def HasField( + self, + field_name: typing.Literal[ + "ai_build", + b"ai_build", + "difficulty", + b"difficulty", + "player_id", + b"player_id", + "player_name", + b"player_name", + "race_actual", + b"race_actual", + "race_requested", + b"race_requested", + "type", + b"type", + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "ai_build", + b"ai_build", + "difficulty", + b"difficulty", + "player_id", + b"player_id", + "player_name", + b"player_name", + "race_actual", + b"race_actual", + "race_requested", + b"race_requested", + "type", + b"type", + ], + ) -> None: ... global___PlayerInfo = PlayerInfo @@ -1827,8 +2465,60 @@ class PlayerCommon(google.protobuf.message.Message): warp_gate_count: builtins.int | None = ..., larva_count: builtins.int | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["army_count", b"army_count", "food_army", b"food_army", "food_cap", b"food_cap", "food_used", b"food_used", "food_workers", b"food_workers", "idle_worker_count", b"idle_worker_count", "larva_count", b"larva_count", "minerals", b"minerals", "player_id", b"player_id", "vespene", b"vespene", "warp_gate_count", b"warp_gate_count"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["army_count", b"army_count", "food_army", b"food_army", "food_cap", b"food_cap", "food_used", b"food_used", "food_workers", b"food_workers", "idle_worker_count", b"idle_worker_count", "larva_count", b"larva_count", "minerals", b"minerals", "player_id", b"player_id", "vespene", b"vespene", "warp_gate_count", b"warp_gate_count"]) -> None: ... + def HasField( + self, + field_name: typing.Literal[ + "army_count", + b"army_count", + "food_army", + b"food_army", + "food_cap", + b"food_cap", + "food_used", + b"food_used", + "food_workers", + b"food_workers", + "idle_worker_count", + b"idle_worker_count", + "larva_count", + b"larva_count", + "minerals", + b"minerals", + "player_id", + b"player_id", + "vespene", + b"vespene", + "warp_gate_count", + b"warp_gate_count", + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "army_count", + b"army_count", + "food_army", + b"food_army", + "food_cap", + b"food_cap", + "food_used", + b"food_used", + "food_workers", + b"food_workers", + "idle_worker_count", + b"idle_worker_count", + "larva_count", + b"larva_count", + "minerals", + b"minerals", + "player_id", + b"player_id", + "vespene", + b"vespene", + "warp_gate_count", + b"warp_gate_count", + ], + ) -> None: ... global___PlayerCommon = PlayerCommon @@ -1851,7 +2541,9 @@ class Observation(google.protobuf.message.Message): @property def alerts(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[global___Alert.ValueType]: ... @property - def abilities(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[s2clientprotocol.common_pb2.AvailableAbility]: + def abilities( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[s2clientprotocol.common_pb2.AvailableAbility]: """Abilities available in the selection. Enabled if in this list, disabled otherwise.""" @property @@ -1885,8 +2577,48 @@ class Observation(google.protobuf.message.Message): render_data: s2clientprotocol.spatial_pb2.ObservationRender | None = ..., ui_data: s2clientprotocol.ui_pb2.ObservationUI | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["feature_layer_data", b"feature_layer_data", "game_loop", b"game_loop", "player_common", b"player_common", "raw_data", b"raw_data", "render_data", b"render_data", "score", b"score", "ui_data", b"ui_data"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["abilities", b"abilities", "alerts", b"alerts", "feature_layer_data", b"feature_layer_data", "game_loop", b"game_loop", "player_common", b"player_common", "raw_data", b"raw_data", "render_data", b"render_data", "score", b"score", "ui_data", b"ui_data"]) -> None: ... + def HasField( + self, + field_name: typing.Literal[ + "feature_layer_data", + b"feature_layer_data", + "game_loop", + b"game_loop", + "player_common", + b"player_common", + "raw_data", + b"raw_data", + "render_data", + b"render_data", + "score", + b"score", + "ui_data", + b"ui_data", + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "abilities", + b"abilities", + "alerts", + b"alerts", + "feature_layer_data", + b"feature_layer_data", + "game_loop", + b"game_loop", + "player_common", + b"player_common", + "raw_data", + b"raw_data", + "render_data", + b"render_data", + "score", + b"score", + "ui_data", + b"ui_data", + ], + ) -> None: ... global___Observation = Observation @@ -1932,8 +2664,40 @@ class Action(google.protobuf.message.Message): action_chat: global___ActionChat | None = ..., game_loop: builtins.int | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["action_chat", b"action_chat", "action_feature_layer", b"action_feature_layer", "action_raw", b"action_raw", "action_render", b"action_render", "action_ui", b"action_ui", "game_loop", b"game_loop"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["action_chat", b"action_chat", "action_feature_layer", b"action_feature_layer", "action_raw", b"action_raw", "action_render", b"action_render", "action_ui", b"action_ui", "game_loop", b"game_loop"]) -> None: ... + def HasField( + self, + field_name: typing.Literal[ + "action_chat", + b"action_chat", + "action_feature_layer", + b"action_feature_layer", + "action_raw", + b"action_raw", + "action_render", + b"action_render", + "action_ui", + b"action_ui", + "game_loop", + b"game_loop", + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "action_chat", + b"action_chat", + "action_feature_layer", + b"action_feature_layer", + "action_raw", + b"action_raw", + "action_render", + b"action_render", + "action_ui", + b"action_ui", + "game_loop", + b"game_loop", + ], + ) -> None: ... global___Action = Action @@ -1945,7 +2709,9 @@ class ActionChat(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _ChannelEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[ActionChat._Channel.ValueType], builtins.type): + class _ChannelEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[ActionChat._Channel.ValueType], builtins.type + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor Broadcast: ActionChat._Channel.ValueType # 1 Team: ActionChat._Channel.ValueType # 2 @@ -1959,10 +2725,7 @@ class ActionChat(google.protobuf.message.Message): channel: global___ActionChat.Channel.ValueType message: builtins.str def __init__( - self, - *, - channel: global___ActionChat.Channel.ValueType | None = ..., - message: builtins.str | None = ..., + self, *, channel: global___ActionChat.Channel.ValueType | None = ..., message: builtins.str | None = ... ) -> None: ... def HasField(self, field_name: typing.Literal["channel", b"channel", "message", b"message"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["channel", b"channel", "message", b"message"]) -> None: ... @@ -1987,8 +2750,12 @@ class ActionError(google.protobuf.message.Message): ability_id: builtins.int | None = ..., result: s2clientprotocol.error_pb2.ActionResult.ValueType | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["ability_id", b"ability_id", "result", b"result", "unit_tag", b"unit_tag"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["ability_id", b"ability_id", "result", b"result", "unit_tag", b"unit_tag"]) -> None: ... + def HasField( + self, field_name: typing.Literal["ability_id", b"ability_id", "result", b"result", "unit_tag", b"unit_tag"] + ) -> builtins.bool: ... + def ClearField( + self, field_name: typing.Literal["ability_id", b"ability_id", "result", b"result", "unit_tag", b"unit_tag"] + ) -> None: ... global___ActionError = ActionError @@ -2020,9 +2787,39 @@ class ObserverAction(google.protobuf.message.Message): camera_follow_player: global___ActionObserverCameraFollowPlayer | None = ..., camera_follow_units: global___ActionObserverCameraFollowUnits | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["action", b"action", "camera_follow_player", b"camera_follow_player", "camera_follow_units", b"camera_follow_units", "camera_move", b"camera_move", "player_perspective", b"player_perspective"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["action", b"action", "camera_follow_player", b"camera_follow_player", "camera_follow_units", b"camera_follow_units", "camera_move", b"camera_move", "player_perspective", b"player_perspective"]) -> None: ... - def WhichOneof(self, oneof_group: typing.Literal["action", b"action"]) -> typing.Literal["player_perspective", "camera_move", "camera_follow_player", "camera_follow_units"] | None: ... + def HasField( + self, + field_name: typing.Literal[ + "action", + b"action", + "camera_follow_player", + b"camera_follow_player", + "camera_follow_units", + b"camera_follow_units", + "camera_move", + b"camera_move", + "player_perspective", + b"player_perspective", + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "action", + b"action", + "camera_follow_player", + b"camera_follow_player", + "camera_follow_units", + b"camera_follow_units", + "camera_move", + b"camera_move", + "player_perspective", + b"player_perspective", + ], + ) -> None: ... + def WhichOneof( + self, oneof_group: typing.Literal["action", b"action"] + ) -> typing.Literal["player_perspective", "camera_move", "camera_follow_player", "camera_follow_units"] | None: ... global___ObserverAction = ObserverAction @@ -2033,11 +2830,7 @@ class ActionObserverPlayerPerspective(google.protobuf.message.Message): PLAYER_ID_FIELD_NUMBER: builtins.int player_id: builtins.int """0 to observe "Everyone" """ - def __init__( - self, - *, - player_id: builtins.int | None = ..., - ) -> None: ... + def __init__(self, *, player_id: builtins.int | None = ...) -> None: ... def HasField(self, field_name: typing.Literal["player_id", b"player_id"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["player_id", b"player_id"]) -> None: ... @@ -2056,10 +2849,7 @@ class ActionObserverCameraMove(google.protobuf.message.Message): @property def world_pos(self) -> s2clientprotocol.common_pb2.Point2D: ... def __init__( - self, - *, - world_pos: s2clientprotocol.common_pb2.Point2D | None = ..., - distance: builtins.float | None = ..., + self, *, world_pos: s2clientprotocol.common_pb2.Point2D | None = ..., distance: builtins.float | None = ... ) -> None: ... def HasField(self, field_name: typing.Literal["distance", b"distance", "world_pos", b"world_pos"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["distance", b"distance", "world_pos", b"world_pos"]) -> None: ... @@ -2073,11 +2863,7 @@ class ActionObserverCameraFollowPlayer(google.protobuf.message.Message): PLAYER_ID_FIELD_NUMBER: builtins.int player_id: builtins.int """Not implemented. Value must be [1, 15]""" - def __init__( - self, - *, - player_id: builtins.int | None = ..., - ) -> None: ... + def __init__(self, *, player_id: builtins.int | None = ...) -> None: ... def HasField(self, field_name: typing.Literal["player_id", b"player_id"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["player_id", b"player_id"]) -> None: ... @@ -2090,11 +2876,7 @@ class ActionObserverCameraFollowUnits(google.protobuf.message.Message): UNIT_TAGS_FIELD_NUMBER: builtins.int @property def unit_tags(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.int]: ... - def __init__( - self, - *, - unit_tags: collections.abc.Iterable[builtins.int] | None = ..., - ) -> None: ... + def __init__(self, *, unit_tags: collections.abc.Iterable[builtins.int] | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["unit_tags", b"unit_tags"]) -> None: ... global___ActionObserverCameraFollowUnits = ActionObserverCameraFollowUnits @@ -2107,12 +2889,7 @@ class PlayerResult(google.protobuf.message.Message): RESULT_FIELD_NUMBER: builtins.int player_id: builtins.int result: global___Result.ValueType - def __init__( - self, - *, - player_id: builtins.int | None = ..., - result: global___Result.ValueType | None = ..., - ) -> None: ... + def __init__(self, *, player_id: builtins.int | None = ..., result: global___Result.ValueType | None = ...) -> None: ... def HasField(self, field_name: typing.Literal["player_id", b"player_id", "result", b"result"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["player_id", b"player_id", "result", b"result"]) -> None: ... diff --git a/stubs/s2clientprotocol/s2clientprotocol/score_pb2.pyi b/stubs/s2clientprotocol/s2clientprotocol/score_pb2.pyi index 95ee7dd7fde5..611578a5c889 100644 --- a/stubs/s2clientprotocol/s2clientprotocol/score_pb2.pyi +++ b/stubs/s2clientprotocol/s2clientprotocol/score_pb2.pyi @@ -26,7 +26,9 @@ class Score(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _ScoreTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[Score._ScoreType.ValueType], builtins.type): + class _ScoreTypeEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[Score._ScoreType.ValueType], builtins.type + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor Curriculum: Score._ScoreType.ValueType # 1 """map generated score (from curriculum maps with special scoring)""" @@ -54,8 +56,12 @@ class Score(google.protobuf.message.Message): score: builtins.int | None = ..., score_details: global___ScoreDetails | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["score", b"score", "score_details", b"score_details", "score_type", b"score_type"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["score", b"score", "score_details", b"score_details", "score_type", b"score_type"]) -> None: ... + def HasField( + self, field_name: typing.Literal["score", b"score", "score_details", b"score_details", "score_type", b"score_type"] + ) -> builtins.bool: ... + def ClearField( + self, field_name: typing.Literal["score", b"score", "score_details", b"score_details", "score_type", b"score_type"] + ) -> None: ... global___Score = Score @@ -83,8 +89,18 @@ class CategoryScoreDetails(google.protobuf.message.Message): technology: builtins.float | None = ..., upgrade: builtins.float | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["army", b"army", "economy", b"economy", "none", b"none", "technology", b"technology", "upgrade", b"upgrade"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["army", b"army", "economy", b"economy", "none", b"none", "technology", b"technology", "upgrade", b"upgrade"]) -> None: ... + def HasField( + self, + field_name: typing.Literal[ + "army", b"army", "economy", b"economy", "none", b"none", "technology", b"technology", "upgrade", b"upgrade" + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "army", b"army", "economy", b"economy", "none", b"none", "technology", b"technology", "upgrade", b"upgrade" + ], + ) -> None: ... global___CategoryScoreDetails = CategoryScoreDetails @@ -99,13 +115,11 @@ class VitalScoreDetails(google.protobuf.message.Message): shields: builtins.float energy: builtins.float def __init__( - self, - *, - life: builtins.float | None = ..., - shields: builtins.float | None = ..., - energy: builtins.float | None = ..., + self, *, life: builtins.float | None = ..., shields: builtins.float | None = ..., energy: builtins.float | None = ... ) -> None: ... - def HasField(self, field_name: typing.Literal["energy", b"energy", "life", b"life", "shields", b"shields"]) -> builtins.bool: ... + def HasField( + self, field_name: typing.Literal["energy", b"energy", "life", b"life", "shields", b"shields"] + ) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["energy", b"energy", "life", b"life", "shields", b"shields"]) -> None: ... global___VitalScoreDetails = VitalScoreDetails @@ -192,7 +206,7 @@ class ScoreDetails(google.protobuf.message.Message): @property def lost_minerals(self) -> global___CategoryScoreDetails: - """ Sum of lost minerals for the player in each category.""" + """Sum of lost minerals for the player in each category.""" @property def lost_vespene(self) -> global___CategoryScoreDetails: @@ -266,7 +280,127 @@ class ScoreDetails(google.protobuf.message.Message): current_apm: builtins.float | None = ..., current_effective_apm: builtins.float | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["collected_minerals", b"collected_minerals", "collected_vespene", b"collected_vespene", "collection_rate_minerals", b"collection_rate_minerals", "collection_rate_vespene", b"collection_rate_vespene", "current_apm", b"current_apm", "current_effective_apm", b"current_effective_apm", "food_used", b"food_used", "friendly_fire_minerals", b"friendly_fire_minerals", "friendly_fire_vespene", b"friendly_fire_vespene", "idle_production_time", b"idle_production_time", "idle_worker_time", b"idle_worker_time", "killed_minerals", b"killed_minerals", "killed_value_structures", b"killed_value_structures", "killed_value_units", b"killed_value_units", "killed_vespene", b"killed_vespene", "lost_minerals", b"lost_minerals", "lost_vespene", b"lost_vespene", "spent_minerals", b"spent_minerals", "spent_vespene", b"spent_vespene", "total_damage_dealt", b"total_damage_dealt", "total_damage_taken", b"total_damage_taken", "total_healed", b"total_healed", "total_used_minerals", b"total_used_minerals", "total_used_vespene", b"total_used_vespene", "total_value_structures", b"total_value_structures", "total_value_units", b"total_value_units", "used_minerals", b"used_minerals", "used_vespene", b"used_vespene"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["collected_minerals", b"collected_minerals", "collected_vespene", b"collected_vespene", "collection_rate_minerals", b"collection_rate_minerals", "collection_rate_vespene", b"collection_rate_vespene", "current_apm", b"current_apm", "current_effective_apm", b"current_effective_apm", "food_used", b"food_used", "friendly_fire_minerals", b"friendly_fire_minerals", "friendly_fire_vespene", b"friendly_fire_vespene", "idle_production_time", b"idle_production_time", "idle_worker_time", b"idle_worker_time", "killed_minerals", b"killed_minerals", "killed_value_structures", b"killed_value_structures", "killed_value_units", b"killed_value_units", "killed_vespene", b"killed_vespene", "lost_minerals", b"lost_minerals", "lost_vespene", b"lost_vespene", "spent_minerals", b"spent_minerals", "spent_vespene", b"spent_vespene", "total_damage_dealt", b"total_damage_dealt", "total_damage_taken", b"total_damage_taken", "total_healed", b"total_healed", "total_used_minerals", b"total_used_minerals", "total_used_vespene", b"total_used_vespene", "total_value_structures", b"total_value_structures", "total_value_units", b"total_value_units", "used_minerals", b"used_minerals", "used_vespene", b"used_vespene"]) -> None: ... + def HasField( + self, + field_name: typing.Literal[ + "collected_minerals", + b"collected_minerals", + "collected_vespene", + b"collected_vespene", + "collection_rate_minerals", + b"collection_rate_minerals", + "collection_rate_vespene", + b"collection_rate_vespene", + "current_apm", + b"current_apm", + "current_effective_apm", + b"current_effective_apm", + "food_used", + b"food_used", + "friendly_fire_minerals", + b"friendly_fire_minerals", + "friendly_fire_vespene", + b"friendly_fire_vespene", + "idle_production_time", + b"idle_production_time", + "idle_worker_time", + b"idle_worker_time", + "killed_minerals", + b"killed_minerals", + "killed_value_structures", + b"killed_value_structures", + "killed_value_units", + b"killed_value_units", + "killed_vespene", + b"killed_vespene", + "lost_minerals", + b"lost_minerals", + "lost_vespene", + b"lost_vespene", + "spent_minerals", + b"spent_minerals", + "spent_vespene", + b"spent_vespene", + "total_damage_dealt", + b"total_damage_dealt", + "total_damage_taken", + b"total_damage_taken", + "total_healed", + b"total_healed", + "total_used_minerals", + b"total_used_minerals", + "total_used_vespene", + b"total_used_vespene", + "total_value_structures", + b"total_value_structures", + "total_value_units", + b"total_value_units", + "used_minerals", + b"used_minerals", + "used_vespene", + b"used_vespene", + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "collected_minerals", + b"collected_minerals", + "collected_vespene", + b"collected_vespene", + "collection_rate_minerals", + b"collection_rate_minerals", + "collection_rate_vespene", + b"collection_rate_vespene", + "current_apm", + b"current_apm", + "current_effective_apm", + b"current_effective_apm", + "food_used", + b"food_used", + "friendly_fire_minerals", + b"friendly_fire_minerals", + "friendly_fire_vespene", + b"friendly_fire_vespene", + "idle_production_time", + b"idle_production_time", + "idle_worker_time", + b"idle_worker_time", + "killed_minerals", + b"killed_minerals", + "killed_value_structures", + b"killed_value_structures", + "killed_value_units", + b"killed_value_units", + "killed_vespene", + b"killed_vespene", + "lost_minerals", + b"lost_minerals", + "lost_vespene", + b"lost_vespene", + "spent_minerals", + b"spent_minerals", + "spent_vespene", + b"spent_vespene", + "total_damage_dealt", + b"total_damage_dealt", + "total_damage_taken", + b"total_damage_taken", + "total_healed", + b"total_healed", + "total_used_minerals", + b"total_used_minerals", + "total_used_vespene", + b"total_used_vespene", + "total_value_structures", + b"total_value_structures", + "total_value_units", + b"total_value_units", + "used_minerals", + b"used_minerals", + "used_vespene", + b"used_vespene", + ], + ) -> None: ... global___ScoreDetails = ScoreDetails diff --git a/stubs/s2clientprotocol/s2clientprotocol/spatial_pb2.pyi b/stubs/s2clientprotocol/s2clientprotocol/spatial_pb2.pyi index f2e686f08242..5fd2301ac84b 100644 --- a/stubs/s2clientprotocol/s2clientprotocol/spatial_pb2.pyi +++ b/stubs/s2clientprotocol/s2clientprotocol/spatial_pb2.pyi @@ -36,12 +36,11 @@ class ObservationFeatureLayer(google.protobuf.message.Message): @property def minimap_renders(self) -> global___FeatureLayersMinimap: ... def __init__( - self, - *, - renders: global___FeatureLayers | None = ..., - minimap_renders: global___FeatureLayersMinimap | None = ..., + self, *, renders: global___FeatureLayers | None = ..., minimap_renders: global___FeatureLayersMinimap | None = ... ) -> None: ... - def HasField(self, field_name: typing.Literal["minimap_renders", b"minimap_renders", "renders", b"renders"]) -> builtins.bool: ... + def HasField( + self, field_name: typing.Literal["minimap_renders", b"minimap_renders", "renders", b"renders"] + ) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["minimap_renders", b"minimap_renders", "renders", b"renders"]) -> None: ... global___ObservationFeatureLayer = ObservationFeatureLayer @@ -216,8 +215,124 @@ class FeatureLayers(google.protobuf.message.Message): pathable: s2clientprotocol.common_pb2.ImageData | None = ..., placeholder: s2clientprotocol.common_pb2.ImageData | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["active", b"active", "blip", b"blip", "buff_duration", b"buff_duration", "buffs", b"buffs", "build_progress", b"build_progress", "buildable", b"buildable", "cloaked", b"cloaked", "creep", b"creep", "effects", b"effects", "hallucinations", b"hallucinations", "height_map", b"height_map", "pathable", b"pathable", "placeholder", b"placeholder", "player_id", b"player_id", "player_relative", b"player_relative", "power", b"power", "selected", b"selected", "unit_density", b"unit_density", "unit_density_aa", b"unit_density_aa", "unit_energy", b"unit_energy", "unit_energy_ratio", b"unit_energy_ratio", "unit_hit_points", b"unit_hit_points", "unit_hit_points_ratio", b"unit_hit_points_ratio", "unit_shields", b"unit_shields", "unit_shields_ratio", b"unit_shields_ratio", "unit_type", b"unit_type", "visibility_map", b"visibility_map"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["active", b"active", "blip", b"blip", "buff_duration", b"buff_duration", "buffs", b"buffs", "build_progress", b"build_progress", "buildable", b"buildable", "cloaked", b"cloaked", "creep", b"creep", "effects", b"effects", "hallucinations", b"hallucinations", "height_map", b"height_map", "pathable", b"pathable", "placeholder", b"placeholder", "player_id", b"player_id", "player_relative", b"player_relative", "power", b"power", "selected", b"selected", "unit_density", b"unit_density", "unit_density_aa", b"unit_density_aa", "unit_energy", b"unit_energy", "unit_energy_ratio", b"unit_energy_ratio", "unit_hit_points", b"unit_hit_points", "unit_hit_points_ratio", b"unit_hit_points_ratio", "unit_shields", b"unit_shields", "unit_shields_ratio", b"unit_shields_ratio", "unit_type", b"unit_type", "visibility_map", b"visibility_map"]) -> None: ... + def HasField( + self, + field_name: typing.Literal[ + "active", + b"active", + "blip", + b"blip", + "buff_duration", + b"buff_duration", + "buffs", + b"buffs", + "build_progress", + b"build_progress", + "buildable", + b"buildable", + "cloaked", + b"cloaked", + "creep", + b"creep", + "effects", + b"effects", + "hallucinations", + b"hallucinations", + "height_map", + b"height_map", + "pathable", + b"pathable", + "placeholder", + b"placeholder", + "player_id", + b"player_id", + "player_relative", + b"player_relative", + "power", + b"power", + "selected", + b"selected", + "unit_density", + b"unit_density", + "unit_density_aa", + b"unit_density_aa", + "unit_energy", + b"unit_energy", + "unit_energy_ratio", + b"unit_energy_ratio", + "unit_hit_points", + b"unit_hit_points", + "unit_hit_points_ratio", + b"unit_hit_points_ratio", + "unit_shields", + b"unit_shields", + "unit_shields_ratio", + b"unit_shields_ratio", + "unit_type", + b"unit_type", + "visibility_map", + b"visibility_map", + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "active", + b"active", + "blip", + b"blip", + "buff_duration", + b"buff_duration", + "buffs", + b"buffs", + "build_progress", + b"build_progress", + "buildable", + b"buildable", + "cloaked", + b"cloaked", + "creep", + b"creep", + "effects", + b"effects", + "hallucinations", + b"hallucinations", + "height_map", + b"height_map", + "pathable", + b"pathable", + "placeholder", + b"placeholder", + "player_id", + b"player_id", + "player_relative", + b"player_relative", + "power", + b"power", + "selected", + b"selected", + "unit_density", + b"unit_density", + "unit_density_aa", + b"unit_density_aa", + "unit_energy", + b"unit_energy", + "unit_energy_ratio", + b"unit_energy_ratio", + "unit_hit_points", + b"unit_hit_points", + "unit_hit_points_ratio", + b"unit_hit_points_ratio", + "unit_shields", + b"unit_shields", + "unit_shields_ratio", + b"unit_shields_ratio", + "unit_type", + b"unit_type", + "visibility_map", + b"visibility_map", + ], + ) -> None: ... global___FeatureLayers = FeatureLayers @@ -297,8 +412,60 @@ class FeatureLayersMinimap(google.protobuf.message.Message): pathable: s2clientprotocol.common_pb2.ImageData | None = ..., unit_type: s2clientprotocol.common_pb2.ImageData | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["alerts", b"alerts", "buildable", b"buildable", "camera", b"camera", "creep", b"creep", "height_map", b"height_map", "pathable", b"pathable", "player_id", b"player_id", "player_relative", b"player_relative", "selected", b"selected", "unit_type", b"unit_type", "visibility_map", b"visibility_map"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["alerts", b"alerts", "buildable", b"buildable", "camera", b"camera", "creep", b"creep", "height_map", b"height_map", "pathable", b"pathable", "player_id", b"player_id", "player_relative", b"player_relative", "selected", b"selected", "unit_type", b"unit_type", "visibility_map", b"visibility_map"]) -> None: ... + def HasField( + self, + field_name: typing.Literal[ + "alerts", + b"alerts", + "buildable", + b"buildable", + "camera", + b"camera", + "creep", + b"creep", + "height_map", + b"height_map", + "pathable", + b"pathable", + "player_id", + b"player_id", + "player_relative", + b"player_relative", + "selected", + b"selected", + "unit_type", + b"unit_type", + "visibility_map", + b"visibility_map", + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "alerts", + b"alerts", + "buildable", + b"buildable", + "camera", + b"camera", + "creep", + b"creep", + "height_map", + b"height_map", + "pathable", + b"pathable", + "player_id", + b"player_id", + "player_relative", + b"player_relative", + "selected", + b"selected", + "unit_type", + b"unit_type", + "visibility_map", + b"visibility_map", + ], + ) -> None: ... global___FeatureLayersMinimap = FeatureLayersMinimap @@ -355,9 +522,39 @@ class ActionSpatial(google.protobuf.message.Message): unit_selection_point: global___ActionSpatialUnitSelectionPoint | None = ..., unit_selection_rect: global___ActionSpatialUnitSelectionRect | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["action", b"action", "camera_move", b"camera_move", "unit_command", b"unit_command", "unit_selection_point", b"unit_selection_point", "unit_selection_rect", b"unit_selection_rect"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["action", b"action", "camera_move", b"camera_move", "unit_command", b"unit_command", "unit_selection_point", b"unit_selection_point", "unit_selection_rect", b"unit_selection_rect"]) -> None: ... - def WhichOneof(self, oneof_group: typing.Literal["action", b"action"]) -> typing.Literal["unit_command", "camera_move", "unit_selection_point", "unit_selection_rect"] | None: ... + def HasField( + self, + field_name: typing.Literal[ + "action", + b"action", + "camera_move", + b"camera_move", + "unit_command", + b"unit_command", + "unit_selection_point", + b"unit_selection_point", + "unit_selection_rect", + b"unit_selection_rect", + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "action", + b"action", + "camera_move", + b"camera_move", + "unit_command", + b"unit_command", + "unit_selection_point", + b"unit_selection_point", + "unit_selection_rect", + b"unit_selection_rect", + ], + ) -> None: ... + def WhichOneof( + self, oneof_group: typing.Literal["action", b"action"] + ) -> typing.Literal["unit_command", "camera_move", "unit_selection_point", "unit_selection_rect"] | None: ... global___ActionSpatial = ActionSpatial @@ -384,9 +581,39 @@ class ActionSpatialUnitCommand(google.protobuf.message.Message): target_minimap_coord: s2clientprotocol.common_pb2.PointI | None = ..., queue_command: builtins.bool | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["ability_id", b"ability_id", "queue_command", b"queue_command", "target", b"target", "target_minimap_coord", b"target_minimap_coord", "target_screen_coord", b"target_screen_coord"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["ability_id", b"ability_id", "queue_command", b"queue_command", "target", b"target", "target_minimap_coord", b"target_minimap_coord", "target_screen_coord", b"target_screen_coord"]) -> None: ... - def WhichOneof(self, oneof_group: typing.Literal["target", b"target"]) -> typing.Literal["target_screen_coord", "target_minimap_coord"] | None: ... + def HasField( + self, + field_name: typing.Literal[ + "ability_id", + b"ability_id", + "queue_command", + b"queue_command", + "target", + b"target", + "target_minimap_coord", + b"target_minimap_coord", + "target_screen_coord", + b"target_screen_coord", + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "ability_id", + b"ability_id", + "queue_command", + b"queue_command", + "target", + b"target", + "target_minimap_coord", + b"target_minimap_coord", + "target_screen_coord", + b"target_screen_coord", + ], + ) -> None: ... + def WhichOneof( + self, oneof_group: typing.Literal["target", b"target"] + ) -> typing.Literal["target_screen_coord", "target_minimap_coord"] | None: ... global___ActionSpatialUnitCommand = ActionSpatialUnitCommand @@ -399,11 +626,7 @@ class ActionSpatialCameraMove(google.protobuf.message.Message): def center_minimap(self) -> s2clientprotocol.common_pb2.PointI: """Simulates a click on the minimap to move the camera.""" - def __init__( - self, - *, - center_minimap: s2clientprotocol.common_pb2.PointI | None = ..., - ) -> None: ... + def __init__(self, *, center_minimap: s2clientprotocol.common_pb2.PointI | None = ...) -> None: ... def HasField(self, field_name: typing.Literal["center_minimap", b"center_minimap"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["center_minimap", b"center_minimap"]) -> None: ... @@ -417,7 +640,10 @@ class ActionSpatialUnitSelectionPoint(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _TypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[ActionSpatialUnitSelectionPoint._Type.ValueType], builtins.type): + class _TypeEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[ActionSpatialUnitSelectionPoint._Type.ValueType], + builtins.type, + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor Select: ActionSpatialUnitSelectionPoint._Type.ValueType # 1 """Equivalent to normal click. Changes selection to unit.""" @@ -449,8 +675,12 @@ class ActionSpatialUnitSelectionPoint(google.protobuf.message.Message): selection_screen_coord: s2clientprotocol.common_pb2.PointI | None = ..., type: global___ActionSpatialUnitSelectionPoint.Type.ValueType | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["selection_screen_coord", b"selection_screen_coord", "type", b"type"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["selection_screen_coord", b"selection_screen_coord", "type", b"type"]) -> None: ... + def HasField( + self, field_name: typing.Literal["selection_screen_coord", b"selection_screen_coord", "type", b"type"] + ) -> builtins.bool: ... + def ClearField( + self, field_name: typing.Literal["selection_screen_coord", b"selection_screen_coord", "type", b"type"] + ) -> None: ... global___ActionSpatialUnitSelectionPoint = ActionSpatialUnitSelectionPoint @@ -463,7 +693,9 @@ class ActionSpatialUnitSelectionRect(google.protobuf.message.Message): selection_add: builtins.bool """Equivalent to shift+drag. Adds units to selection.""" @property - def selection_screen_coord(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[s2clientprotocol.common_pb2.RectangleI]: + def selection_screen_coord( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[s2clientprotocol.common_pb2.RectangleI]: """Eventually this should not be an array, but a single field (multiple would be cheating).""" def __init__( @@ -473,6 +705,8 @@ class ActionSpatialUnitSelectionRect(google.protobuf.message.Message): selection_add: builtins.bool | None = ..., ) -> None: ... def HasField(self, field_name: typing.Literal["selection_add", b"selection_add"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["selection_add", b"selection_add", "selection_screen_coord", b"selection_screen_coord"]) -> None: ... + def ClearField( + self, field_name: typing.Literal["selection_add", b"selection_add", "selection_screen_coord", b"selection_screen_coord"] + ) -> None: ... global___ActionSpatialUnitSelectionRect = ActionSpatialUnitSelectionRect diff --git a/stubs/s2clientprotocol/s2clientprotocol/ui_pb2.pyi b/stubs/s2clientprotocol/s2clientprotocol/ui_pb2.pyi index 65a0e165a43d..2608a8aa142c 100644 --- a/stubs/s2clientprotocol/s2clientprotocol/ui_pb2.pyi +++ b/stubs/s2clientprotocol/s2clientprotocol/ui_pb2.pyi @@ -52,9 +52,32 @@ class ObservationUI(google.protobuf.message.Message): cargo: global___CargoPanel | None = ..., production: global___ProductionPanel | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["cargo", b"cargo", "multi", b"multi", "panel", b"panel", "production", b"production", "single", b"single"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["cargo", b"cargo", "groups", b"groups", "multi", b"multi", "panel", b"panel", "production", b"production", "single", b"single"]) -> None: ... - def WhichOneof(self, oneof_group: typing.Literal["panel", b"panel"]) -> typing.Literal["single", "multi", "cargo", "production"] | None: ... + def HasField( + self, + field_name: typing.Literal[ + "cargo", b"cargo", "multi", b"multi", "panel", b"panel", "production", b"production", "single", b"single" + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "cargo", + b"cargo", + "groups", + b"groups", + "multi", + b"multi", + "panel", + b"panel", + "production", + b"production", + "single", + b"single", + ], + ) -> None: ... + def WhichOneof( + self, oneof_group: typing.Literal["panel", b"panel"] + ) -> typing.Literal["single", "multi", "cargo", "production"] | None: ... global___ObservationUI = ObservationUI @@ -75,8 +98,18 @@ class ControlGroup(google.protobuf.message.Message): leader_unit_type: builtins.int | None = ..., count: builtins.int | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["control_group_index", b"control_group_index", "count", b"count", "leader_unit_type", b"leader_unit_type"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["control_group_index", b"control_group_index", "count", b"count", "leader_unit_type", b"leader_unit_type"]) -> None: ... + def HasField( + self, + field_name: typing.Literal[ + "control_group_index", b"control_group_index", "count", b"count", "leader_unit_type", b"leader_unit_type" + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "control_group_index", b"control_group_index", "count", b"count", "leader_unit_type", b"leader_unit_type" + ], + ) -> None: ... global___ControlGroup = ControlGroup @@ -123,8 +156,60 @@ class UnitInfo(google.protobuf.message.Message): max_shields: builtins.int | None = ..., max_energy: builtins.int | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["add_on", b"add_on", "build_progress", b"build_progress", "energy", b"energy", "health", b"health", "max_energy", b"max_energy", "max_health", b"max_health", "max_shields", b"max_shields", "player_relative", b"player_relative", "shields", b"shields", "transport_slots_taken", b"transport_slots_taken", "unit_type", b"unit_type"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["add_on", b"add_on", "build_progress", b"build_progress", "energy", b"energy", "health", b"health", "max_energy", b"max_energy", "max_health", b"max_health", "max_shields", b"max_shields", "player_relative", b"player_relative", "shields", b"shields", "transport_slots_taken", b"transport_slots_taken", "unit_type", b"unit_type"]) -> None: ... + def HasField( + self, + field_name: typing.Literal[ + "add_on", + b"add_on", + "build_progress", + b"build_progress", + "energy", + b"energy", + "health", + b"health", + "max_energy", + b"max_energy", + "max_health", + b"max_health", + "max_shields", + b"max_shields", + "player_relative", + b"player_relative", + "shields", + b"shields", + "transport_slots_taken", + b"transport_slots_taken", + "unit_type", + b"unit_type", + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "add_on", + b"add_on", + "build_progress", + b"build_progress", + "energy", + b"energy", + "health", + b"health", + "max_energy", + b"max_energy", + "max_health", + b"max_health", + "max_shields", + b"max_shields", + "player_relative", + b"player_relative", + "shields", + b"shields", + "transport_slots_taken", + b"transport_slots_taken", + "unit_type", + b"unit_type", + ], + ) -> None: ... global___UnitInfo = UnitInfo @@ -153,8 +238,34 @@ class SinglePanel(google.protobuf.message.Message): shield_upgrade_level: builtins.int | None = ..., buffs: collections.abc.Iterable[builtins.int] | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["armor_upgrade_level", b"armor_upgrade_level", "attack_upgrade_level", b"attack_upgrade_level", "shield_upgrade_level", b"shield_upgrade_level", "unit", b"unit"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["armor_upgrade_level", b"armor_upgrade_level", "attack_upgrade_level", b"attack_upgrade_level", "buffs", b"buffs", "shield_upgrade_level", b"shield_upgrade_level", "unit", b"unit"]) -> None: ... + def HasField( + self, + field_name: typing.Literal[ + "armor_upgrade_level", + b"armor_upgrade_level", + "attack_upgrade_level", + b"attack_upgrade_level", + "shield_upgrade_level", + b"shield_upgrade_level", + "unit", + b"unit", + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "armor_upgrade_level", + b"armor_upgrade_level", + "attack_upgrade_level", + b"attack_upgrade_level", + "buffs", + b"buffs", + "shield_upgrade_level", + b"shield_upgrade_level", + "unit", + b"unit", + ], + ) -> None: ... global___SinglePanel = SinglePanel @@ -165,11 +276,7 @@ class MultiPanel(google.protobuf.message.Message): UNITS_FIELD_NUMBER: builtins.int @property def units(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___UnitInfo]: ... - def __init__( - self, - *, - units: collections.abc.Iterable[global___UnitInfo] | None = ..., - ) -> None: ... + def __init__(self, *, units: collections.abc.Iterable[global___UnitInfo] | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["units", b"units"]) -> None: ... global___MultiPanel = MultiPanel @@ -195,7 +302,9 @@ class CargoPanel(google.protobuf.message.Message): slots_available: builtins.int | None = ..., ) -> None: ... def HasField(self, field_name: typing.Literal["slots_available", b"slots_available", "unit", b"unit"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["passengers", b"passengers", "slots_available", b"slots_available", "unit", b"unit"]) -> None: ... + def ClearField( + self, field_name: typing.Literal["passengers", b"passengers", "slots_available", b"slots_available", "unit", b"unit"] + ) -> None: ... global___CargoPanel = CargoPanel @@ -208,14 +317,13 @@ class BuildItem(google.protobuf.message.Message): ability_id: builtins.int build_progress: builtins.float """Range: [0.0, 1.0]""" - def __init__( - self, - *, - ability_id: builtins.int | None = ..., - build_progress: builtins.float | None = ..., + def __init__(self, *, ability_id: builtins.int | None = ..., build_progress: builtins.float | None = ...) -> None: ... + def HasField( + self, field_name: typing.Literal["ability_id", b"ability_id", "build_progress", b"build_progress"] + ) -> builtins.bool: ... + def ClearField( + self, field_name: typing.Literal["ability_id", b"ability_id", "build_progress", b"build_progress"] ) -> None: ... - def HasField(self, field_name: typing.Literal["ability_id", b"ability_id", "build_progress", b"build_progress"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["ability_id", b"ability_id", "build_progress", b"build_progress"]) -> None: ... global___BuildItem = BuildItem @@ -244,7 +352,9 @@ class ProductionPanel(google.protobuf.message.Message): production_queue: collections.abc.Iterable[global___BuildItem] | None = ..., ) -> None: ... def HasField(self, field_name: typing.Literal["unit", b"unit"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["build_queue", b"build_queue", "production_queue", b"production_queue", "unit", b"unit"]) -> None: ... + def ClearField( + self, field_name: typing.Literal["build_queue", b"build_queue", "production_queue", b"production_queue", "unit", b"unit"] + ) -> None: ... global___ProductionPanel = ProductionPanel @@ -296,9 +406,72 @@ class ActionUI(google.protobuf.message.Message): production_panel: global___ActionProductionPanelRemoveFromQueue | None = ..., toggle_autocast: global___ActionToggleAutocast | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["action", b"action", "cargo_panel", b"cargo_panel", "control_group", b"control_group", "multi_panel", b"multi_panel", "production_panel", b"production_panel", "select_army", b"select_army", "select_idle_worker", b"select_idle_worker", "select_larva", b"select_larva", "select_warp_gates", b"select_warp_gates", "toggle_autocast", b"toggle_autocast"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["action", b"action", "cargo_panel", b"cargo_panel", "control_group", b"control_group", "multi_panel", b"multi_panel", "production_panel", b"production_panel", "select_army", b"select_army", "select_idle_worker", b"select_idle_worker", "select_larva", b"select_larva", "select_warp_gates", b"select_warp_gates", "toggle_autocast", b"toggle_autocast"]) -> None: ... - def WhichOneof(self, oneof_group: typing.Literal["action", b"action"]) -> typing.Literal["control_group", "select_army", "select_warp_gates", "select_larva", "select_idle_worker", "multi_panel", "cargo_panel", "production_panel", "toggle_autocast"] | None: ... + def HasField( + self, + field_name: typing.Literal[ + "action", + b"action", + "cargo_panel", + b"cargo_panel", + "control_group", + b"control_group", + "multi_panel", + b"multi_panel", + "production_panel", + b"production_panel", + "select_army", + b"select_army", + "select_idle_worker", + b"select_idle_worker", + "select_larva", + b"select_larva", + "select_warp_gates", + b"select_warp_gates", + "toggle_autocast", + b"toggle_autocast", + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "action", + b"action", + "cargo_panel", + b"cargo_panel", + "control_group", + b"control_group", + "multi_panel", + b"multi_panel", + "production_panel", + b"production_panel", + "select_army", + b"select_army", + "select_idle_worker", + b"select_idle_worker", + "select_larva", + b"select_larva", + "select_warp_gates", + b"select_warp_gates", + "toggle_autocast", + b"toggle_autocast", + ], + ) -> None: ... + def WhichOneof( + self, oneof_group: typing.Literal["action", b"action"] + ) -> ( + typing.Literal[ + "control_group", + "select_army", + "select_warp_gates", + "select_larva", + "select_idle_worker", + "multi_panel", + "cargo_panel", + "production_panel", + "toggle_autocast", + ] + | None + ): ... global___ActionUI = ActionUI @@ -310,7 +483,10 @@ class ActionControlGroup(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _ControlGroupActionEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[ActionControlGroup._ControlGroupAction.ValueType], builtins.type): + class _ControlGroupActionEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[ActionControlGroup._ControlGroupAction.ValueType], + builtins.type, + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor Recall: ActionControlGroup._ControlGroupAction.ValueType # 1 """Equivalent to number hotkey. Replaces current selection with control group.""" @@ -345,8 +521,12 @@ class ActionControlGroup(google.protobuf.message.Message): action: global___ActionControlGroup.ControlGroupAction.ValueType | None = ..., control_group_index: builtins.int | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["action", b"action", "control_group_index", b"control_group_index"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["action", b"action", "control_group_index", b"control_group_index"]) -> None: ... + def HasField( + self, field_name: typing.Literal["action", b"action", "control_group_index", b"control_group_index"] + ) -> builtins.bool: ... + def ClearField( + self, field_name: typing.Literal["action", b"action", "control_group_index", b"control_group_index"] + ) -> None: ... global___ActionControlGroup = ActionControlGroup @@ -356,11 +536,7 @@ class ActionSelectArmy(google.protobuf.message.Message): SELECTION_ADD_FIELD_NUMBER: builtins.int selection_add: builtins.bool - def __init__( - self, - *, - selection_add: builtins.bool | None = ..., - ) -> None: ... + def __init__(self, *, selection_add: builtins.bool | None = ...) -> None: ... def HasField(self, field_name: typing.Literal["selection_add", b"selection_add"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["selection_add", b"selection_add"]) -> None: ... @@ -372,11 +548,7 @@ class ActionSelectWarpGates(google.protobuf.message.Message): SELECTION_ADD_FIELD_NUMBER: builtins.int selection_add: builtins.bool - def __init__( - self, - *, - selection_add: builtins.bool | None = ..., - ) -> None: ... + def __init__(self, *, selection_add: builtins.bool | None = ...) -> None: ... def HasField(self, field_name: typing.Literal["selection_add", b"selection_add"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["selection_add", b"selection_add"]) -> None: ... @@ -386,9 +558,7 @@ global___ActionSelectWarpGates = ActionSelectWarpGates class ActionSelectLarva(google.protobuf.message.Message): DESCRIPTOR: google.protobuf.descriptor.Descriptor - def __init__( - self, - ) -> None: ... + def __init__(self) -> None: ... global___ActionSelectLarva = ActionSelectLarva @@ -400,7 +570,9 @@ class ActionSelectIdleWorker(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _TypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[ActionSelectIdleWorker._Type.ValueType], builtins.type): + class _TypeEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[ActionSelectIdleWorker._Type.ValueType], builtins.type + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor Set: ActionSelectIdleWorker._Type.ValueType # 1 """Equivalent to click with no modifiers. Replaces selection with single idle worker.""" @@ -423,11 +595,7 @@ class ActionSelectIdleWorker(google.protobuf.message.Message): TYPE_FIELD_NUMBER: builtins.int type: global___ActionSelectIdleWorker.Type.ValueType - def __init__( - self, - *, - type: global___ActionSelectIdleWorker.Type.ValueType | None = ..., - ) -> None: ... + def __init__(self, *, type: global___ActionSelectIdleWorker.Type.ValueType | None = ...) -> None: ... def HasField(self, field_name: typing.Literal["type", b"type"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["type", b"type"]) -> None: ... @@ -441,7 +609,9 @@ class ActionMultiPanel(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _TypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[ActionMultiPanel._Type.ValueType], builtins.type): + class _TypeEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[ActionMultiPanel._Type.ValueType], builtins.type + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor SingleSelect: ActionMultiPanel._Type.ValueType # 1 """Click on icon""" @@ -467,10 +637,7 @@ class ActionMultiPanel(google.protobuf.message.Message): type: global___ActionMultiPanel.Type.ValueType unit_index: builtins.int def __init__( - self, - *, - type: global___ActionMultiPanel.Type.ValueType | None = ..., - unit_index: builtins.int | None = ..., + self, *, type: global___ActionMultiPanel.Type.ValueType | None = ..., unit_index: builtins.int | None = ... ) -> None: ... def HasField(self, field_name: typing.Literal["type", b"type", "unit_index", b"unit_index"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["type", b"type", "unit_index", b"unit_index"]) -> None: ... @@ -483,11 +650,7 @@ class ActionCargoPanelUnload(google.protobuf.message.Message): UNIT_INDEX_FIELD_NUMBER: builtins.int unit_index: builtins.int - def __init__( - self, - *, - unit_index: builtins.int | None = ..., - ) -> None: ... + def __init__(self, *, unit_index: builtins.int | None = ...) -> None: ... def HasField(self, field_name: typing.Literal["unit_index", b"unit_index"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["unit_index", b"unit_index"]) -> None: ... @@ -499,11 +662,7 @@ class ActionProductionPanelRemoveFromQueue(google.protobuf.message.Message): UNIT_INDEX_FIELD_NUMBER: builtins.int unit_index: builtins.int - def __init__( - self, - *, - unit_index: builtins.int | None = ..., - ) -> None: ... + def __init__(self, *, unit_index: builtins.int | None = ...) -> None: ... def HasField(self, field_name: typing.Literal["unit_index", b"unit_index"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["unit_index", b"unit_index"]) -> None: ... @@ -515,11 +674,7 @@ class ActionToggleAutocast(google.protobuf.message.Message): ABILITY_ID_FIELD_NUMBER: builtins.int ability_id: builtins.int - def __init__( - self, - *, - ability_id: builtins.int | None = ..., - ) -> None: ... + def __init__(self, *, ability_id: builtins.int | None = ...) -> None: ... def HasField(self, field_name: typing.Literal["ability_id", b"ability_id"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["ability_id", b"ability_id"]) -> None: ... diff --git a/stubs/tensorflow/tensorflow/compiler/xla/service/hlo_pb2.pyi b/stubs/tensorflow/tensorflow/compiler/xla/service/hlo_pb2.pyi index 25dd0678d051..9057959cc005 100644 --- a/stubs/tensorflow/tensorflow/compiler/xla/service/hlo_pb2.pyi +++ b/stubs/tensorflow/tensorflow/compiler/xla/service/hlo_pb2.pyi @@ -38,7 +38,9 @@ class _CustomCallSchedule: ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType -class _CustomCallScheduleEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_CustomCallSchedule.ValueType], builtins.type): +class _CustomCallScheduleEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_CustomCallSchedule.ValueType], builtins.type +): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor SCHEDULE_NONE: _CustomCallSchedule.ValueType # 0 SCHEDULE_LATEST: _CustomCallSchedule.ValueType # 1 @@ -55,7 +57,9 @@ class _CustomCallApiVersion: ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType -class _CustomCallApiVersionEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_CustomCallApiVersion.ValueType], builtins.type): +class _CustomCallApiVersionEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_CustomCallApiVersion.ValueType], builtins.type +): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor API_VERSION_UNSPECIFIED: _CustomCallApiVersion.ValueType # 0 API_VERSION_ORIGINAL: _CustomCallApiVersion.ValueType # 1 @@ -237,11 +241,7 @@ class HloInstructionProto(google.protobuf.message.Message): limit: builtins.int stride: builtins.int def __init__( - self, - *, - start: builtins.int | None = ..., - limit: builtins.int | None = ..., - stride: builtins.int | None = ..., + self, *, start: builtins.int | None = ..., limit: builtins.int | None = ..., stride: builtins.int | None = ... ) -> None: ... def ClearField(self, field_name: typing.Literal["limit", b"limit", "start", b"start", "stride", b"stride"]) -> None: ... @@ -461,7 +461,9 @@ class HloInstructionProto(google.protobuf.message.Message): """Describes the dimension numbers used for a convolution.""" @property - def slice_dimensions(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___HloInstructionProto.SliceDimensions]: ... + def slice_dimensions( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___HloInstructionProto.SliceDimensions]: ... @property def dynamic_slice_sizes(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.int]: """Describes the [start, start + size) range size for a dynamic slice @@ -501,7 +503,9 @@ class HloInstructionProto(google.protobuf.message.Message): @property def sharding(self) -> tensorflow.compiler.xla.xla_data_pb2.OpSharding: ... @property - def replica_groups(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[tensorflow.compiler.xla.xla_data_pb2.ReplicaGroup]: + def replica_groups( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[tensorflow.compiler.xla.xla_data_pb2.ReplicaGroup]: """Deprecated, but keeping for backward compatibility. Use collective_device_list. Cross replica op fields. """ @@ -513,7 +517,9 @@ class HloInstructionProto(google.protobuf.message.Message): """Precision configuration for the instruction. Has backend-specific meaning.""" @property - def source_target_pairs(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[tensorflow.compiler.xla.xla_data_pb2.SourceTarget]: + def source_target_pairs( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[tensorflow.compiler.xla.xla_data_pb2.SourceTarget]: """Collective permute field.""" @property @@ -523,7 +529,9 @@ class HloInstructionProto(google.protobuf.message.Message): @property def domain_exit_sharding(self) -> tensorflow.compiler.xla.xla_data_pb2.OpSharding: ... @property - def operand_shapes_with_layout(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[tensorflow.compiler.xla.xla_data_pb2.ShapeProto]: ... + def operand_shapes_with_layout( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[tensorflow.compiler.xla.xla_data_pb2.ShapeProto]: ... @property def triangular_solve_options(self) -> tensorflow.compiler.xla.xla_data_pb2.TriangularSolveOptions: """Options for TriangularSolve""" @@ -537,7 +545,11 @@ class HloInstructionProto(google.protobuf.message.Message): """Describes how parameters behave with regards to replicas.""" @property - def output_operand_aliasing(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[tensorflow.compiler.xla.xla_data_pb2.OutputOperandAliasing]: + def output_operand_aliasing( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[ + tensorflow.compiler.xla.xla_data_pb2.OutputOperandAliasing + ]: """A list of OutputOperandAliasing pairs that specifies aliasing buffers between output and operands for kCustomCall and kFusion. """ @@ -553,7 +565,11 @@ class HloInstructionProto(google.protobuf.message.Message): """ @property - def dot_sparsity(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[tensorflow.compiler.xla.xla_data_pb2.SparsityDescriptor]: + def dot_sparsity( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[ + tensorflow.compiler.xla.xla_data_pb2.SparsityDescriptor + ]: """Sparsity descriptor for dot operation.""" @property @@ -621,7 +637,9 @@ class HloInstructionProto(google.protobuf.message.Message): cholesky_options: tensorflow.compiler.xla.xla_data_pb2.CholeskyOptions | None = ..., parameter_replication: tensorflow.compiler.xla.xla_data_pb2.ParameterReplication | None = ..., custom_call_has_side_effect: builtins.bool | None = ..., - output_operand_aliasing: collections.abc.Iterable[tensorflow.compiler.xla.xla_data_pb2.OutputOperandAliasing] | None = ..., + output_operand_aliasing: ( + collections.abc.Iterable[tensorflow.compiler.xla.xla_data_pb2.OutputOperandAliasing] | None + ) = ..., custom_call_schedule: global___CustomCallSchedule.ValueType | None = ..., delta: builtins.int | None = ..., indices_are_sorted: builtins.bool | None = ..., @@ -642,9 +660,215 @@ class HloInstructionProto(google.protobuf.message.Message): original_value: tensorflow.compiler.xla.xla_data_pb2.OriginalValueProto | None = ..., is_composite: builtins.bool | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["cholesky_options", b"cholesky_options", "collective_device_list", b"collective_device_list", "convolution_dimension_numbers", b"convolution_dimension_numbers", "cross_program_prefetch_index", b"cross_program_prefetch_index", "domain_entry_sharding", b"domain_entry_sharding", "domain_exit_sharding", b"domain_exit_sharding", "dot_dimension_numbers", b"dot_dimension_numbers", "frontend_attributes", b"frontend_attributes", "gather_dimension_numbers", b"gather_dimension_numbers", "literal", b"literal", "metadata", b"metadata", "optional_cross_program_prefetch_index", b"optional_cross_program_prefetch_index", "original_value", b"original_value", "outfeed_shape", b"outfeed_shape", "padding_config", b"padding_config", "parameter_replication", b"parameter_replication", "precision_config", b"precision_config", "scatter_dimension_numbers", b"scatter_dimension_numbers", "shape", b"shape", "sharding", b"sharding", "statistics_viz", b"statistics_viz", "triangular_solve_options", b"triangular_solve_options", "window", b"window"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["all_reduce_id", b"all_reduce_id", "async_execution_thread", b"async_execution_thread", "backend_config", b"backend_config", "batch_group_count", b"batch_group_count", "called_computation_ids", b"called_computation_ids", "channel_id", b"channel_id", "cholesky_options", b"cholesky_options", "collective_device_list", b"collective_device_list", "comparison_direction", b"comparison_direction", "comparison_type", b"comparison_type", "constrain_layout", b"constrain_layout", "control_predecessor_ids", b"control_predecessor_ids", "convolution_dimension_numbers", b"convolution_dimension_numbers", "cross_program_prefetch_index", b"cross_program_prefetch_index", "custom_call_api_version", b"custom_call_api_version", "custom_call_has_side_effect", b"custom_call_has_side_effect", "custom_call_schedule", b"custom_call_schedule", "custom_call_target", b"custom_call_target", "delta", b"delta", "dimensions", b"dimensions", "distribution", b"distribution", "domain_entry_sharding", b"domain_entry_sharding", "domain_exit_sharding", b"domain_exit_sharding", "dot_dimension_numbers", b"dot_dimension_numbers", "dot_sparsity", b"dot_sparsity", "dynamic_slice_sizes", b"dynamic_slice_sizes", "epsilon", b"epsilon", "exponent_bits", b"exponent_bits", "feature_group_count", b"feature_group_count", "feature_index", b"feature_index", "fft_length", b"fft_length", "fft_type", b"fft_type", "frontend_attributes", b"frontend_attributes", "fusion_kind", b"fusion_kind", "gather_dimension_numbers", b"gather_dimension_numbers", "gather_slice_sizes", b"gather_slice_sizes", "id", b"id", "indices_are_sorted", b"indices_are_sorted", "infeed_config", b"infeed_config", "is_composite", b"is_composite", "is_cross_program_prefetch", b"is_cross_program_prefetch", "is_host_transfer", b"is_host_transfer", "is_stable", b"is_stable", "k", b"k", "largest", b"largest", "literal", b"literal", "mantissa_bits", b"mantissa_bits", "metadata", b"metadata", "name", b"name", "opcode", b"opcode", "operand_ids", b"operand_ids", "operand_shapes_with_layout", b"operand_shapes_with_layout", "optional_cross_program_prefetch_index", b"optional_cross_program_prefetch_index", "original_value", b"original_value", "outfeed_config", b"outfeed_config", "outfeed_shape", b"outfeed_shape", "output_operand_aliasing", b"output_operand_aliasing", "padding_config", b"padding_config", "padding_type", b"padding_type", "parameter_number", b"parameter_number", "parameter_replication", b"parameter_replication", "precision_config", b"precision_config", "replica_groups", b"replica_groups", "rng_algorithm", b"rng_algorithm", "scatter_dimension_numbers", b"scatter_dimension_numbers", "shape", b"shape", "sharding", b"sharding", "slice_dimensions", b"slice_dimensions", "source_target_pairs", b"source_target_pairs", "statistics_viz", b"statistics_viz", "triangular_solve_options", b"triangular_solve_options", "tuple_index", b"tuple_index", "unique_indices", b"unique_indices", "use_global_device_ids", b"use_global_device_ids", "window", b"window"]) -> None: ... - def WhichOneof(self, oneof_group: typing.Literal["optional_cross_program_prefetch_index", b"optional_cross_program_prefetch_index"]) -> typing.Literal["cross_program_prefetch_index"] | None: ... + def HasField( + self, + field_name: typing.Literal[ + "cholesky_options", + b"cholesky_options", + "collective_device_list", + b"collective_device_list", + "convolution_dimension_numbers", + b"convolution_dimension_numbers", + "cross_program_prefetch_index", + b"cross_program_prefetch_index", + "domain_entry_sharding", + b"domain_entry_sharding", + "domain_exit_sharding", + b"domain_exit_sharding", + "dot_dimension_numbers", + b"dot_dimension_numbers", + "frontend_attributes", + b"frontend_attributes", + "gather_dimension_numbers", + b"gather_dimension_numbers", + "literal", + b"literal", + "metadata", + b"metadata", + "optional_cross_program_prefetch_index", + b"optional_cross_program_prefetch_index", + "original_value", + b"original_value", + "outfeed_shape", + b"outfeed_shape", + "padding_config", + b"padding_config", + "parameter_replication", + b"parameter_replication", + "precision_config", + b"precision_config", + "scatter_dimension_numbers", + b"scatter_dimension_numbers", + "shape", + b"shape", + "sharding", + b"sharding", + "statistics_viz", + b"statistics_viz", + "triangular_solve_options", + b"triangular_solve_options", + "window", + b"window", + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "all_reduce_id", + b"all_reduce_id", + "async_execution_thread", + b"async_execution_thread", + "backend_config", + b"backend_config", + "batch_group_count", + b"batch_group_count", + "called_computation_ids", + b"called_computation_ids", + "channel_id", + b"channel_id", + "cholesky_options", + b"cholesky_options", + "collective_device_list", + b"collective_device_list", + "comparison_direction", + b"comparison_direction", + "comparison_type", + b"comparison_type", + "constrain_layout", + b"constrain_layout", + "control_predecessor_ids", + b"control_predecessor_ids", + "convolution_dimension_numbers", + b"convolution_dimension_numbers", + "cross_program_prefetch_index", + b"cross_program_prefetch_index", + "custom_call_api_version", + b"custom_call_api_version", + "custom_call_has_side_effect", + b"custom_call_has_side_effect", + "custom_call_schedule", + b"custom_call_schedule", + "custom_call_target", + b"custom_call_target", + "delta", + b"delta", + "dimensions", + b"dimensions", + "distribution", + b"distribution", + "domain_entry_sharding", + b"domain_entry_sharding", + "domain_exit_sharding", + b"domain_exit_sharding", + "dot_dimension_numbers", + b"dot_dimension_numbers", + "dot_sparsity", + b"dot_sparsity", + "dynamic_slice_sizes", + b"dynamic_slice_sizes", + "epsilon", + b"epsilon", + "exponent_bits", + b"exponent_bits", + "feature_group_count", + b"feature_group_count", + "feature_index", + b"feature_index", + "fft_length", + b"fft_length", + "fft_type", + b"fft_type", + "frontend_attributes", + b"frontend_attributes", + "fusion_kind", + b"fusion_kind", + "gather_dimension_numbers", + b"gather_dimension_numbers", + "gather_slice_sizes", + b"gather_slice_sizes", + "id", + b"id", + "indices_are_sorted", + b"indices_are_sorted", + "infeed_config", + b"infeed_config", + "is_composite", + b"is_composite", + "is_cross_program_prefetch", + b"is_cross_program_prefetch", + "is_host_transfer", + b"is_host_transfer", + "is_stable", + b"is_stable", + "k", + b"k", + "largest", + b"largest", + "literal", + b"literal", + "mantissa_bits", + b"mantissa_bits", + "metadata", + b"metadata", + "name", + b"name", + "opcode", + b"opcode", + "operand_ids", + b"operand_ids", + "operand_shapes_with_layout", + b"operand_shapes_with_layout", + "optional_cross_program_prefetch_index", + b"optional_cross_program_prefetch_index", + "original_value", + b"original_value", + "outfeed_config", + b"outfeed_config", + "outfeed_shape", + b"outfeed_shape", + "output_operand_aliasing", + b"output_operand_aliasing", + "padding_config", + b"padding_config", + "padding_type", + b"padding_type", + "parameter_number", + b"parameter_number", + "parameter_replication", + b"parameter_replication", + "precision_config", + b"precision_config", + "replica_groups", + b"replica_groups", + "rng_algorithm", + b"rng_algorithm", + "scatter_dimension_numbers", + b"scatter_dimension_numbers", + "shape", + b"shape", + "sharding", + b"sharding", + "slice_dimensions", + b"slice_dimensions", + "source_target_pairs", + b"source_target_pairs", + "statistics_viz", + b"statistics_viz", + "triangular_solve_options", + b"triangular_solve_options", + "tuple_index", + b"tuple_index", + "unique_indices", + b"unique_indices", + "use_global_device_ids", + b"use_global_device_ids", + "window", + b"window", + ], + ) -> None: ... + def WhichOneof( + self, oneof_group: typing.Literal["optional_cross_program_prefetch_index", b"optional_cross_program_prefetch_index"] + ) -> typing.Literal["cross_program_prefetch_index"] | None: ... global___HloInstructionProto = HloInstructionProto @@ -695,7 +919,25 @@ class HloComputationProto(google.protobuf.message.Message): execution_thread: builtins.str | None = ..., ) -> None: ... def HasField(self, field_name: typing.Literal["program_shape", b"program_shape"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["execution_thread", b"execution_thread", "id", b"id", "instructions", b"instructions", "is_fusion_computation", b"is_fusion_computation", "name", b"name", "program_shape", b"program_shape", "root_id", b"root_id"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "execution_thread", + b"execution_thread", + "id", + b"id", + "instructions", + b"instructions", + "is_fusion_computation", + b"is_fusion_computation", + "name", + b"name", + "program_shape", + b"program_shape", + "root_id", + b"root_id", + ], + ) -> None: ... global___HloComputationProto = HloComputationProto @@ -714,11 +956,7 @@ class HloScheduleProto(google.protobuf.message.Message): INSTRUCTION_IDS_FIELD_NUMBER: builtins.int @property def instruction_ids(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.int]: ... - def __init__( - self, - *, - instruction_ids: collections.abc.Iterable[builtins.int] | None = ..., - ) -> None: ... + def __init__(self, *, instruction_ids: collections.abc.Iterable[builtins.int] | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["instruction_ids", b"instruction_ids"]) -> None: ... @typing.final @@ -731,23 +969,20 @@ class HloScheduleProto(google.protobuf.message.Message): @property def value(self) -> global___HloScheduleProto.InstructionSequence: ... def __init__( - self, - *, - key: builtins.int | None = ..., - value: global___HloScheduleProto.InstructionSequence | None = ..., + self, *, key: builtins.int | None = ..., value: global___HloScheduleProto.InstructionSequence | None = ... ) -> None: ... def HasField(self, field_name: typing.Literal["value", b"value"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["key", b"key", "value", b"value"]) -> None: ... SEQUENCES_FIELD_NUMBER: builtins.int @property - def sequences(self) -> google.protobuf.internal.containers.MessageMap[builtins.int, global___HloScheduleProto.InstructionSequence]: + def sequences( + self, + ) -> google.protobuf.internal.containers.MessageMap[builtins.int, global___HloScheduleProto.InstructionSequence]: """Map from computation id to sequence.""" def __init__( - self, - *, - sequences: collections.abc.Mapping[builtins.int, global___HloScheduleProto.InstructionSequence] | None = ..., + self, *, sequences: collections.abc.Mapping[builtins.int, global___HloScheduleProto.InstructionSequence] | None = ... ) -> None: ... def ClearField(self, field_name: typing.Literal["sequences", b"sequences"]) -> None: ... @@ -800,15 +1035,29 @@ class HloInputOutputAliasProto(google.protobuf.message.Message): parameter_shape_index: collections.abc.Iterable[builtins.int] | None = ..., kind: global___Kind.ValueType | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["kind", b"kind", "output_shape_index", b"output_shape_index", "parameter_number", b"parameter_number", "parameter_shape_index", b"parameter_shape_index"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "kind", + b"kind", + "output_shape_index", + b"output_shape_index", + "parameter_number", + b"parameter_number", + "parameter_shape_index", + b"parameter_shape_index", + ], + ) -> None: ... ENTRIES_FIELD_NUMBER: builtins.int @property - def entries(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___HloInputOutputAliasProto.AliasEntryProto]: ... - def __init__( + def entries( self, - *, - entries: collections.abc.Iterable[global___HloInputOutputAliasProto.AliasEntryProto] | None = ..., + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[ + global___HloInputOutputAliasProto.AliasEntryProto + ]: ... + def __init__( + self, *, entries: collections.abc.Iterable[global___HloInputOutputAliasProto.AliasEntryProto] | None = ... ) -> None: ... def ClearField(self, field_name: typing.Literal["entries", b"entries"]) -> None: ... @@ -850,15 +1099,22 @@ class HloBufferDonorProto(google.protobuf.message.Message): parameter_number: builtins.int | None = ..., parameter_shape_index: collections.abc.Iterable[builtins.int] | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["parameter_number", b"parameter_number", "parameter_shape_index", b"parameter_shape_index"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "parameter_number", b"parameter_number", "parameter_shape_index", b"parameter_shape_index" + ], + ) -> None: ... ENTRIES_FIELD_NUMBER: builtins.int @property - def entries(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___HloBufferDonorProto.BufferDonorEntryProto]: ... - def __init__( + def entries( self, - *, - entries: collections.abc.Iterable[global___HloBufferDonorProto.BufferDonorEntryProto] | None = ..., + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[ + global___HloBufferDonorProto.BufferDonorEntryProto + ]: ... + def __init__( + self, *, entries: collections.abc.Iterable[global___HloBufferDonorProto.BufferDonorEntryProto] | None = ... ) -> None: ... def ClearField(self, field_name: typing.Literal["entries", b"entries"]) -> None: ... @@ -882,7 +1138,9 @@ class CrossProgramPrefetch(google.protobuf.message.Message): index: collections.abc.Iterable[builtins.int] | None = ..., offset: builtins.int | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["index", b"index", "offset", b"offset", "parameter", b"parameter"]) -> None: ... + def ClearField( + self, field_name: typing.Literal["index", b"index", "offset", b"offset", "parameter", b"parameter"] + ) -> None: ... global___CrossProgramPrefetch = CrossProgramPrefetch @@ -927,7 +1185,12 @@ class StackFrameIndexProto(google.protobuf.message.Message): line: builtins.int | None = ..., column: builtins.int | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["column", b"column", "file_name_id", b"file_name_id", "function_name_id", b"function_name_id", "line", b"line"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "column", b"column", "file_name_id", b"file_name_id", "function_name_id", b"function_name_id", "line", b"line" + ], + ) -> None: ... @typing.final class StackFrame(google.protobuf.message.Message): @@ -942,12 +1205,11 @@ class StackFrameIndexProto(google.protobuf.message.Message): parent_frame_id: builtins.int """1-based position of the parent frame.""" def __init__( - self, - *, - file_location_id: builtins.int | None = ..., - parent_frame_id: builtins.int | None = ..., + self, *, file_location_id: builtins.int | None = ..., parent_frame_id: builtins.int | None = ... + ) -> None: ... + def ClearField( + self, field_name: typing.Literal["file_location_id", b"file_location_id", "parent_frame_id", b"parent_frame_id"] ) -> None: ... - def ClearField(self, field_name: typing.Literal["file_location_id", b"file_location_id", "parent_frame_id", b"parent_frame_id"]) -> None: ... FILE_NAMES_FIELD_NUMBER: builtins.int FUNCTION_NAMES_FIELD_NUMBER: builtins.int @@ -962,11 +1224,15 @@ class StackFrameIndexProto(google.protobuf.message.Message): """Flat index array of function names.""" @property - def file_locations(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___StackFrameIndexProto.FileLocation]: + def file_locations( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___StackFrameIndexProto.FileLocation]: """Flat index array of file locations.""" @property - def stack_frames(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___StackFrameIndexProto.StackFrame]: + def stack_frames( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___StackFrameIndexProto.StackFrame]: """Flat index array of frames.""" def __init__( @@ -977,7 +1243,19 @@ class StackFrameIndexProto(google.protobuf.message.Message): file_locations: collections.abc.Iterable[global___StackFrameIndexProto.FileLocation] | None = ..., stack_frames: collections.abc.Iterable[global___StackFrameIndexProto.StackFrame] | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["file_locations", b"file_locations", "file_names", b"file_names", "function_names", b"function_names", "stack_frames", b"stack_frames"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "file_locations", + b"file_locations", + "file_names", + b"file_names", + "function_names", + b"function_names", + "stack_frames", + b"stack_frames", + ], + ) -> None: ... global___StackFrameIndexProto = StackFrameIndexProto @@ -991,7 +1269,9 @@ class HloModuleProto(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _ProfileTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[HloModuleProto._ProfileType.ValueType], builtins.type): + class _ProfileTypeEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[HloModuleProto._ProfileType.ValueType], builtins.type + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor INVALID: HloModuleProto._ProfileType.ValueType # 0 FLAG: HloModuleProto._ProfileType.ValueType # 1 @@ -1040,7 +1320,21 @@ class HloModuleProto(google.protobuf.message.Message): compilation_event: tensorflow.compiler.xla.xla_data_pb2.CompilationEvent.ValueType | None = ..., fingerprint: builtins.str | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["compilation_event", b"compilation_event", "fingerprint", b"fingerprint", "profile_source", b"profile_source", "profile_type", b"profile_type", "relative_speedup", b"relative_speedup"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "compilation_event", + b"compilation_event", + "fingerprint", + b"fingerprint", + "profile_source", + b"profile_source", + "profile_type", + b"profile_type", + "relative_speedup", + b"relative_speedup", + ], + ) -> None: ... NAME_FIELD_NUMBER: builtins.int ENTRY_COMPUTATION_NAME_FIELD_NUMBER: builtins.int @@ -1092,13 +1386,19 @@ class HloModuleProto(google.protobuf.message.Message): """Describes the information of input buffer donors.""" @property - def cross_program_prefetches(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___CrossProgramPrefetch]: ... + def cross_program_prefetches( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___CrossProgramPrefetch]: ... @property def spmd_output_sharding(self) -> tensorflow.compiler.xla.xla_data_pb2.OpSharding: ... @property - def spmd_parameters_shardings(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[tensorflow.compiler.xla.xla_data_pb2.OpSharding]: ... + def spmd_parameters_shardings( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[tensorflow.compiler.xla.xla_data_pb2.OpSharding]: ... @property - def profile_info(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___HloModuleProto.ProfileInfo]: + def profile_info( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___HloModuleProto.ProfileInfo]: """Profile information for the HLO module.""" @property @@ -1135,8 +1435,68 @@ class HloModuleProto(google.protobuf.message.Message): stack_frame_index: global___StackFrameIndexProto | None = ..., frontend_attributes: tensorflow.compiler.xla.xla_data_pb2.FrontendAttributes | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["buffer_donor", b"buffer_donor", "device_assignment", b"device_assignment", "frontend_attributes", b"frontend_attributes", "host_program_shape", b"host_program_shape", "input_output_alias", b"input_output_alias", "schedule", b"schedule", "spmd_output_sharding", b"spmd_output_sharding", "stack_frame_index", b"stack_frame_index"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["buffer_donor", b"buffer_donor", "computations", b"computations", "cross_program_prefetches", b"cross_program_prefetches", "device_assignment", b"device_assignment", "entry_computation_id", b"entry_computation_id", "entry_computation_name", b"entry_computation_name", "frontend_attributes", b"frontend_attributes", "host_program_shape", b"host_program_shape", "id", b"id", "input_output_alias", b"input_output_alias", "is_dynamic", b"is_dynamic", "name", b"name", "profile_info", b"profile_info", "schedule", b"schedule", "spmd_output_sharding", b"spmd_output_sharding", "spmd_parameters_shardings", b"spmd_parameters_shardings", "stack_frame_index", b"stack_frame_index", "use_auto_spmd_partitioning", b"use_auto_spmd_partitioning"]) -> None: ... + def HasField( + self, + field_name: typing.Literal[ + "buffer_donor", + b"buffer_donor", + "device_assignment", + b"device_assignment", + "frontend_attributes", + b"frontend_attributes", + "host_program_shape", + b"host_program_shape", + "input_output_alias", + b"input_output_alias", + "schedule", + b"schedule", + "spmd_output_sharding", + b"spmd_output_sharding", + "stack_frame_index", + b"stack_frame_index", + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "buffer_donor", + b"buffer_donor", + "computations", + b"computations", + "cross_program_prefetches", + b"cross_program_prefetches", + "device_assignment", + b"device_assignment", + "entry_computation_id", + b"entry_computation_id", + "entry_computation_name", + b"entry_computation_name", + "frontend_attributes", + b"frontend_attributes", + "host_program_shape", + b"host_program_shape", + "id", + b"id", + "input_output_alias", + b"input_output_alias", + "is_dynamic", + b"is_dynamic", + "name", + b"name", + "profile_info", + b"profile_info", + "schedule", + b"schedule", + "spmd_output_sharding", + b"spmd_output_sharding", + "spmd_parameters_shardings", + b"spmd_parameters_shardings", + "stack_frame_index", + b"stack_frame_index", + "use_auto_spmd_partitioning", + b"use_auto_spmd_partitioning", + ], + ) -> None: ... global___HloModuleProto = HloModuleProto @@ -1169,7 +1529,12 @@ class LogicalBufferProto(google.protobuf.message.Message): instruction_id: builtins.int | None = ..., shape_index: collections.abc.Iterable[builtins.int] | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["instruction_id", b"instruction_id", "instruction_name", b"instruction_name", "shape_index", b"shape_index"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "instruction_id", b"instruction_id", "instruction_name", b"instruction_name", "shape_index", b"shape_index" + ], + ) -> None: ... ID_FIELD_NUMBER: builtins.int SIZE_FIELD_NUMBER: builtins.int @@ -1191,7 +1556,9 @@ class LogicalBufferProto(google.protobuf.message.Message): color: builtins.int | None = ..., ) -> None: ... def HasField(self, field_name: typing.Literal["defined_at", b"defined_at"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["color", b"color", "defined_at", b"defined_at", "id", b"id", "size", b"size"]) -> None: ... + def ClearField( + self, field_name: typing.Literal["color", b"color", "defined_at", b"defined_at", "id", b"id", "size", b"size"] + ) -> None: ... global___LogicalBufferProto = LogicalBufferProto @@ -1222,7 +1589,9 @@ class BufferAllocationProto(google.protobuf.message.Message): offset: builtins.int | None = ..., size: builtins.int | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["logical_buffer_id", b"logical_buffer_id", "offset", b"offset", "size", b"size"]) -> None: ... + def ClearField( + self, field_name: typing.Literal["logical_buffer_id", b"logical_buffer_id", "offset", b"offset", "size", b"size"] + ) -> None: ... INDEX_FIELD_NUMBER: builtins.int SIZE_FIELD_NUMBER: builtins.int @@ -1247,7 +1616,9 @@ class BufferAllocationProto(google.protobuf.message.Message): @property def parameter_shape_index(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.int]: ... @property - def assigned(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___BufferAllocationProto.Assigned]: ... + def assigned( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___BufferAllocationProto.Assigned]: ... def __init__( self, *, @@ -1263,7 +1634,33 @@ class BufferAllocationProto(google.protobuf.message.Message): color: builtins.int | None = ..., assigned: collections.abc.Iterable[global___BufferAllocationProto.Assigned] | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["assigned", b"assigned", "color", b"color", "index", b"index", "is_constant", b"is_constant", "is_entry_computation_parameter", b"is_entry_computation_parameter", "is_thread_local", b"is_thread_local", "is_tuple", b"is_tuple", "maybe_live_out", b"maybe_live_out", "parameter_number", b"parameter_number", "parameter_shape_index", b"parameter_shape_index", "size", b"size"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "assigned", + b"assigned", + "color", + b"color", + "index", + b"index", + "is_constant", + b"is_constant", + "is_entry_computation_parameter", + b"is_entry_computation_parameter", + "is_thread_local", + b"is_thread_local", + "is_tuple", + b"is_tuple", + "maybe_live_out", + b"maybe_live_out", + "parameter_number", + b"parameter_number", + "parameter_shape_index", + b"parameter_shape_index", + "size", + b"size", + ], + ) -> None: ... global___BufferAllocationProto = BufferAllocationProto @@ -1285,7 +1682,9 @@ class HeapSimulatorTrace(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _KindEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[HeapSimulatorTrace.Event._Kind.ValueType], builtins.type): + class _KindEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[HeapSimulatorTrace.Event._Kind.ValueType], builtins.type + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor ALLOC: HeapSimulatorTrace.Event._Kind.ValueType # 0 """A memory region was allocated for the buffer.""" @@ -1339,7 +1738,21 @@ class HeapSimulatorTrace(google.protobuf.message.Message): instruction_name: builtins.str | None = ..., share_with_canonical_id: builtins.int | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["buffer_id", b"buffer_id", "computation_name", b"computation_name", "instruction_name", b"instruction_name", "kind", b"kind", "share_with_canonical_id", b"share_with_canonical_id"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "buffer_id", + b"buffer_id", + "computation_name", + b"computation_name", + "instruction_name", + b"instruction_name", + "kind", + b"kind", + "share_with_canonical_id", + b"share_with_canonical_id", + ], + ) -> None: ... EVENTS_FIELD_NUMBER: builtins.int WHOLE_MODULE_SIMULATION_FIELD_NUMBER: builtins.int @@ -1347,7 +1760,9 @@ class HeapSimulatorTrace(google.protobuf.message.Message): whole_module_simulation: builtins.bool buffer_allocation_index: builtins.int @property - def events(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___HeapSimulatorTrace.Event]: ... + def events( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___HeapSimulatorTrace.Event]: ... def __init__( self, *, @@ -1355,7 +1770,17 @@ class HeapSimulatorTrace(google.protobuf.message.Message): whole_module_simulation: builtins.bool | None = ..., buffer_allocation_index: builtins.int | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["buffer_allocation_index", b"buffer_allocation_index", "events", b"events", "whole_module_simulation", b"whole_module_simulation"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "buffer_allocation_index", + b"buffer_allocation_index", + "events", + b"events", + "whole_module_simulation", + b"whole_module_simulation", + ], + ) -> None: ... global___HeapSimulatorTrace = HeapSimulatorTrace @@ -1373,10 +1798,7 @@ class HloModuleGroupProto(google.protobuf.message.Message): @property def hlo_modules(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___HloModuleProto]: ... def __init__( - self, - *, - name: builtins.str | None = ..., - hlo_modules: collections.abc.Iterable[global___HloModuleProto] | None = ..., + self, *, name: builtins.str | None = ..., hlo_modules: collections.abc.Iterable[global___HloModuleProto] | None = ... ) -> None: ... def ClearField(self, field_name: typing.Literal["hlo_modules", b"hlo_modules", "name", b"name"]) -> None: ... @@ -1402,26 +1824,33 @@ class BufferAssignmentProto(google.protobuf.message.Message): @property def location(self) -> global___LogicalBufferProto.Location: ... def __init__( - self, - *, - source_buffer_id: builtins.int | None = ..., - location: global___LogicalBufferProto.Location | None = ..., + self, *, source_buffer_id: builtins.int | None = ..., location: global___LogicalBufferProto.Location | None = ... ) -> None: ... def HasField(self, field_name: typing.Literal["location", b"location"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["location", b"location", "source_buffer_id", b"source_buffer_id"]) -> None: ... + def ClearField( + self, field_name: typing.Literal["location", b"location", "source_buffer_id", b"source_buffer_id"] + ) -> None: ... LOGICAL_BUFFERS_FIELD_NUMBER: builtins.int BUFFER_ALIASES_FIELD_NUMBER: builtins.int BUFFER_ALLOCATIONS_FIELD_NUMBER: builtins.int HEAP_SIMULATOR_TRACES_FIELD_NUMBER: builtins.int @property - def logical_buffers(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___LogicalBufferProto]: ... + def logical_buffers( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___LogicalBufferProto]: ... @property - def buffer_aliases(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___BufferAssignmentProto.BufferAlias]: ... + def buffer_aliases( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___BufferAssignmentProto.BufferAlias]: ... @property - def buffer_allocations(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___BufferAllocationProto]: ... + def buffer_allocations( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___BufferAllocationProto]: ... @property - def heap_simulator_traces(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___HeapSimulatorTrace]: ... + def heap_simulator_traces( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___HeapSimulatorTrace]: ... def __init__( self, *, @@ -1430,7 +1859,19 @@ class BufferAssignmentProto(google.protobuf.message.Message): buffer_allocations: collections.abc.Iterable[global___BufferAllocationProto] | None = ..., heap_simulator_traces: collections.abc.Iterable[global___HeapSimulatorTrace] | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["buffer_aliases", b"buffer_aliases", "buffer_allocations", b"buffer_allocations", "heap_simulator_traces", b"heap_simulator_traces", "logical_buffers", b"logical_buffers"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "buffer_aliases", + b"buffer_aliases", + "buffer_allocations", + b"buffer_allocations", + "heap_simulator_traces", + b"heap_simulator_traces", + "logical_buffers", + b"logical_buffers", + ], + ) -> None: ... global___BufferAssignmentProto = BufferAssignmentProto @@ -1447,13 +1888,14 @@ class HloProto(google.protobuf.message.Message): @property def buffer_assignment(self) -> global___BufferAssignmentProto: ... def __init__( - self, - *, - hlo_module: global___HloModuleProto | None = ..., - buffer_assignment: global___BufferAssignmentProto | None = ..., + self, *, hlo_module: global___HloModuleProto | None = ..., buffer_assignment: global___BufferAssignmentProto | None = ... + ) -> None: ... + def HasField( + self, field_name: typing.Literal["buffer_assignment", b"buffer_assignment", "hlo_module", b"hlo_module"] + ) -> builtins.bool: ... + def ClearField( + self, field_name: typing.Literal["buffer_assignment", b"buffer_assignment", "hlo_module", b"hlo_module"] ) -> None: ... - def HasField(self, field_name: typing.Literal["buffer_assignment", b"buffer_assignment", "hlo_module", b"hlo_module"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["buffer_assignment", b"buffer_assignment", "hlo_module", b"hlo_module"]) -> None: ... global___HloProto = HloProto @@ -1477,7 +1919,9 @@ class HloSnapshot(google.protobuf.message.Message): """The hlo graph.""" @property - def arguments(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[tensorflow.compiler.xla.xla_data_pb2.LiteralProto]: + def arguments( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[tensorflow.compiler.xla.xla_data_pb2.LiteralProto]: """The arguments passed to the graph.""" @property @@ -1493,7 +1937,12 @@ class HloSnapshot(google.protobuf.message.Message): execution_platform: builtins.str | None = ..., ) -> None: ... def HasField(self, field_name: typing.Literal["hlo", b"hlo", "result", b"result"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["arguments", b"arguments", "execution_platform", b"execution_platform", "hlo", b"hlo", "result", b"result"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "arguments", b"arguments", "execution_platform", b"execution_platform", "hlo", b"hlo", "result", b"result" + ], + ) -> None: ... global___HloSnapshot = HloSnapshot @@ -1542,7 +1991,21 @@ class HloModuleMetadataProto(google.protobuf.message.Message): partitioned_module_ids: collections.abc.Iterable[builtins.int] | None = ..., pass_metadata: collections.abc.Iterable[global___HloPassMetadata] | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["canonical_module_id", b"canonical_module_id", "module_group_name", b"module_group_name", "original_module_id", b"original_module_id", "partitioned_module_ids", b"partitioned_module_ids", "pass_metadata", b"pass_metadata"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "canonical_module_id", + b"canonical_module_id", + "module_group_name", + b"module_group_name", + "original_module_id", + b"original_module_id", + "partitioned_module_ids", + b"partitioned_module_ids", + "pass_metadata", + b"pass_metadata", + ], + ) -> None: ... global___HloModuleMetadataProto = HloModuleMetadataProto @@ -1621,6 +2084,30 @@ class HloPassMetadata(google.protobuf.message.Message): custom_metadata: google.protobuf.any_pb2.Any | None = ..., ) -> None: ... def HasField(self, field_name: typing.Literal["custom_metadata", b"custom_metadata"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["custom_metadata", b"custom_metadata", "dump_filenames", b"dump_filenames", "end_timestamp_usec", b"end_timestamp_usec", "module_changed", b"module_changed", "module_group_module_ids", b"module_group_module_ids", "module_id", b"module_id", "pass_id", b"pass_id", "pass_name", b"pass_name", "pipeline_name", b"pipeline_name", "start_timestamp_usec", b"start_timestamp_usec"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "custom_metadata", + b"custom_metadata", + "dump_filenames", + b"dump_filenames", + "end_timestamp_usec", + b"end_timestamp_usec", + "module_changed", + b"module_changed", + "module_group_module_ids", + b"module_group_module_ids", + "module_id", + b"module_id", + "pass_id", + b"pass_id", + "pass_name", + b"pass_name", + "pipeline_name", + b"pipeline_name", + "start_timestamp_usec", + b"start_timestamp_usec", + ], + ) -> None: ... global___HloPassMetadata = HloPassMetadata diff --git a/stubs/tensorflow/tensorflow/compiler/xla/service/hlo_profile_printer_data_pb2.pyi b/stubs/tensorflow/tensorflow/compiler/xla/service/hlo_profile_printer_data_pb2.pyi index d1871110d55b..5699e8cc13fc 100644 --- a/stubs/tensorflow/tensorflow/compiler/xla/service/hlo_profile_printer_data_pb2.pyi +++ b/stubs/tensorflow/tensorflow/compiler/xla/service/hlo_profile_printer_data_pb2.pyi @@ -73,7 +73,27 @@ class HloProfilePrinterData(google.protobuf.message.Message): optimal_seconds: builtins.float | None = ..., profile_index: builtins.int | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["bytes_accessed", b"bytes_accessed", "category", b"category", "flop_count", b"flop_count", "long_name", b"long_name", "optimal_seconds", b"optimal_seconds", "profile_index", b"profile_index", "short_name", b"short_name", "transcendental_count", b"transcendental_count"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "bytes_accessed", + b"bytes_accessed", + "category", + b"category", + "flop_count", + b"flop_count", + "long_name", + b"long_name", + "optimal_seconds", + b"optimal_seconds", + "profile_index", + b"profile_index", + "short_name", + b"short_name", + "transcendental_count", + b"transcendental_count", + ], + ) -> None: ... @typing.final class HloComputationInfo(google.protobuf.message.Message): @@ -90,7 +110,11 @@ class HloProfilePrinterData(google.protobuf.message.Message): corresponding to this HloComputationInfo. """ @property - def instruction_infos(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___HloProfilePrinterData.HloInstructionInfo]: + def instruction_infos( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[ + global___HloProfilePrinterData.HloInstructionInfo + ]: """HloInstructionInfos for every HloInstruction in the HloComputation for corresponding to this HloComputattionInfo. """ @@ -102,7 +126,12 @@ class HloProfilePrinterData(google.protobuf.message.Message): profile_index: builtins.int | None = ..., instruction_infos: collections.abc.Iterable[global___HloProfilePrinterData.HloInstructionInfo] | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["instruction_infos", b"instruction_infos", "name", b"name", "profile_index", b"profile_index"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "instruction_infos", b"instruction_infos", "name", b"name", "profile_index", b"profile_index" + ], + ) -> None: ... @typing.final class ExtraMetricsEntry(google.protobuf.message.Message): @@ -112,12 +141,7 @@ class HloProfilePrinterData(google.protobuf.message.Message): VALUE_FIELD_NUMBER: builtins.int key: builtins.str value: builtins.int - def __init__( - self, - *, - key: builtins.str | None = ..., - value: builtins.int | None = ..., - ) -> None: ... + def __init__(self, *, key: builtins.str | None = ..., value: builtins.int | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["key", b"key", "value", b"value"]) -> None: ... COMPUTATION_INFOS_FIELD_NUMBER: builtins.int @@ -129,7 +153,9 @@ class HloProfilePrinterData(google.protobuf.message.Message): entry_computation: builtins.str """Name of the entry computation.""" @property - def computation_infos(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___HloProfilePrinterData.HloComputationInfo]: + def computation_infos( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___HloProfilePrinterData.HloComputationInfo]: """HloComputationInfos for every HloComputation in the HloModule.""" @property @@ -144,6 +170,18 @@ class HloProfilePrinterData(google.protobuf.message.Message): extra_metrics: collections.abc.Mapping[builtins.str, builtins.int] | None = ..., entry_computation: builtins.str | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["computation_infos", b"computation_infos", "entry_computation", b"entry_computation", "extra_metrics", b"extra_metrics", "profile_counters_size", b"profile_counters_size"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "computation_infos", + b"computation_infos", + "entry_computation", + b"entry_computation", + "extra_metrics", + b"extra_metrics", + "profile_counters_size", + b"profile_counters_size", + ], + ) -> None: ... global___HloProfilePrinterData = HloProfilePrinterData diff --git a/stubs/tensorflow/tensorflow/compiler/xla/service/metrics_pb2.pyi b/stubs/tensorflow/tensorflow/compiler/xla/service/metrics_pb2.pyi index 44592002fdd9..05eaf7379849 100644 --- a/stubs/tensorflow/tensorflow/compiler/xla/service/metrics_pb2.pyi +++ b/stubs/tensorflow/tensorflow/compiler/xla/service/metrics_pb2.pyi @@ -56,8 +56,22 @@ class PassMetrics(google.protobuf.message.Message): pass_duration: google.protobuf.duration_pb2.Duration | None = ..., custom_metrics: google.protobuf.any_pb2.Any | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["custom_metrics", b"custom_metrics", "pass_duration", b"pass_duration"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["custom_metrics", b"custom_metrics", "module_id", b"module_id", "pass_duration", b"pass_duration", "pass_name", b"pass_name"]) -> None: ... + def HasField( + self, field_name: typing.Literal["custom_metrics", b"custom_metrics", "pass_duration", b"pass_duration"] + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "custom_metrics", + b"custom_metrics", + "module_id", + b"module_id", + "pass_duration", + b"pass_duration", + "pass_name", + b"pass_name", + ], + ) -> None: ... global___PassMetrics = PassMetrics @@ -95,8 +109,64 @@ class JobInfo(google.protobuf.message.Message): task_id: builtins.int | None = ..., task_uid: builtins.int | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["_cell", b"_cell", "_name", b"_name", "_task_id", b"_task_id", "_task_uid", b"_task_uid", "_uid", b"_uid", "_user", b"_user", "cell", b"cell", "name", b"name", "task_id", b"task_id", "task_uid", b"task_uid", "uid", b"uid", "user", b"user"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["_cell", b"_cell", "_name", b"_name", "_task_id", b"_task_id", "_task_uid", b"_task_uid", "_uid", b"_uid", "_user", b"_user", "cell", b"cell", "name", b"name", "task_id", b"task_id", "task_uid", b"task_uid", "uid", b"uid", "user", b"user"]) -> None: ... + def HasField( + self, + field_name: typing.Literal[ + "_cell", + b"_cell", + "_name", + b"_name", + "_task_id", + b"_task_id", + "_task_uid", + b"_task_uid", + "_uid", + b"_uid", + "_user", + b"_user", + "cell", + b"cell", + "name", + b"name", + "task_id", + b"task_id", + "task_uid", + b"task_uid", + "uid", + b"uid", + "user", + b"user", + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "_cell", + b"_cell", + "_name", + b"_name", + "_task_id", + b"_task_id", + "_task_uid", + b"_task_uid", + "_uid", + b"_uid", + "_user", + b"_user", + "cell", + b"cell", + "name", + b"name", + "task_id", + b"task_id", + "task_uid", + b"task_uid", + "uid", + b"uid", + "user", + b"user", + ], + ) -> None: ... @typing.overload def WhichOneof(self, oneof_group: typing.Literal["_cell", b"_cell"]) -> typing.Literal["cell"] | None: ... @typing.overload @@ -122,7 +192,10 @@ class CompilationLogEntry(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _CompilationStageEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[CompilationLogEntry._CompilationStage.ValueType], builtins.type): + class _CompilationStageEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[CompilationLogEntry._CompilationStage.ValueType], + builtins.type, + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor UNSPECIFIED: CompilationLogEntry._CompilationStage.ValueType # 0 END_TO_END: CompilationLogEntry._CompilationStage.ValueType # 1 @@ -184,7 +257,27 @@ class CompilationLogEntry(google.protobuf.message.Message): module_ids: collections.abc.Iterable[builtins.int] | None = ..., job_info: global___JobInfo | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["duration", b"duration", "job_info", b"job_info", "timestamp", b"timestamp"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["duration", b"duration", "job_info", b"job_info", "module_ids", b"module_ids", "pass_metrics", b"pass_metrics", "stage", b"stage", "task_index", b"task_index", "timestamp", b"timestamp"]) -> None: ... + def HasField( + self, field_name: typing.Literal["duration", b"duration", "job_info", b"job_info", "timestamp", b"timestamp"] + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "duration", + b"duration", + "job_info", + b"job_info", + "module_ids", + b"module_ids", + "pass_metrics", + b"pass_metrics", + "stage", + b"stage", + "task_index", + b"task_index", + "timestamp", + b"timestamp", + ], + ) -> None: ... global___CompilationLogEntry = CompilationLogEntry diff --git a/stubs/tensorflow/tensorflow/compiler/xla/service/test_compilation_environment_pb2.pyi b/stubs/tensorflow/tensorflow/compiler/xla/service/test_compilation_environment_pb2.pyi index c7a849570d8d..6b3b31b94f3d 100644 --- a/stubs/tensorflow/tensorflow/compiler/xla/service/test_compilation_environment_pb2.pyi +++ b/stubs/tensorflow/tensorflow/compiler/xla/service/test_compilation_environment_pb2.pyi @@ -31,11 +31,7 @@ class TestCompilationEnvironment1(google.protobuf.message.Message): SOME_FLAG_FIELD_NUMBER: builtins.int some_flag: builtins.int - def __init__( - self, - *, - some_flag: builtins.int | None = ..., - ) -> None: ... + def __init__(self, *, some_flag: builtins.int | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["some_flag", b"some_flag"]) -> None: ... global___TestCompilationEnvironment1 = TestCompilationEnvironment1 @@ -46,11 +42,7 @@ class TestCompilationEnvironment2(google.protobuf.message.Message): SOME_OTHER_FLAG_FIELD_NUMBER: builtins.int some_other_flag: builtins.int - def __init__( - self, - *, - some_other_flag: builtins.int | None = ..., - ) -> None: ... + def __init__(self, *, some_other_flag: builtins.int | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["some_other_flag", b"some_other_flag"]) -> None: ... global___TestCompilationEnvironment2 = TestCompilationEnvironment2 @@ -61,11 +53,7 @@ class TestCompilationEnvironment3(google.protobuf.message.Message): A_THIRD_FLAG_FIELD_NUMBER: builtins.int a_third_flag: builtins.int - def __init__( - self, - *, - a_third_flag: builtins.int | None = ..., - ) -> None: ... + def __init__(self, *, a_third_flag: builtins.int | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["a_third_flag", b"a_third_flag"]) -> None: ... global___TestCompilationEnvironment3 = TestCompilationEnvironment3 diff --git a/stubs/tensorflow/tensorflow/compiler/xla/service/xla_compile_result_pb2.pyi b/stubs/tensorflow/tensorflow/compiler/xla/service/xla_compile_result_pb2.pyi index 3c5dde2bac5c..e17f9559b648 100644 --- a/stubs/tensorflow/tensorflow/compiler/xla/service/xla_compile_result_pb2.pyi +++ b/stubs/tensorflow/tensorflow/compiler/xla/service/xla_compile_result_pb2.pyi @@ -73,8 +73,36 @@ class CompilerPerfStats(google.protobuf.message.Message): compilation_duration: google.protobuf.duration_pb2.Duration | None = ..., total_duration: google.protobuf.duration_pb2.Duration | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["compilation_duration", b"compilation_duration", "compilation_prologue_duration", b"compilation_prologue_duration", "hlo_verification_duration", b"hlo_verification_duration", "init_duration", b"init_duration", "total_duration", b"total_duration"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["compilation_duration", b"compilation_duration", "compilation_prologue_duration", b"compilation_prologue_duration", "hlo_verification_duration", b"hlo_verification_duration", "init_duration", b"init_duration", "total_duration", b"total_duration"]) -> None: ... + def HasField( + self, + field_name: typing.Literal[ + "compilation_duration", + b"compilation_duration", + "compilation_prologue_duration", + b"compilation_prologue_duration", + "hlo_verification_duration", + b"hlo_verification_duration", + "init_duration", + b"init_duration", + "total_duration", + b"total_duration", + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "compilation_duration", + b"compilation_duration", + "compilation_prologue_duration", + b"compilation_prologue_duration", + "hlo_verification_duration", + b"hlo_verification_duration", + "init_duration", + b"init_duration", + "total_duration", + b"total_duration", + ], + ) -> None: ... global___CompilerPerfStats = CompilerPerfStats @@ -90,12 +118,7 @@ class CompilationResult(google.protobuf.message.Message): VALUE_FIELD_NUMBER: builtins.int key: builtins.str value: builtins.int - def __init__( - self, - *, - key: builtins.str | None = ..., - value: builtins.int | None = ..., - ) -> None: ... + def __init__(self, *, key: builtins.str | None = ..., value: builtins.int | None = ...) -> None: ... def HasField(self, field_name: typing.Literal["key", b"key", "value", b"value"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["key", b"key", "value", b"value"]) -> None: ... @@ -131,7 +154,14 @@ class CompilationResult(google.protobuf.message.Message): status: tensorflow.tsl.protobuf.status_pb2.StatusProto | None = ..., counters: collections.abc.Mapping[builtins.str, builtins.int] | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["hlo_module", b"hlo_module", "perf_stats", b"perf_stats", "status", b"status"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["counters", b"counters", "hlo_module", b"hlo_module", "perf_stats", b"perf_stats", "status", b"status"]) -> None: ... + def HasField( + self, field_name: typing.Literal["hlo_module", b"hlo_module", "perf_stats", b"perf_stats", "status", b"status"] + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "counters", b"counters", "hlo_module", b"hlo_module", "perf_stats", b"perf_stats", "status", b"status" + ], + ) -> None: ... global___CompilationResult = CompilationResult diff --git a/stubs/tensorflow/tensorflow/compiler/xla/tsl/protobuf/bfc_memory_map_pb2.pyi b/stubs/tensorflow/tensorflow/compiler/xla/tsl/protobuf/bfc_memory_map_pb2.pyi index 9e1871be16a8..0fe33b725b8e 100644 --- a/stubs/tensorflow/tensorflow/compiler/xla/tsl/protobuf/bfc_memory_map_pb2.pyi +++ b/stubs/tensorflow/tensorflow/compiler/xla/tsl/protobuf/bfc_memory_map_pb2.pyi @@ -38,7 +38,21 @@ class MemAllocatorStats(google.protobuf.message.Message): largest_alloc_size: builtins.int | None = ..., fragmentation_metric: builtins.float | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["bytes_in_use", b"bytes_in_use", "fragmentation_metric", b"fragmentation_metric", "largest_alloc_size", b"largest_alloc_size", "num_allocs", b"num_allocs", "peak_bytes_in_use", b"peak_bytes_in_use"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "bytes_in_use", + b"bytes_in_use", + "fragmentation_metric", + b"fragmentation_metric", + "largest_alloc_size", + b"largest_alloc_size", + "num_allocs", + b"num_allocs", + "peak_bytes_in_use", + b"peak_bytes_in_use", + ], + ) -> None: ... global___MemAllocatorStats = MemAllocatorStats @@ -77,7 +91,29 @@ class MemChunk(google.protobuf.message.Message): in_use: builtins.bool | None = ..., step_id: builtins.int | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["action_count", b"action_count", "address", b"address", "bin", b"bin", "freed_at_count", b"freed_at_count", "in_use", b"in_use", "op_name", b"op_name", "requested_size", b"requested_size", "size", b"size", "step_id", b"step_id"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "action_count", + b"action_count", + "address", + b"address", + "bin", + b"bin", + "freed_at_count", + b"freed_at_count", + "in_use", + b"in_use", + "op_name", + b"op_name", + "requested_size", + b"requested_size", + "size", + b"size", + "step_id", + b"step_id", + ], + ) -> None: ... global___MemChunk = MemChunk @@ -104,7 +140,21 @@ class BinSummary(google.protobuf.message.Message): total_chunks_in_use: builtins.int | None = ..., total_chunks_in_bin: builtins.int | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["bin", b"bin", "total_bytes_in_bin", b"total_bytes_in_bin", "total_bytes_in_use", b"total_bytes_in_use", "total_chunks_in_bin", b"total_chunks_in_bin", "total_chunks_in_use", b"total_chunks_in_use"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "bin", + b"bin", + "total_bytes_in_bin", + b"total_bytes_in_bin", + "total_bytes_in_use", + b"total_bytes_in_use", + "total_chunks_in_bin", + b"total_chunks_in_bin", + "total_chunks_in_use", + b"total_chunks_in_use", + ], + ) -> None: ... global___BinSummary = BinSummary @@ -116,12 +166,7 @@ class SnapShot(google.protobuf.message.Message): SIZE_FIELD_NUMBER: builtins.int action_count: builtins.int size: builtins.int - def __init__( - self, - *, - action_count: builtins.int | None = ..., - size: builtins.int | None = ..., - ) -> None: ... + def __init__(self, *, action_count: builtins.int | None = ..., size: builtins.int | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["action_count", b"action_count", "size", b"size"]) -> None: ... global___SnapShot = SnapShot @@ -154,6 +199,20 @@ class MemoryDump(google.protobuf.message.Message): stats: global___MemAllocatorStats | None = ..., ) -> None: ... def HasField(self, field_name: typing.Literal["stats", b"stats"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["allocator_name", b"allocator_name", "bin_summary", b"bin_summary", "chunk", b"chunk", "snap_shot", b"snap_shot", "stats", b"stats"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "allocator_name", + b"allocator_name", + "bin_summary", + b"bin_summary", + "chunk", + b"chunk", + "snap_shot", + b"snap_shot", + "stats", + b"stats", + ], + ) -> None: ... global___MemoryDump = MemoryDump diff --git a/stubs/tensorflow/tensorflow/compiler/xla/tsl/protobuf/test_log_pb2.pyi b/stubs/tensorflow/tensorflow/compiler/xla/tsl/protobuf/test_log_pb2.pyi index 0e06ea85faa7..a0cd6ef6a44b 100644 --- a/stubs/tensorflow/tensorflow/compiler/xla/tsl/protobuf/test_log_pb2.pyi +++ b/stubs/tensorflow/tensorflow/compiler/xla/tsl/protobuf/test_log_pb2.pyi @@ -30,15 +30,16 @@ class EntryValue(google.protobuf.message.Message): STRING_VALUE_FIELD_NUMBER: builtins.int double_value: builtins.float string_value: builtins.str - def __init__( - self, - *, - double_value: builtins.float | None = ..., - string_value: builtins.str | None = ..., + def __init__(self, *, double_value: builtins.float | None = ..., string_value: builtins.str | None = ...) -> None: ... + def HasField( + self, field_name: typing.Literal["double_value", b"double_value", "kind", b"kind", "string_value", b"string_value"] + ) -> builtins.bool: ... + def ClearField( + self, field_name: typing.Literal["double_value", b"double_value", "kind", b"kind", "string_value", b"string_value"] ) -> None: ... - def HasField(self, field_name: typing.Literal["double_value", b"double_value", "kind", b"kind", "string_value", b"string_value"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["double_value", b"double_value", "kind", b"kind", "string_value", b"string_value"]) -> None: ... - def WhichOneof(self, oneof_group: typing.Literal["kind", b"kind"]) -> typing.Literal["double_value", "string_value"] | None: ... + def WhichOneof( + self, oneof_group: typing.Literal["kind", b"kind"] + ) -> typing.Literal["double_value", "string_value"] | None: ... global___EntryValue = EntryValue @@ -71,7 +72,9 @@ class MetricEntry(google.protobuf.message.Message): max_value: google.protobuf.wrappers_pb2.DoubleValue | None = ..., ) -> None: ... def HasField(self, field_name: typing.Literal["max_value", b"max_value", "min_value", b"min_value"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["max_value", b"max_value", "min_value", b"min_value", "name", b"name", "value", b"value"]) -> None: ... + def ClearField( + self, field_name: typing.Literal["max_value", b"max_value", "min_value", b"min_value", "name", b"name", "value", b"value"] + ) -> None: ... global___MetricEntry = MetricEntry @@ -97,12 +100,7 @@ class BenchmarkEntry(google.protobuf.message.Message): key: builtins.str @property def value(self) -> global___EntryValue: ... - def __init__( - self, - *, - key: builtins.str | None = ..., - value: global___EntryValue | None = ..., - ) -> None: ... + def __init__(self, *, key: builtins.str | None = ..., value: global___EntryValue | None = ...) -> None: ... def HasField(self, field_name: typing.Literal["value", b"value"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["key", b"key", "value", b"value"]) -> None: ... @@ -146,7 +144,25 @@ class BenchmarkEntry(google.protobuf.message.Message): extras: collections.abc.Mapping[builtins.str, global___EntryValue] | None = ..., metrics: collections.abc.Iterable[global___MetricEntry] | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["cpu_time", b"cpu_time", "extras", b"extras", "iters", b"iters", "metrics", b"metrics", "name", b"name", "throughput", b"throughput", "wall_time", b"wall_time"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "cpu_time", + b"cpu_time", + "extras", + b"extras", + "iters", + b"iters", + "metrics", + b"metrics", + "name", + b"name", + "throughput", + b"throughput", + "wall_time", + b"wall_time", + ], + ) -> None: ... global___BenchmarkEntry = BenchmarkEntry @@ -157,11 +173,7 @@ class BenchmarkEntries(google.protobuf.message.Message): ENTRY_FIELD_NUMBER: builtins.int @property def entry(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___BenchmarkEntry]: ... - def __init__( - self, - *, - entry: collections.abc.Iterable[global___BenchmarkEntry] | None = ..., - ) -> None: ... + def __init__(self, *, entry: collections.abc.Iterable[global___BenchmarkEntry] | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["entry", b"entry"]) -> None: ... global___BenchmarkEntries = BenchmarkEntries @@ -219,8 +231,24 @@ class CommitId(google.protobuf.message.Message): snapshot: builtins.str | None = ..., pending_changelist: builtins.int | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["changelist", b"changelist", "hash", b"hash", "kind", b"kind"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["changelist", b"changelist", "hash", b"hash", "kind", b"kind", "pending_changelist", b"pending_changelist", "snapshot", b"snapshot"]) -> None: ... + def HasField( + self, field_name: typing.Literal["changelist", b"changelist", "hash", b"hash", "kind", b"kind"] + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "changelist", + b"changelist", + "hash", + b"hash", + "kind", + b"kind", + "pending_changelist", + b"pending_changelist", + "snapshot", + b"snapshot", + ], + ) -> None: ... def WhichOneof(self, oneof_group: typing.Literal["kind", b"kind"]) -> typing.Literal["changelist", "hash"] | None: ... global___CommitId = CommitId @@ -237,12 +265,7 @@ class CPUInfo(google.protobuf.message.Message): VALUE_FIELD_NUMBER: builtins.int key: builtins.str value: builtins.int - def __init__( - self, - *, - key: builtins.str | None = ..., - value: builtins.int | None = ..., - ) -> None: ... + def __init__(self, *, key: builtins.str | None = ..., value: builtins.int | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["key", b"key", "value", b"value"]) -> None: ... NUM_CORES_FIELD_NUMBER: builtins.int @@ -277,7 +300,23 @@ class CPUInfo(google.protobuf.message.Message): cpu_governor: builtins.str | None = ..., cache_size: collections.abc.Mapping[builtins.str, builtins.int] | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["cache_size", b"cache_size", "cpu_governor", b"cpu_governor", "cpu_info", b"cpu_info", "mhz_per_cpu", b"mhz_per_cpu", "num_cores", b"num_cores", "num_cores_allowed", b"num_cores_allowed"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "cache_size", + b"cache_size", + "cpu_governor", + b"cpu_governor", + "cpu_info", + b"cpu_info", + "mhz_per_cpu", + b"mhz_per_cpu", + "num_cores", + b"num_cores", + "num_cores_allowed", + b"num_cores_allowed", + ], + ) -> None: ... global___CPUInfo = CPUInfo @@ -291,12 +330,7 @@ class MemoryInfo(google.protobuf.message.Message): """Total virtual memory in bytes""" available: builtins.int """Immediately available memory in bytes""" - def __init__( - self, - *, - total: builtins.int | None = ..., - available: builtins.int | None = ..., - ) -> None: ... + def __init__(self, *, total: builtins.int | None = ..., available: builtins.int | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["available", b"available", "total", b"total"]) -> None: ... global___MemoryInfo = MemoryInfo @@ -315,11 +349,7 @@ class GPUInfo(google.protobuf.message.Message): bus_id: builtins.str """e.g. "0000:04:00.0" """ def __init__( - self, - *, - model: builtins.str | None = ..., - uuid: builtins.str | None = ..., - bus_id: builtins.str | None = ..., + self, *, model: builtins.str | None = ..., uuid: builtins.str | None = ..., bus_id: builtins.str | None = ... ) -> None: ... def ClearField(self, field_name: typing.Literal["bus_id", b"bus_id", "model", b"model", "uuid", b"uuid"]) -> None: ... @@ -357,7 +387,23 @@ class PlatformInfo(google.protobuf.message.Message): system: builtins.str | None = ..., version: builtins.str | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["bits", b"bits", "linkage", b"linkage", "machine", b"machine", "release", b"release", "system", b"system", "version", b"version"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "bits", + b"bits", + "linkage", + b"linkage", + "machine", + b"machine", + "release", + b"release", + "system", + b"system", + "version", + b"version", + ], + ) -> None: ... global___PlatformInfo = PlatformInfo @@ -387,7 +433,12 @@ class AvailableDeviceInfo(google.protobuf.message.Message): memory_limit: builtins.int | None = ..., physical_description: builtins.str | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["memory_limit", b"memory_limit", "name", b"name", "physical_description", b"physical_description", "type", b"type"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "memory_limit", b"memory_limit", "name", b"name", "physical_description", b"physical_description", "type", b"type" + ], + ) -> None: ... global___AvailableDeviceInfo = AvailableDeviceInfo @@ -419,7 +470,9 @@ class MachineConfiguration(google.protobuf.message.Message): """Other devices that are attached and relevant (e.g. GPUInfo).""" @property - def available_device_info(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___AvailableDeviceInfo]: + def available_device_info( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___AvailableDeviceInfo]: """Devices accessible to the test (e.g. as given by list_local_devices).""" @property @@ -435,8 +488,29 @@ class MachineConfiguration(google.protobuf.message.Message): available_device_info: collections.abc.Iterable[global___AvailableDeviceInfo] | None = ..., memory_info: global___MemoryInfo | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["cpu_info", b"cpu_info", "memory_info", b"memory_info", "platform_info", b"platform_info"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["available_device_info", b"available_device_info", "cpu_info", b"cpu_info", "device_info", b"device_info", "hostname", b"hostname", "memory_info", b"memory_info", "platform_info", b"platform_info", "serial_identifier", b"serial_identifier"]) -> None: ... + def HasField( + self, + field_name: typing.Literal["cpu_info", b"cpu_info", "memory_info", b"memory_info", "platform_info", b"platform_info"], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "available_device_info", + b"available_device_info", + "cpu_info", + b"cpu_info", + "device_info", + b"device_info", + "hostname", + b"hostname", + "memory_info", + b"memory_info", + "platform_info", + b"platform_info", + "serial_identifier", + b"serial_identifier", + ], + ) -> None: ... global___MachineConfiguration = MachineConfiguration @@ -454,12 +528,7 @@ class RunConfiguration(google.protobuf.message.Message): VALUE_FIELD_NUMBER: builtins.int key: builtins.str value: builtins.str - def __init__( - self, - *, - key: builtins.str | None = ..., - value: builtins.str | None = ..., - ) -> None: ... + def __init__(self, *, key: builtins.str | None = ..., value: builtins.str | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["key", b"key", "value", b"value"]) -> None: ... ARGUMENT_FIELD_NUMBER: builtins.int @@ -497,7 +566,9 @@ class TestResults(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _BenchmarkTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[TestResults._BenchmarkType.ValueType], builtins.type): + class _BenchmarkTypeEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[TestResults._BenchmarkType.ValueType], builtins.type + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor UNKNOWN: TestResults._BenchmarkType.ValueType # 0 """Fallback for protos written before Type was introduced.""" @@ -588,7 +659,49 @@ class TestResults(google.protobuf.message.Message): run_mode: builtins.str | None = ..., tf_version: builtins.str | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["build_configuration", b"build_configuration", "commit_id", b"commit_id", "entries", b"entries", "machine_configuration", b"machine_configuration", "run_configuration", b"run_configuration"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["benchmark_type", b"benchmark_type", "build_configuration", b"build_configuration", "commit_id", b"commit_id", "entries", b"entries", "machine_configuration", b"machine_configuration", "name", b"name", "run_configuration", b"run_configuration", "run_mode", b"run_mode", "run_time", b"run_time", "start_time", b"start_time", "target", b"target", "tf_version", b"tf_version"]) -> None: ... + def HasField( + self, + field_name: typing.Literal[ + "build_configuration", + b"build_configuration", + "commit_id", + b"commit_id", + "entries", + b"entries", + "machine_configuration", + b"machine_configuration", + "run_configuration", + b"run_configuration", + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "benchmark_type", + b"benchmark_type", + "build_configuration", + b"build_configuration", + "commit_id", + b"commit_id", + "entries", + b"entries", + "machine_configuration", + b"machine_configuration", + "name", + b"name", + "run_configuration", + b"run_configuration", + "run_mode", + b"run_mode", + "run_time", + b"run_time", + "start_time", + b"start_time", + "target", + b"target", + "tf_version", + b"tf_version", + ], + ) -> None: ... global___TestResults = TestResults diff --git a/stubs/tensorflow/tensorflow/compiler/xla/xla_data_pb2.pyi b/stubs/tensorflow/tensorflow/compiler/xla/xla_data_pb2.pyi index 572077b07eaa..de86f1c8f255 100644 --- a/stubs/tensorflow/tensorflow/compiler/xla/xla_data_pb2.pyi +++ b/stubs/tensorflow/tensorflow/compiler/xla/xla_data_pb2.pyi @@ -38,7 +38,9 @@ class _PrimitiveType: ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType -class _PrimitiveTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_PrimitiveType.ValueType], builtins.type): +class _PrimitiveTypeEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_PrimitiveType.ValueType], builtins.type +): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor PRIMITIVE_TYPE_INVALID: _PrimitiveType.ValueType # 0 """Invalid primitive type to serve as default.""" @@ -251,7 +253,9 @@ class _DimLevelType: ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType -class _DimLevelTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_DimLevelType.ValueType], builtins.type): +class _DimLevelTypeEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_DimLevelType.ValueType], builtins.type +): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor DIM_DENSE: _DimLevelType.ValueType # 0 """The corresponding dimension is Dense, every entry is stored.""" @@ -294,7 +298,9 @@ class _ProfileType: ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType -class _ProfileTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_ProfileType.ValueType], builtins.type): +class _ProfileTypeEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_ProfileType.ValueType], builtins.type +): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor INVALID: _ProfileType.ValueType # 0 WINDOW: _ProfileType.ValueType # 1 @@ -314,7 +320,9 @@ class _ProfileSource: ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType -class _ProfileSourceEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_ProfileSource.ValueType], builtins.type): +class _ProfileSourceEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_ProfileSource.ValueType], builtins.type +): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor PROFILE_SOURCE_UNKNOWN_SOURCE: _ProfileSource.ValueType # 0 PROFILE_SOURCE_EMBEDDED: _ProfileSource.ValueType # 1 @@ -332,7 +340,9 @@ class _CompilationEvent: ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType -class _CompilationEventEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_CompilationEvent.ValueType], builtins.type): +class _CompilationEventEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_CompilationEvent.ValueType], builtins.type +): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor COMPILATION_EVENT_UNKNOWN_EVENT: _CompilationEvent.ValueType # 0 COMPILATION_EVENT_FIRST_COMPILATION: _CompilationEvent.ValueType # 1 @@ -350,7 +360,9 @@ class _PaddingType: ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType -class _PaddingTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_PaddingType.ValueType], builtins.type): +class _PaddingTypeEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_PaddingType.ValueType], builtins.type +): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor PADDING_INVALID: _PaddingType.ValueType # 0 PADDING_VALID: _PaddingType.ValueType # 1 @@ -398,7 +410,9 @@ class _SparsityType: ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType -class _SparsityTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_SparsityType.ValueType], builtins.type): +class _SparsityTypeEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_SparsityType.ValueType], builtins.type +): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor SPARSITY_INVALID: _SparsityType.ValueType # 0 SPARSITY_STRUCTURED_N_M: _SparsityType.ValueType # 1 @@ -415,7 +429,9 @@ class _RandomDistribution: ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType -class _RandomDistributionEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_RandomDistribution.ValueType], builtins.type): +class _RandomDistributionEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_RandomDistribution.ValueType], builtins.type +): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor RNG_INVALID: _RandomDistribution.ValueType # 0 RNG_UNIFORM: _RandomDistribution.ValueType # 1 @@ -444,7 +460,9 @@ class _RandomAlgorithm: ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType -class _RandomAlgorithmEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_RandomAlgorithm.ValueType], builtins.type): +class _RandomAlgorithmEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_RandomAlgorithm.ValueType], builtins.type +): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor RNG_DEFAULT: _RandomAlgorithm.ValueType # 0 """Backend dependent default algorithm.""" @@ -493,17 +511,27 @@ class PaddingConfig(google.protobuf.message.Message): edge_padding_high: builtins.int | None = ..., interior_padding: builtins.int | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["edge_padding_high", b"edge_padding_high", "edge_padding_low", b"edge_padding_low", "interior_padding", b"interior_padding"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "edge_padding_high", + b"edge_padding_high", + "edge_padding_low", + b"edge_padding_low", + "interior_padding", + b"interior_padding", + ], + ) -> None: ... DIMENSIONS_FIELD_NUMBER: builtins.int @property - def dimensions(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___PaddingConfig.PaddingConfigDimension]: + def dimensions( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___PaddingConfig.PaddingConfigDimension]: """The padding configuration for all dimensions.""" def __init__( - self, - *, - dimensions: collections.abc.Iterable[global___PaddingConfig.PaddingConfigDimension] | None = ..., + self, *, dimensions: collections.abc.Iterable[global___PaddingConfig.PaddingConfigDimension] | None = ... ) -> None: ... def ClearField(self, field_name: typing.Literal["dimensions", b"dimensions"]) -> None: ... @@ -527,11 +555,7 @@ class TileProto(google.protobuf.message.Message): tiled. """ - def __init__( - self, - *, - dimensions: collections.abc.Iterable[builtins.int] | None = ..., - ) -> None: ... + def __init__(self, *, dimensions: collections.abc.Iterable[builtins.int] | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["dimensions", b"dimensions"]) -> None: ... global___TileProto = TileProto @@ -554,10 +578,7 @@ class SplitConfigProto(google.protobuf.message.Message): """ def __init__( - self, - *, - dimension: builtins.int | None = ..., - split_indices: collections.abc.Iterable[builtins.int] | None = ..., + self, *, dimension: builtins.int | None = ..., split_indices: collections.abc.Iterable[builtins.int] | None = ... ) -> None: ... def ClearField(self, field_name: typing.Literal["dimension", b"dimension", "split_indices", b"split_indices"]) -> None: ... @@ -624,7 +645,9 @@ class LayoutProto(google.protobuf.message.Message): dynamic shape, e.g. a result of SliceToDynamic. """ @property - def dim_level_types(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[global___DimLevelType.ValueType]: + def dim_level_types( + self, + ) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[global___DimLevelType.ValueType]: """The dimension level type list for this array, specifying the way in which each array dimension is represented in memory. If this list is empty, the array is assumed to be dense. @@ -688,7 +711,37 @@ class LayoutProto(google.protobuf.message.Message): split_configs: collections.abc.Iterable[global___SplitConfigProto] | None = ..., ) -> None: ... def HasField(self, field_name: typing.Literal["physical_shape", b"physical_shape"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["dim_level_types", b"dim_level_types", "dim_ordered", b"dim_ordered", "dim_unique", b"dim_unique", "dynamic_shape_metadata_prefix_bytes", b"dynamic_shape_metadata_prefix_bytes", "element_size_in_bits", b"element_size_in_bits", "index_primitive_type", b"index_primitive_type", "memory_space", b"memory_space", "minor_to_major", b"minor_to_major", "physical_shape", b"physical_shape", "pointer_primitive_type", b"pointer_primitive_type", "split_configs", b"split_configs", "tail_padding_alignment_in_elements", b"tail_padding_alignment_in_elements", "tiles", b"tiles"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "dim_level_types", + b"dim_level_types", + "dim_ordered", + b"dim_ordered", + "dim_unique", + b"dim_unique", + "dynamic_shape_metadata_prefix_bytes", + b"dynamic_shape_metadata_prefix_bytes", + "element_size_in_bits", + b"element_size_in_bits", + "index_primitive_type", + b"index_primitive_type", + "memory_space", + b"memory_space", + "minor_to_major", + b"minor_to_major", + "physical_shape", + b"physical_shape", + "pointer_primitive_type", + b"pointer_primitive_type", + "split_configs", + b"split_configs", + "tail_padding_alignment_in_elements", + b"tail_padding_alignment_in_elements", + "tiles", + b"tiles", + ], + ) -> None: ... global___LayoutProto = LayoutProto @@ -752,7 +805,21 @@ class ShapeProto(google.protobuf.message.Message): is_dynamic_dimension: collections.abc.Iterable[builtins.bool] | None = ..., ) -> None: ... def HasField(self, field_name: typing.Literal["layout", b"layout"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["dimensions", b"dimensions", "element_type", b"element_type", "is_dynamic_dimension", b"is_dynamic_dimension", "layout", b"layout", "tuple_shapes", b"tuple_shapes"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "dimensions", + b"dimensions", + "element_type", + b"element_type", + "is_dynamic_dimension", + b"is_dynamic_dimension", + "layout", + b"layout", + "tuple_shapes", + b"tuple_shapes", + ], + ) -> None: ... global___ShapeProto = ShapeProto @@ -781,7 +848,9 @@ class ProgramShapeProto(google.protobuf.message.Message): parameter_names: collections.abc.Iterable[builtins.str] | None = ..., ) -> None: ... def HasField(self, field_name: typing.Literal["result", b"result"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["parameter_names", b"parameter_names", "parameters", b"parameters", "result", b"result"]) -> None: ... + def ClearField( + self, field_name: typing.Literal["parameter_names", b"parameter_names", "parameters", b"parameters", "result", b"result"] + ) -> None: ... global___ProgramShapeProto = ProgramShapeProto @@ -797,13 +866,10 @@ class ComputationStats(google.protobuf.message.Message): """The number of floating point operations in the computation.""" transcendental_count: builtins.float """The number of transcendental operations (e.g., exp) in the computation.""" - def __init__( - self, - *, - flop_count: builtins.float | None = ..., - transcendental_count: builtins.float | None = ..., + def __init__(self, *, flop_count: builtins.float | None = ..., transcendental_count: builtins.float | None = ...) -> None: ... + def ClearField( + self, field_name: typing.Literal["flop_count", b"flop_count", "transcendental_count", b"transcendental_count"] ) -> None: ... - def ClearField(self, field_name: typing.Literal["flop_count", b"flop_count", "transcendental_count", b"transcendental_count"]) -> None: ... global___ComputationStats = ComputationStats @@ -836,7 +902,9 @@ class OpMetadata(google.protobuf.message.Message): compilation_event: global___CompilationEvent.ValueType """The compilation event that triggered the use of the profiles.""" @property - def profile_type(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[global___ProfileType.ValueType]: + def profile_type( + self, + ) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[global___ProfileType.ValueType]: """The type of optimization profiles that this operation contains.""" def __init__( @@ -847,7 +915,19 @@ class OpMetadata(google.protobuf.message.Message): profile_source: global___ProfileSource.ValueType | None = ..., compilation_event: global___CompilationEvent.ValueType | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["compilation_event", b"compilation_event", "profile_source", b"profile_source", "profile_type", b"profile_type", "relative_speedup", b"relative_speedup"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "compilation_event", + b"compilation_event", + "profile_source", + b"profile_source", + "profile_type", + b"profile_type", + "relative_speedup", + b"relative_speedup", + ], + ) -> None: ... OP_TYPE_FIELD_NUMBER: builtins.int OP_NAME_FIELD_NUMBER: builtins.int @@ -929,7 +1009,35 @@ class OpMetadata(google.protobuf.message.Message): scheduling_name: builtins.str | None = ..., ) -> None: ... def HasField(self, field_name: typing.Literal["profile_info", b"profile_info"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["deduplicated_name", b"deduplicated_name", "op_name", b"op_name", "op_type", b"op_type", "preserve_layout", b"preserve_layout", "profile_info", b"profile_info", "profile_type", b"profile_type", "scheduling_name", b"scheduling_name", "size_of_generated_code_in_bytes", b"size_of_generated_code_in_bytes", "size_of_memory_working_set_in_bytes", b"size_of_memory_working_set_in_bytes", "source_file", b"source_file", "source_line", b"source_line", "stack_frame_id", b"stack_frame_id"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "deduplicated_name", + b"deduplicated_name", + "op_name", + b"op_name", + "op_type", + b"op_type", + "preserve_layout", + b"preserve_layout", + "profile_info", + b"profile_info", + "profile_type", + b"profile_type", + "scheduling_name", + b"scheduling_name", + "size_of_generated_code_in_bytes", + b"size_of_generated_code_in_bytes", + "size_of_memory_working_set_in_bytes", + b"size_of_memory_working_set_in_bytes", + "source_file", + b"source_file", + "source_line", + b"source_line", + "stack_frame_id", + b"stack_frame_id", + ], + ) -> None: ... global___OpMetadata = OpMetadata @@ -989,7 +1097,27 @@ class ExecutionProfile(google.protobuf.message.Message): profile_cache_hit: builtins.bool | None = ..., warmup_run_executed: builtins.bool | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["compilation_cache_hit", b"compilation_cache_hit", "compile_time_ms", b"compile_time_ms", "compute_and_transfer_time_ns", b"compute_and_transfer_time_ns", "compute_cycle_count", b"compute_cycle_count", "compute_time_ns", b"compute_time_ns", "executable_size_in_bytes", b"executable_size_in_bytes", "profile_cache_hit", b"profile_cache_hit", "warmup_run_executed", b"warmup_run_executed"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "compilation_cache_hit", + b"compilation_cache_hit", + "compile_time_ms", + b"compile_time_ms", + "compute_and_transfer_time_ns", + b"compute_and_transfer_time_ns", + "compute_cycle_count", + b"compute_cycle_count", + "compute_time_ns", + b"compute_time_ns", + "executable_size_in_bytes", + b"executable_size_in_bytes", + "profile_cache_hit", + b"profile_cache_hit", + "warmup_run_executed", + b"warmup_run_executed", + ], + ) -> None: ... global___ExecutionProfile = ExecutionProfile @@ -1003,11 +1131,7 @@ class ExecutionHandle(google.protobuf.message.Message): HANDLE_FIELD_NUMBER: builtins.int handle: builtins.int - def __init__( - self, - *, - handle: builtins.int | None = ..., - ) -> None: ... + def __init__(self, *, handle: builtins.int | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["handle", b"handle"]) -> None: ... global___ExecutionHandle = ExecutionHandle @@ -1023,11 +1147,7 @@ class GlobalDataHandle(google.protobuf.message.Message): HANDLE_FIELD_NUMBER: builtins.int handle: builtins.int - def __init__( - self, - *, - handle: builtins.int | None = ..., - ) -> None: ... + def __init__(self, *, handle: builtins.int | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["handle", b"handle"]) -> None: ... global___GlobalDataHandle = GlobalDataHandle @@ -1048,12 +1168,7 @@ class DeviceHandle(google.protobuf.message.Message): """The number of model-parallel virtual devices that communicate via XLA Send/Recv instructions. """ - def __init__( - self, - *, - handle: builtins.int | None = ..., - device_count: builtins.int | None = ..., - ) -> None: ... + def __init__(self, *, handle: builtins.int | None = ..., device_count: builtins.int | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["device_count", b"device_count", "handle", b"handle"]) -> None: ... global___DeviceHandle = DeviceHandle @@ -1071,7 +1186,9 @@ class ChannelHandle(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _ChannelTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[ChannelHandle._ChannelType.ValueType], builtins.type): + class _ChannelTypeEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[ChannelHandle._ChannelType.ValueType], builtins.type + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor CHANNEL_TYPE_INVALID: ChannelHandle._ChannelType.ValueType # 0 """Invalid primitive type to serve as default.""" @@ -1105,10 +1222,7 @@ class ChannelHandle(google.protobuf.message.Message): handle: builtins.int type: global___ChannelHandle.ChannelType.ValueType def __init__( - self, - *, - handle: builtins.int | None = ..., - type: global___ChannelHandle.ChannelType.ValueType | None = ..., + self, *, handle: builtins.int | None = ..., type: global___ChannelHandle.ChannelType.ValueType | None = ... ) -> None: ... def ClearField(self, field_name: typing.Literal["handle", b"handle", "type", b"type"]) -> None: ... @@ -1134,11 +1248,7 @@ class DeviceAssignmentProto(google.protobuf.message.Message): REPLICA_DEVICE_IDS_FIELD_NUMBER: builtins.int @property def replica_device_ids(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.int]: ... - def __init__( - self, - *, - replica_device_ids: collections.abc.Iterable[builtins.int] | None = ..., - ) -> None: ... + def __init__(self, *, replica_device_ids: collections.abc.Iterable[builtins.int] | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["replica_device_ids", b"replica_device_ids"]) -> None: ... REPLICA_COUNT_FIELD_NUMBER: builtins.int @@ -1147,7 +1257,11 @@ class DeviceAssignmentProto(google.protobuf.message.Message): replica_count: builtins.int computation_count: builtins.int @property - def computation_devices(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___DeviceAssignmentProto.ComputationDevice]: ... + def computation_devices( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[ + global___DeviceAssignmentProto.ComputationDevice + ]: ... def __init__( self, *, @@ -1155,7 +1269,17 @@ class DeviceAssignmentProto(google.protobuf.message.Message): computation_count: builtins.int | None = ..., computation_devices: collections.abc.Iterable[global___DeviceAssignmentProto.ComputationDevice] | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["computation_count", b"computation_count", "computation_devices", b"computation_devices", "replica_count", b"replica_count"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "computation_count", + b"computation_count", + "computation_devices", + b"computation_devices", + "replica_count", + b"replica_count", + ], + ) -> None: ... global___DeviceAssignmentProto = DeviceAssignmentProto @@ -1276,7 +1400,65 @@ class LiteralProto(google.protobuf.message.Message): sparse_indices: collections.abc.Iterable[builtins.int] | None = ..., ) -> None: ... def HasField(self, field_name: typing.Literal["shape", b"shape"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["bf16s", b"bf16s", "c128s", b"c128s", "c64s", b"c64s", "f16s", b"f16s", "f32s", b"f32s", "f64s", b"f64s", "f8e4m3b11fnuzs", b"f8e4m3b11fnuzs", "f8e4m3fns", b"f8e4m3fns", "f8e4m3fnuzs", b"f8e4m3fnuzs", "f8e5m2fnuzs", b"f8e5m2fnuzs", "f8e5m2s", b"f8e5m2s", "preds", b"preds", "s16s", b"s16s", "s2s", b"s2s", "s32s", b"s32s", "s4s", b"s4s", "s64s", b"s64s", "s8s", b"s8s", "shape", b"shape", "sparse_indices", b"sparse_indices", "tuple_literals", b"tuple_literals", "u16s", b"u16s", "u2s", b"u2s", "u32s", b"u32s", "u4s", b"u4s", "u64s", b"u64s", "u8s", b"u8s"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "bf16s", + b"bf16s", + "c128s", + b"c128s", + "c64s", + b"c64s", + "f16s", + b"f16s", + "f32s", + b"f32s", + "f64s", + b"f64s", + "f8e4m3b11fnuzs", + b"f8e4m3b11fnuzs", + "f8e4m3fns", + b"f8e4m3fns", + "f8e4m3fnuzs", + b"f8e4m3fnuzs", + "f8e5m2fnuzs", + b"f8e5m2fnuzs", + "f8e5m2s", + b"f8e5m2s", + "preds", + b"preds", + "s16s", + b"s16s", + "s2s", + b"s2s", + "s32s", + b"s32s", + "s4s", + b"s4s", + "s64s", + b"s64s", + "s8s", + b"s8s", + "shape", + b"shape", + "sparse_indices", + b"sparse_indices", + "tuple_literals", + b"tuple_literals", + "u16s", + b"u16s", + "u2s", + b"u2s", + "u32s", + b"u32s", + "u4s", + b"u4s", + "u64s", + b"u64s", + "u8s", + b"u8s", + ], + ) -> None: ... global___LiteralProto = LiteralProto @@ -1342,7 +1524,25 @@ class WindowDimension(google.protobuf.message.Message): base_dilation: builtins.int | None = ..., window_reversal: builtins.bool | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["base_dilation", b"base_dilation", "padding_high", b"padding_high", "padding_low", b"padding_low", "size", b"size", "stride", b"stride", "window_dilation", b"window_dilation", "window_reversal", b"window_reversal"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "base_dilation", + b"base_dilation", + "padding_high", + b"padding_high", + "padding_low", + b"padding_low", + "size", + b"size", + "stride", + b"stride", + "window_dilation", + b"window_dilation", + "window_reversal", + b"window_reversal", + ], + ) -> None: ... global___WindowDimension = WindowDimension @@ -1360,11 +1560,7 @@ class Window(google.protobuf.message.Message): DIMENSIONS_FIELD_NUMBER: builtins.int @property def dimensions(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___WindowDimension]: ... - def __init__( - self, - *, - dimensions: collections.abc.Iterable[global___WindowDimension] | None = ..., - ) -> None: ... + def __init__(self, *, dimensions: collections.abc.Iterable[global___WindowDimension] | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["dimensions", b"dimensions"]) -> None: ... global___Window = Window @@ -1391,7 +1587,7 @@ class GatherDimensionNumbers(google.protobuf.message.Message): """ @property def offset_dims(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.int]: - """"Window indices" is a term for a set of indices that index into the + """ "Window indices" is a term for a set of indices that index into the interior of a dynamic-slice from the input tensor, the starting indices for which were computed from output_gather_dims (see the operation semantic for how this is defined) and the start_indices tensor. @@ -1435,7 +1631,23 @@ class GatherDimensionNumbers(google.protobuf.message.Message): operand_batching_dims: collections.abc.Iterable[builtins.int] | None = ..., start_indices_batching_dims: collections.abc.Iterable[builtins.int] | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["collapsed_slice_dims", b"collapsed_slice_dims", "index_vector_dim", b"index_vector_dim", "offset_dims", b"offset_dims", "operand_batching_dims", b"operand_batching_dims", "start_index_map", b"start_index_map", "start_indices_batching_dims", b"start_indices_batching_dims"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "collapsed_slice_dims", + b"collapsed_slice_dims", + "index_vector_dim", + b"index_vector_dim", + "offset_dims", + b"offset_dims", + "operand_batching_dims", + b"operand_batching_dims", + "start_index_map", + b"start_index_map", + "start_indices_batching_dims", + b"start_indices_batching_dims", + ], + ) -> None: ... global___GatherDimensionNumbers = GatherDimensionNumbers @@ -1484,7 +1696,23 @@ class ScatterDimensionNumbers(google.protobuf.message.Message): input_batching_dims: collections.abc.Iterable[builtins.int] | None = ..., scatter_indices_batching_dims: collections.abc.Iterable[builtins.int] | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["index_vector_dim", b"index_vector_dim", "input_batching_dims", b"input_batching_dims", "inserted_window_dims", b"inserted_window_dims", "scatter_dims_to_operand_dims", b"scatter_dims_to_operand_dims", "scatter_indices_batching_dims", b"scatter_indices_batching_dims", "update_window_dims", b"update_window_dims"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "index_vector_dim", + b"index_vector_dim", + "input_batching_dims", + b"input_batching_dims", + "inserted_window_dims", + b"inserted_window_dims", + "scatter_dims_to_operand_dims", + b"scatter_dims_to_operand_dims", + "scatter_indices_batching_dims", + b"scatter_indices_batching_dims", + "update_window_dims", + b"update_window_dims", + ], + ) -> None: ... global___ScatterDimensionNumbers = ScatterDimensionNumbers @@ -1549,7 +1777,29 @@ class ConvolutionDimensionNumbers(google.protobuf.message.Message): output_feature_dimension: builtins.int | None = ..., output_spatial_dimensions: collections.abc.Iterable[builtins.int] | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["input_batch_dimension", b"input_batch_dimension", "input_feature_dimension", b"input_feature_dimension", "input_spatial_dimensions", b"input_spatial_dimensions", "kernel_input_feature_dimension", b"kernel_input_feature_dimension", "kernel_output_feature_dimension", b"kernel_output_feature_dimension", "kernel_spatial_dimensions", b"kernel_spatial_dimensions", "output_batch_dimension", b"output_batch_dimension", "output_feature_dimension", b"output_feature_dimension", "output_spatial_dimensions", b"output_spatial_dimensions"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "input_batch_dimension", + b"input_batch_dimension", + "input_feature_dimension", + b"input_feature_dimension", + "input_spatial_dimensions", + b"input_spatial_dimensions", + "kernel_input_feature_dimension", + b"kernel_input_feature_dimension", + "kernel_output_feature_dimension", + b"kernel_output_feature_dimension", + "kernel_spatial_dimensions", + b"kernel_spatial_dimensions", + "output_batch_dimension", + b"output_batch_dimension", + "output_feature_dimension", + b"output_feature_dimension", + "output_spatial_dimensions", + b"output_spatial_dimensions", + ], + ) -> None: ... global___ConvolutionDimensionNumbers = ConvolutionDimensionNumbers @@ -1585,7 +1835,19 @@ class DotDimensionNumbers(google.protobuf.message.Message): lhs_batch_dimensions: collections.abc.Iterable[builtins.int] | None = ..., rhs_batch_dimensions: collections.abc.Iterable[builtins.int] | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["lhs_batch_dimensions", b"lhs_batch_dimensions", "lhs_contracting_dimensions", b"lhs_contracting_dimensions", "rhs_batch_dimensions", b"rhs_batch_dimensions", "rhs_contracting_dimensions", b"rhs_contracting_dimensions"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "lhs_batch_dimensions", + b"lhs_batch_dimensions", + "lhs_contracting_dimensions", + b"lhs_contracting_dimensions", + "rhs_batch_dimensions", + b"rhs_batch_dimensions", + "rhs_contracting_dimensions", + b"rhs_contracting_dimensions", + ], + ) -> None: ... global___DotDimensionNumbers = DotDimensionNumbers @@ -1623,7 +1885,9 @@ class SparsityDescriptor(google.protobuf.message.Message): n: builtins.int | None = ..., m: builtins.int | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["dimension", b"dimension", "index", b"index", "m", b"m", "n", b"n", "type", b"type"]) -> None: ... + def ClearField( + self, field_name: typing.Literal["dimension", b"dimension", "index", b"index", "m", b"m", "n", b"n", "type", b"type"] + ) -> None: ... global___SparsityDescriptor = SparsityDescriptor @@ -1635,7 +1899,9 @@ class TriangularSolveOptions(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _TransposeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[TriangularSolveOptions._Transpose.ValueType], builtins.type): + class _TransposeEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[TriangularSolveOptions._Transpose.ValueType], builtins.type + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor TRANSPOSE_INVALID: TriangularSolveOptions._Transpose.ValueType # 0 NO_TRANSPOSE: TriangularSolveOptions._Transpose.ValueType # 1 @@ -1675,7 +1941,12 @@ class TriangularSolveOptions(google.protobuf.message.Message): unit_diagonal: builtins.bool | None = ..., transpose_a: global___TriangularSolveOptions.Transpose.ValueType | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["left_side", b"left_side", "lower", b"lower", "transpose_a", b"transpose_a", "unit_diagonal", b"unit_diagonal"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "left_side", b"left_side", "lower", b"lower", "transpose_a", b"transpose_a", "unit_diagonal", b"unit_diagonal" + ], + ) -> None: ... global___TriangularSolveOptions = TriangularSolveOptions @@ -1688,11 +1959,7 @@ class CholeskyOptions(google.protobuf.message.Message): """If true, uses the lower triangle of `a`. If false, uses the upper triangle of `a`. """ - def __init__( - self, - *, - lower: builtins.bool | None = ..., - ) -> None: ... + def __init__(self, *, lower: builtins.bool | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["lower", b"lower"]) -> None: ... global___CholeskyOptions = CholeskyOptions @@ -1705,11 +1972,7 @@ class SortOptions(google.protobuf.message.Message): DESCENDING_FIELD_NUMBER: builtins.int descending: builtins.bool - def __init__( - self, - *, - descending: builtins.bool | None = ..., - ) -> None: ... + def __init__(self, *, descending: builtins.bool | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["descending", b"descending"]) -> None: ... global___SortOptions = SortOptions @@ -1730,22 +1993,13 @@ class FrontendAttributes(google.protobuf.message.Message): VALUE_FIELD_NUMBER: builtins.int key: builtins.str value: builtins.str - def __init__( - self, - *, - key: builtins.str | None = ..., - value: builtins.str | None = ..., - ) -> None: ... + def __init__(self, *, key: builtins.str | None = ..., value: builtins.str | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["key", b"key", "value", b"value"]) -> None: ... MAP_FIELD_NUMBER: builtins.int @property def map(self) -> google.protobuf.internal.containers.ScalarMap[builtins.str, builtins.str]: ... - def __init__( - self, - *, - map: collections.abc.Mapping[builtins.str, builtins.str] | None = ..., - ) -> None: ... + def __init__(self, *, map: collections.abc.Mapping[builtins.str, builtins.str] | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["map", b"map"]) -> None: ... global___FrontendAttributes = FrontendAttributes @@ -1764,12 +2018,7 @@ class Statistic(google.protobuf.message.Message): """Must be within a range of [0, 100], in order for the graph dumper to properly render the statistic onto the graph. """ - def __init__( - self, - *, - stat_name: builtins.str | None = ..., - stat_val: builtins.float | None = ..., - ) -> None: ... + def __init__(self, *, stat_name: builtins.str | None = ..., stat_val: builtins.float | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["stat_name", b"stat_name", "stat_val", b"stat_val"]) -> None: ... global___Statistic = Statistic @@ -1794,7 +2043,9 @@ class StatisticsViz(google.protobuf.message.Message): stat_index_to_visualize: builtins.int | None = ..., statistics: collections.abc.Iterable[global___Statistic] | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["stat_index_to_visualize", b"stat_index_to_visualize", "statistics", b"statistics"]) -> None: ... + def ClearField( + self, field_name: typing.Literal["stat_index_to_visualize", b"stat_index_to_visualize", "statistics", b"statistics"] + ) -> None: ... global___StatisticsViz = StatisticsViz @@ -1808,7 +2059,9 @@ class OpSharding(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _TypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[OpSharding._Type.ValueType], builtins.type): + class _TypeEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[OpSharding._Type.ValueType], builtins.type + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor REPLICATED: OpSharding._Type.ValueType # 0 """This sharding is replicated across all devices (implies maximal, @@ -1853,7 +2106,9 @@ class OpSharding(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _ShardGroupTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[OpSharding._ShardGroupType.ValueType], builtins.type): + class _ShardGroupTypeEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[OpSharding._ShardGroupType.ValueType], builtins.type + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor AS: OpSharding._ShardGroupType.ValueType # 0 """This op will be sharded exactly the same as the other op. (hard @@ -1944,7 +2199,9 @@ class OpSharding(google.protobuf.message.Message): """ @property - def last_tile_dims(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[global___OpSharding.Type.ValueType]: + def last_tile_dims( + self, + ) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[global___OpSharding.Type.ValueType]: """This field is used to represented the sharding type of each subgroup. For example, sharding={devices=[2,2,2,2]0,1,2,...,15 last_tile_dims={ replicate, manual, unreduced}} means that each of the last 3 dimensions @@ -1983,7 +2240,37 @@ class OpSharding(google.protobuf.message.Message): shard_group_type: global___OpSharding.ShardGroupType.ValueType | None = ..., ) -> None: ... def HasField(self, field_name: typing.Literal["tile_shape", b"tile_shape"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["iota_reshape_dims", b"iota_reshape_dims", "iota_transpose_perm", b"iota_transpose_perm", "is_shard_group", b"is_shard_group", "last_tile_dims", b"last_tile_dims", "metadata", b"metadata", "replicate_on_last_tile_dim", b"replicate_on_last_tile_dim", "shard_group_id", b"shard_group_id", "shard_group_type", b"shard_group_type", "tile_assignment_devices", b"tile_assignment_devices", "tile_assignment_dimensions", b"tile_assignment_dimensions", "tile_shape", b"tile_shape", "tuple_shardings", b"tuple_shardings", "type", b"type"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "iota_reshape_dims", + b"iota_reshape_dims", + "iota_transpose_perm", + b"iota_transpose_perm", + "is_shard_group", + b"is_shard_group", + "last_tile_dims", + b"last_tile_dims", + "metadata", + b"metadata", + "replicate_on_last_tile_dim", + b"replicate_on_last_tile_dim", + "shard_group_id", + b"shard_group_id", + "shard_group_type", + b"shard_group_type", + "tile_assignment_devices", + b"tile_assignment_devices", + "tile_assignment_dimensions", + b"tile_assignment_dimensions", + "tile_shape", + b"tile_shape", + "tuple_shardings", + b"tuple_shardings", + "type", + b"type", + ], + ) -> None: ... global___OpSharding = OpSharding @@ -2002,11 +2289,7 @@ class ReplicaGroup(google.protobuf.message.Message): ids matters in some ops (e.g., all-to-all). """ - def __init__( - self, - *, - replica_ids: collections.abc.Iterable[builtins.int] | None = ..., - ) -> None: ... + def __init__(self, *, replica_ids: collections.abc.Iterable[builtins.int] | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["replica_ids", b"replica_ids"]) -> None: ... global___ReplicaGroup = ReplicaGroup @@ -2047,7 +2330,19 @@ class IotaReplicaGroupListProto(google.protobuf.message.Message): iota_reshape_dims: collections.abc.Iterable[builtins.int] | None = ..., iota_transpose_perm: collections.abc.Iterable[builtins.int] | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["iota_reshape_dims", b"iota_reshape_dims", "iota_transpose_perm", b"iota_transpose_perm", "num_devices_per_group", b"num_devices_per_group", "num_replica_groups", b"num_replica_groups"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "iota_reshape_dims", + b"iota_reshape_dims", + "iota_transpose_perm", + b"iota_transpose_perm", + "num_devices_per_group", + b"num_devices_per_group", + "num_replica_groups", + b"num_replica_groups", + ], + ) -> None: ... global___IotaReplicaGroupListProto = IotaReplicaGroupListProto @@ -2081,7 +2376,10 @@ class CollectiveDeviceListProto(google.protobuf.message.Message): iota_replica_group_list: global___IotaReplicaGroupListProto | None = ..., ) -> None: ... def HasField(self, field_name: typing.Literal["iota_replica_group_list", b"iota_replica_group_list"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["iota_replica_group_list", b"iota_replica_group_list", "replica_groups", b"replica_groups"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal["iota_replica_group_list", b"iota_replica_group_list", "replica_groups", b"replica_groups"], + ) -> None: ... global___CollectiveDeviceListProto = CollectiveDeviceListProto @@ -2095,12 +2393,7 @@ class SourceTarget(google.protobuf.message.Message): TARGET_FIELD_NUMBER: builtins.int source: builtins.int target: builtins.int - def __init__( - self, - *, - source: builtins.int | None = ..., - target: builtins.int | None = ..., - ) -> None: ... + def __init__(self, *, source: builtins.int | None = ..., target: builtins.int | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["source", b"source", "target", b"target"]) -> None: ... global___SourceTarget = SourceTarget @@ -2117,7 +2410,9 @@ class PrecisionConfig(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _PrecisionEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[PrecisionConfig._Precision.ValueType], builtins.type): + class _PrecisionEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[PrecisionConfig._Precision.ValueType], builtins.type + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor DEFAULT: PrecisionConfig._Precision.ValueType # 0 HIGH: PrecisionConfig._Precision.ValueType # 1 @@ -2136,7 +2431,9 @@ class PrecisionConfig(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _AlgorithmEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[PrecisionConfig._Algorithm.ValueType], builtins.type): + class _AlgorithmEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[PrecisionConfig._Algorithm.ValueType], builtins.type + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor ALG_UNSET: PrecisionConfig._Algorithm.ValueType # 0 """If the algorithm is `ALG_UNSET`, we will decide the algorithm based on @@ -2231,14 +2528,18 @@ class PrecisionConfig(google.protobuf.message.Message): will be ignored. """ @property - def operand_precision(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[global___PrecisionConfig.Precision.ValueType]: ... + def operand_precision( + self, + ) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[global___PrecisionConfig.Precision.ValueType]: ... def __init__( self, *, operand_precision: collections.abc.Iterable[global___PrecisionConfig.Precision.ValueType] | None = ..., algorithm: global___PrecisionConfig.Algorithm.ValueType | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["algorithm", b"algorithm", "operand_precision", b"operand_precision"]) -> None: ... + def ClearField( + self, field_name: typing.Literal["algorithm", b"algorithm", "operand_precision", b"operand_precision"] + ) -> None: ... global___PrecisionConfig = PrecisionConfig @@ -2261,11 +2562,7 @@ class ParameterReplication(google.protobuf.message.Message): the HLO instruction's shape. """ - def __init__( - self, - *, - replicated_at_leaf_buffers: collections.abc.Iterable[builtins.bool] | None = ..., - ) -> None: ... + def __init__(self, *, replicated_at_leaf_buffers: collections.abc.Iterable[builtins.bool] | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["replicated_at_leaf_buffers", b"replicated_at_leaf_buffers"]) -> None: ... global___ParameterReplication = ParameterReplication @@ -2290,11 +2587,7 @@ class WhileLoopBackendConfig(google.protobuf.message.Message): N_FIELD_NUMBER: builtins.int n: builtins.int - def __init__( - self, - *, - n: builtins.int | None = ..., - ) -> None: ... + def __init__(self, *, n: builtins.int | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["n", b"n"]) -> None: ... KNOWN_TRIP_COUNT_FIELD_NUMBER: builtins.int @@ -2304,11 +2597,7 @@ class WhileLoopBackendConfig(google.protobuf.message.Message): unknown-trip-count. """ - def __init__( - self, - *, - known_trip_count: global___WhileLoopBackendConfig.KnownTripCount | None = ..., - ) -> None: ... + def __init__(self, *, known_trip_count: global___WhileLoopBackendConfig.KnownTripCount | None = ...) -> None: ... def HasField(self, field_name: typing.Literal["known_trip_count", b"known_trip_count"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["known_trip_count", b"known_trip_count"]) -> None: ... @@ -2337,7 +2626,17 @@ class OutputOperandAliasing(google.protobuf.message.Message): operand_index: builtins.int | None = ..., operand_shape_index: collections.abc.Iterable[builtins.int] | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["operand_index", b"operand_index", "operand_shape_index", b"operand_shape_index", "output_shape_index", b"output_shape_index"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "operand_index", + b"operand_index", + "operand_shape_index", + b"operand_shape_index", + "output_shape_index", + b"output_shape_index", + ], + ) -> None: ... global___OutputOperandAliasing = OutputOperandAliasing @@ -2360,7 +2659,12 @@ class OriginalArrayProto(google.protobuf.message.Message): instruction_name: builtins.str | None = ..., shape_index: collections.abc.Iterable[builtins.int] | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["instruction_name", b"instruction_name", "leaf_shape_index", b"leaf_shape_index", "shape_index", b"shape_index"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "instruction_name", b"instruction_name", "leaf_shape_index", b"leaf_shape_index", "shape_index", b"shape_index" + ], + ) -> None: ... global___OriginalArrayProto = OriginalArrayProto @@ -2371,11 +2675,7 @@ class OriginalValueProto(google.protobuf.message.Message): LEAVES_FIELD_NUMBER: builtins.int @property def leaves(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___OriginalArrayProto]: ... - def __init__( - self, - *, - leaves: collections.abc.Iterable[global___OriginalArrayProto] | None = ..., - ) -> None: ... + def __init__(self, *, leaves: collections.abc.Iterable[global___OriginalArrayProto] | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["leaves", b"leaves"]) -> None: ... global___OriginalValueProto = OriginalValueProto diff --git a/stubs/tensorflow/tensorflow/compiler/xla/xla_pb2.pyi b/stubs/tensorflow/tensorflow/compiler/xla/xla_pb2.pyi index 086a41fc6528..17820b9e12c0 100644 --- a/stubs/tensorflow/tensorflow/compiler/xla/xla_pb2.pyi +++ b/stubs/tensorflow/tensorflow/compiler/xla/xla_pb2.pyi @@ -45,12 +45,10 @@ class CompilationEnvironmentsProto(google.protobuf.message.Message): ENVIRONMENTS_FIELD_NUMBER: builtins.int @property - def environments(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[google.protobuf.any_pb2.Any]: ... - def __init__( + def environments( self, - *, - environments: collections.abc.Iterable[google.protobuf.any_pb2.Any] | None = ..., - ) -> None: ... + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[google.protobuf.any_pb2.Any]: ... + def __init__(self, *, environments: collections.abc.Iterable[google.protobuf.any_pb2.Any] | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["environments", b"environments"]) -> None: ... global___CompilationEnvironmentsProto = CompilationEnvironmentsProto @@ -79,7 +77,9 @@ class DebugOptions(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _ShapeChecksEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[DebugOptions._ShapeChecks.ValueType], builtins.type): + class _ShapeChecksEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[DebugOptions._ShapeChecks.ValueType], builtins.type + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor IGNORE: DebugOptions._ShapeChecks.ValueType # 0 """Do not insert any shape checks for dynamically shaped operations; output @@ -112,7 +112,9 @@ class DebugOptions(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _StepMarkerLocationEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[DebugOptions._StepMarkerLocation.ValueType], builtins.type): + class _StepMarkerLocationEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[DebugOptions._StepMarkerLocation.ValueType], builtins.type + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor STEP_MARK_AT_ENTRY: DebugOptions._StepMarkerLocation.ValueType # 0 """Generate a step marker at the program entry. This handles the case where @@ -153,7 +155,9 @@ class DebugOptions(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _CollectiveOpTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[DebugOptions._CollectiveOpType.ValueType], builtins.type): + class _CollectiveOpTypeEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[DebugOptions._CollectiveOpType.ValueType], builtins.type + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor NOOP: DebugOptions._CollectiveOpType.ValueType # 0 ALLREDUCE: DebugOptions._CollectiveOpType.ValueType # 1 @@ -180,7 +184,9 @@ class DebugOptions(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _CommandBufferCmdTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[DebugOptions._CommandBufferCmdType.ValueType], builtins.type): + class _CommandBufferCmdTypeEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[DebugOptions._CommandBufferCmdType.ValueType], builtins.type + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor INVALID: DebugOptions._CommandBufferCmdType.ValueType # 0 FUSION: DebugOptions._CommandBufferCmdType.ValueType # 1 @@ -211,7 +217,9 @@ class DebugOptions(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _PartitioningAlgorithmEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[DebugOptions._PartitioningAlgorithm.ValueType], builtins.type): + class _PartitioningAlgorithmEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[DebugOptions._PartitioningAlgorithm.ValueType], builtins.type + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor PARTITIONING_ALGORITHM_NOOP: DebugOptions._PartitioningAlgorithm.ValueType # 0 PARTITIONING_ALGORITHM_EXP0: DebugOptions._PartitioningAlgorithm.ValueType # 1 @@ -228,7 +236,9 @@ class DebugOptions(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _WhileLoopUnrollingEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[DebugOptions._WhileLoopUnrolling.ValueType], builtins.type): + class _WhileLoopUnrollingEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[DebugOptions._WhileLoopUnrolling.ValueType], builtins.type + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor WHILE_LOOP_UNROLLING_NO_UNROLL: DebugOptions._WhileLoopUnrolling.ValueType # 0 WHILE_LOOP_UNROLLING_DOUBLE_BUFFER: DebugOptions._WhileLoopUnrolling.ValueType # 1 @@ -251,7 +261,9 @@ class DebugOptions(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _AutotuneCacheModeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[DebugOptions._AutotuneCacheMode.ValueType], builtins.type): + class _AutotuneCacheModeEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[DebugOptions._AutotuneCacheMode.ValueType], builtins.type + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor AUTOTUNE_CACHE_MODE_UNSPECIFIED: DebugOptions._AutotuneCacheMode.ValueType # 0 AUTOTUNE_CACHE_MODE_UPDATE: DebugOptions._AutotuneCacheMode.ValueType # 1 @@ -282,12 +294,7 @@ class DebugOptions(google.protobuf.message.Message): VALUE_FIELD_NUMBER: builtins.int key: builtins.str value: builtins.str - def __init__( - self, - *, - key: builtins.str | None = ..., - value: builtins.str | None = ..., - ) -> None: ... + def __init__(self, *, key: builtins.str | None = ..., value: builtins.str | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["key", b"key", "value", b"value"]) -> None: ... XLA_CPU_ENABLE_CONCURRENCY_OPTIMIZED_SCHEDULER_FIELD_NUMBER: builtins.int @@ -1170,13 +1177,19 @@ class DebugOptions(google.protobuf.message.Message): """Paths to files with LLVM code.""" @property - def xla_gpu_disable_async_collectives(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[global___DebugOptions.CollectiveOpType.ValueType]: ... + def xla_gpu_disable_async_collectives( + self, + ) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[global___DebugOptions.CollectiveOpType.ValueType]: ... @property - def xla_gpu_enable_command_buffer(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[global___DebugOptions.CommandBufferCmdType.ValueType]: + def xla_gpu_enable_command_buffer( + self, + ) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[global___DebugOptions.CommandBufferCmdType.ValueType]: """Determine the types of commands that are recorded into command buffers.""" @property - def legacy_command_buffer_custom_call_targets(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]: + def legacy_command_buffer_custom_call_targets( + self, + ) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]: """Custom call targets with legacy registry API (non FFI API), that support recording to command buffer custom command, i.e., custom call target supports cuda-graph capturing for CUDA devices. @@ -1271,7 +1284,9 @@ class DebugOptions(google.protobuf.message.Message): xla_gpu_enable_llvm_module_compilation_parallelism: builtins.bool | None = ..., xla_gpu_deterministic_ops: builtins.bool | None = ..., xla_gpu_llvm_ir_file: collections.abc.Iterable[builtins.str] | None = ..., - xla_gpu_disable_async_collectives: collections.abc.Iterable[global___DebugOptions.CollectiveOpType.ValueType] | None = ..., + xla_gpu_disable_async_collectives: ( + collections.abc.Iterable[global___DebugOptions.CollectiveOpType.ValueType] | None + ) = ..., xla_gpu_all_reduce_combine_threshold_bytes: builtins.int | None = ..., xla_gpu_all_gather_combine_threshold_bytes: builtins.int | None = ..., xla_gpu_reduce_scatter_combine_threshold_bytes: builtins.int | None = ..., @@ -1295,7 +1310,9 @@ class DebugOptions(google.protobuf.message.Message): xla_gpu_nccl_termination_timeout_seconds: builtins.int | None = ..., xla_gpu_enable_shared_constants: builtins.bool | None = ..., xla_gpu_enable_cublaslt: builtins.bool | None = ..., - xla_gpu_enable_command_buffer: collections.abc.Iterable[global___DebugOptions.CommandBufferCmdType.ValueType] | None = ..., + xla_gpu_enable_command_buffer: ( + collections.abc.Iterable[global___DebugOptions.CommandBufferCmdType.ValueType] | None + ) = ..., xla_gpu_graph_min_graph_size: builtins.int | None = ..., xla_gpu_graph_enable_concurrent_region: builtins.bool | None = ..., xla_gpu_redzone_scratch_max_megabytes: builtins.int | None = ..., @@ -1399,7 +1416,417 @@ class DebugOptions(google.protobuf.message.Message): xla_experimental_ignore_channel_id: builtins.bool | None = ..., xla_backend_extra_options: collections.abc.Mapping[builtins.str, builtins.str] | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["legacy_command_buffer_custom_call_targets", b"legacy_command_buffer_custom_call_targets", "xla_allow_excess_precision", b"xla_allow_excess_precision", "xla_allow_scalar_index_dynamic_ops", b"xla_allow_scalar_index_dynamic_ops", "xla_backend_extra_options", b"xla_backend_extra_options", "xla_backend_optimization_level", b"xla_backend_optimization_level", "xla_cmd_buffer_trace_cache_size", b"xla_cmd_buffer_trace_cache_size", "xla_cpu_enable_concurrency_optimized_scheduler", b"xla_cpu_enable_concurrency_optimized_scheduler", "xla_cpu_enable_custom_matmul_tiling", b"xla_cpu_enable_custom_matmul_tiling", "xla_cpu_enable_experimental_deallocation", b"xla_cpu_enable_experimental_deallocation", "xla_cpu_enable_fast_math", b"xla_cpu_enable_fast_math", "xla_cpu_enable_fast_min_max", b"xla_cpu_enable_fast_min_max", "xla_cpu_enable_mlir_fusion_outlining", b"xla_cpu_enable_mlir_fusion_outlining", "xla_cpu_enable_mlir_tiling_and_fusion", b"xla_cpu_enable_mlir_tiling_and_fusion", "xla_cpu_enable_xprof_traceme", b"xla_cpu_enable_xprof_traceme", "xla_cpu_fast_math_honor_division", b"xla_cpu_fast_math_honor_division", "xla_cpu_fast_math_honor_functions", b"xla_cpu_fast_math_honor_functions", "xla_cpu_fast_math_honor_infs", b"xla_cpu_fast_math_honor_infs", "xla_cpu_fast_math_honor_nans", b"xla_cpu_fast_math_honor_nans", "xla_cpu_matmul_tiling_k_dim", b"xla_cpu_matmul_tiling_k_dim", "xla_cpu_matmul_tiling_m_dim", b"xla_cpu_matmul_tiling_m_dim", "xla_cpu_matmul_tiling_n_dim", b"xla_cpu_matmul_tiling_n_dim", "xla_cpu_multi_thread_eigen", b"xla_cpu_multi_thread_eigen", "xla_cpu_parallel_codegen_split_count", b"xla_cpu_parallel_codegen_split_count", "xla_cpu_prefer_vector_width", b"xla_cpu_prefer_vector_width", "xla_cpu_strict_dot_conv_math", b"xla_cpu_strict_dot_conv_math", "xla_cpu_use_acl", b"xla_cpu_use_acl", "xla_cpu_use_mkl_dnn", b"xla_cpu_use_mkl_dnn", "xla_cpu_use_thunk_runtime", b"xla_cpu_use_thunk_runtime", "xla_debug_buffer_assignment_show_max", b"xla_debug_buffer_assignment_show_max", "xla_detailed_logging", b"xla_detailed_logging", "xla_disable_all_hlo_passes", b"xla_disable_all_hlo_passes", "xla_disable_hlo_passes", b"xla_disable_hlo_passes", "xla_dump_compress_protos", b"xla_dump_compress_protos", "xla_dump_disable_metadata", b"xla_dump_disable_metadata", "xla_dump_enable_mlir_pretty_form", b"xla_dump_enable_mlir_pretty_form", "xla_dump_fusion_visualization", b"xla_dump_fusion_visualization", "xla_dump_hlo_as_dot", b"xla_dump_hlo_as_dot", "xla_dump_hlo_as_html", b"xla_dump_hlo_as_html", "xla_dump_hlo_as_long_text", b"xla_dump_hlo_as_long_text", "xla_dump_hlo_as_proto", b"xla_dump_hlo_as_proto", "xla_dump_hlo_as_text", b"xla_dump_hlo_as_text", "xla_dump_hlo_as_url", b"xla_dump_hlo_as_url", "xla_dump_hlo_module_re", b"xla_dump_hlo_module_re", "xla_dump_hlo_pass_re", b"xla_dump_hlo_pass_re", "xla_dump_hlo_pipeline_re", b"xla_dump_hlo_pipeline_re", "xla_dump_hlo_snapshots", b"xla_dump_hlo_snapshots", "xla_dump_include_timestamp", b"xla_dump_include_timestamp", "xla_dump_large_constants", b"xla_dump_large_constants", "xla_dump_latency_hiding_schedule", b"xla_dump_latency_hiding_schedule", "xla_dump_max_hlo_modules", b"xla_dump_max_hlo_modules", "xla_dump_module_metadata", b"xla_dump_module_metadata", "xla_dump_to", b"xla_dump_to", "xla_eliminate_hlo_implicit_broadcast", b"xla_eliminate_hlo_implicit_broadcast", "xla_embed_ir_in_executable", b"xla_embed_ir_in_executable", "xla_enable_command_buffers_during_profiling", b"xla_enable_command_buffers_during_profiling", "xla_enable_dumping", b"xla_enable_dumping", "xla_enable_hlo_passes_only", b"xla_enable_hlo_passes_only", "xla_experimental_ignore_channel_id", b"xla_experimental_ignore_channel_id", "xla_force_host_platform_device_count", b"xla_force_host_platform_device_count", "xla_gpu_algorithm_denylist_path", b"xla_gpu_algorithm_denylist_path", "xla_gpu_all_gather_combine_threshold_bytes", b"xla_gpu_all_gather_combine_threshold_bytes", "xla_gpu_all_reduce_blueconnect_num_devices_per_host", b"xla_gpu_all_reduce_blueconnect_num_devices_per_host", "xla_gpu_all_reduce_combine_threshold_bytes", b"xla_gpu_all_reduce_combine_threshold_bytes", "xla_gpu_asm_extra_flags", b"xla_gpu_asm_extra_flags", "xla_gpu_async_dot", b"xla_gpu_async_dot", "xla_gpu_auto_spmd_partitioning_memory_budget_gb", b"xla_gpu_auto_spmd_partitioning_memory_budget_gb", "xla_gpu_auto_spmd_partitioning_memory_budget_ratio", b"xla_gpu_auto_spmd_partitioning_memory_budget_ratio", "xla_gpu_autotune_gemm_rtol", b"xla_gpu_autotune_gemm_rtol", "xla_gpu_autotune_level", b"xla_gpu_autotune_level", "xla_gpu_autotune_max_solutions", b"xla_gpu_autotune_max_solutions", "xla_gpu_collect_cost_model_stats", b"xla_gpu_collect_cost_model_stats", "xla_gpu_collective_inflation_factor", b"xla_gpu_collective_inflation_factor", "xla_gpu_collective_permute_decomposer_threshold", b"xla_gpu_collective_permute_decomposer_threshold", "xla_gpu_copy_insertion_use_region_analysis", b"xla_gpu_copy_insertion_use_region_analysis", "xla_gpu_crash_on_verification_failures", b"xla_gpu_crash_on_verification_failures", "xla_gpu_cublas_fallback", b"xla_gpu_cublas_fallback", "xla_gpu_cuda_data_dir", b"xla_gpu_cuda_data_dir", "xla_gpu_cudnn_gemm_fusion_level", b"xla_gpu_cudnn_gemm_fusion_level", "xla_gpu_cudnn_gemm_max_plans", b"xla_gpu_cudnn_gemm_max_plans", "xla_gpu_deterministic_ops", b"xla_gpu_deterministic_ops", "xla_gpu_disable_async_collectives", b"xla_gpu_disable_async_collectives", "xla_gpu_disable_gpuasm_optimizations", b"xla_gpu_disable_gpuasm_optimizations", "xla_gpu_dump_autotune_logs_to", b"xla_gpu_dump_autotune_logs_to", "xla_gpu_dump_autotune_results_to", b"xla_gpu_dump_autotune_results_to", "xla_gpu_dump_autotuned_gemm_fusions", b"xla_gpu_dump_autotuned_gemm_fusions", "xla_gpu_dump_llvmir", b"xla_gpu_dump_llvmir", "xla_gpu_enable_all_gather_combine_by_dim", b"xla_gpu_enable_all_gather_combine_by_dim", "xla_gpu_enable_analytical_latency_estimator", b"xla_gpu_enable_analytical_latency_estimator", "xla_gpu_enable_approx_costly_collectives", b"xla_gpu_enable_approx_costly_collectives", "xla_gpu_enable_bf16_3way_gemm", b"xla_gpu_enable_bf16_3way_gemm", "xla_gpu_enable_bf16_6way_gemm", b"xla_gpu_enable_bf16_6way_gemm", "xla_gpu_enable_command_buffer", b"xla_gpu_enable_command_buffer", "xla_gpu_enable_cub_radix_sort", b"xla_gpu_enable_cub_radix_sort", "xla_gpu_enable_cublaslt", b"xla_gpu_enable_cublaslt", "xla_gpu_enable_cudnn_fmha", b"xla_gpu_enable_cudnn_fmha", "xla_gpu_enable_cudnn_frontend", b"xla_gpu_enable_cudnn_frontend", "xla_gpu_enable_cudnn_int8x32_convolution_reordering", b"xla_gpu_enable_cudnn_int8x32_convolution_reordering", "xla_gpu_enable_cudnn_layer_norm", b"xla_gpu_enable_cudnn_layer_norm", "xla_gpu_enable_custom_fusions", b"xla_gpu_enable_custom_fusions", "xla_gpu_enable_custom_fusions_re", b"xla_gpu_enable_custom_fusions_re", "xla_gpu_enable_dot_strength_reduction", b"xla_gpu_enable_dot_strength_reduction", "xla_gpu_enable_dynamic_slice_fusion", b"xla_gpu_enable_dynamic_slice_fusion", "xla_gpu_enable_fast_min_max", b"xla_gpu_enable_fast_min_max", "xla_gpu_enable_highest_priority_async_stream", b"xla_gpu_enable_highest_priority_async_stream", "xla_gpu_enable_host_memory_offloading", b"xla_gpu_enable_host_memory_offloading", "xla_gpu_enable_latency_hiding_scheduler", b"xla_gpu_enable_latency_hiding_scheduler", "xla_gpu_enable_libnvjitlink", b"xla_gpu_enable_libnvjitlink", "xla_gpu_enable_libnvptxcompiler", b"xla_gpu_enable_libnvptxcompiler", "xla_gpu_enable_llvm_module_compilation_parallelism", b"xla_gpu_enable_llvm_module_compilation_parallelism", "xla_gpu_enable_nccl_clique_optimization", b"xla_gpu_enable_nccl_clique_optimization", "xla_gpu_enable_nccl_comm_splitting", b"xla_gpu_enable_nccl_comm_splitting", "xla_gpu_enable_nccl_per_stream_comms", b"xla_gpu_enable_nccl_per_stream_comms", "xla_gpu_enable_nccl_user_buffers", b"xla_gpu_enable_nccl_user_buffers", "xla_gpu_enable_pgle_accuracy_checker", b"xla_gpu_enable_pgle_accuracy_checker", "xla_gpu_enable_pipelined_all_gather", b"xla_gpu_enable_pipelined_all_gather", "xla_gpu_enable_pipelined_all_reduce", b"xla_gpu_enable_pipelined_all_reduce", "xla_gpu_enable_pipelined_collectives", b"xla_gpu_enable_pipelined_collectives", "xla_gpu_enable_pipelined_p2p", b"xla_gpu_enable_pipelined_p2p", "xla_gpu_enable_pipelined_reduce_scatter", b"xla_gpu_enable_pipelined_reduce_scatter", "xla_gpu_enable_priority_fusion", b"xla_gpu_enable_priority_fusion", "xla_gpu_enable_reassociation_for_converted_ar", b"xla_gpu_enable_reassociation_for_converted_ar", "xla_gpu_enable_reduce_scatter_combine_by_dim", b"xla_gpu_enable_reduce_scatter_combine_by_dim", "xla_gpu_enable_reduction_epilogue_fusion", b"xla_gpu_enable_reduction_epilogue_fusion", "xla_gpu_enable_shared_constants", b"xla_gpu_enable_shared_constants", "xla_gpu_enable_split_k_autotuning", b"xla_gpu_enable_split_k_autotuning", "xla_gpu_enable_triton_gemm", b"xla_gpu_enable_triton_gemm", "xla_gpu_enable_triton_gemm_int4", b"xla_gpu_enable_triton_gemm_int4", "xla_gpu_enable_triton_hopper", b"xla_gpu_enable_triton_hopper", "xla_gpu_enable_while_loop_double_buffering", b"xla_gpu_enable_while_loop_double_buffering", "xla_gpu_enable_while_loop_reduce_scatter_code_motion", b"xla_gpu_enable_while_loop_reduce_scatter_code_motion", "xla_gpu_enable_while_loop_unrolling", b"xla_gpu_enable_while_loop_unrolling", "xla_gpu_ensure_minor_dot_contraction_dims", b"xla_gpu_ensure_minor_dot_contraction_dims", "xla_gpu_exclude_nondeterministic_ops", b"xla_gpu_exclude_nondeterministic_ops", "xla_gpu_executable_terminate_timeout_seconds", b"xla_gpu_executable_terminate_timeout_seconds", "xla_gpu_executable_warn_stuck_timeout_seconds", b"xla_gpu_executable_warn_stuck_timeout_seconds", "xla_gpu_exhaustive_tiling_search", b"xla_gpu_exhaustive_tiling_search", "xla_gpu_experimental_autotune_cache_mode", b"xla_gpu_experimental_autotune_cache_mode", "xla_gpu_experimental_disable_binary_libraries", b"xla_gpu_experimental_disable_binary_libraries", "xla_gpu_experimental_enable_triton_softmax_priority_fusion", b"xla_gpu_experimental_enable_triton_softmax_priority_fusion", "xla_gpu_filter_kernels_spilling_registers_on_autotuning", b"xla_gpu_filter_kernels_spilling_registers_on_autotuning", "xla_gpu_force_compilation_parallelism", b"xla_gpu_force_compilation_parallelism", "xla_gpu_force_conv_nchw", b"xla_gpu_force_conv_nchw", "xla_gpu_force_conv_nhwc", b"xla_gpu_force_conv_nhwc", "xla_gpu_ftz", b"xla_gpu_ftz", "xla_gpu_fused_attention_use_cudnn_rng", b"xla_gpu_fused_attention_use_cudnn_rng", "xla_gpu_gemm_rewrite_size_threshold", b"xla_gpu_gemm_rewrite_size_threshold", "xla_gpu_graph_enable_concurrent_region", b"xla_gpu_graph_enable_concurrent_region", "xla_gpu_graph_min_graph_size", b"xla_gpu_graph_min_graph_size", "xla_gpu_kernel_cache_file", b"xla_gpu_kernel_cache_file", "xla_gpu_lhs_enable_gpu_async_tracker", b"xla_gpu_lhs_enable_gpu_async_tracker", "xla_gpu_llvm_ir_file", b"xla_gpu_llvm_ir_file", "xla_gpu_llvm_verification_level", b"xla_gpu_llvm_verification_level", "xla_gpu_load_autotune_results_from", b"xla_gpu_load_autotune_results_from", "xla_gpu_memory_limit_slop_factor", b"xla_gpu_memory_limit_slop_factor", "xla_gpu_mlir_emitter_level", b"xla_gpu_mlir_emitter_level", "xla_gpu_mock_custom_calls", b"xla_gpu_mock_custom_calls", "xla_gpu_multi_streamed_windowed_einsum", b"xla_gpu_multi_streamed_windowed_einsum", "xla_gpu_nccl_collective_max_nchannels", b"xla_gpu_nccl_collective_max_nchannels", "xla_gpu_nccl_p2p_max_nchannels", b"xla_gpu_nccl_p2p_max_nchannels", "xla_gpu_nccl_terminate_on_error", b"xla_gpu_nccl_terminate_on_error", "xla_gpu_nccl_termination_timeout_seconds", b"xla_gpu_nccl_termination_timeout_seconds", "xla_gpu_override_gemm_autotuner", b"xla_gpu_override_gemm_autotuner", "xla_gpu_per_fusion_autotune_cache_dir", b"xla_gpu_per_fusion_autotune_cache_dir", "xla_gpu_pgle_profile_file_or_directory_path", b"xla_gpu_pgle_profile_file_or_directory_path", "xla_gpu_ptx_file", b"xla_gpu_ptx_file", "xla_gpu_reduce_scatter_combine_threshold_bytes", b"xla_gpu_reduce_scatter_combine_threshold_bytes", "xla_gpu_redzone_padding_bytes", b"xla_gpu_redzone_padding_bytes", "xla_gpu_redzone_scratch_max_megabytes", b"xla_gpu_redzone_scratch_max_megabytes", "xla_gpu_require_complete_aot_autotune_results", b"xla_gpu_require_complete_aot_autotune_results", "xla_gpu_run_post_layout_collective_pipeliner", b"xla_gpu_run_post_layout_collective_pipeliner", "xla_gpu_shape_checks", b"xla_gpu_shape_checks", "xla_gpu_shard_autotuning", b"xla_gpu_shard_autotuning", "xla_gpu_strict_conv_algorithm_picker", b"xla_gpu_strict_conv_algorithm_picker", "xla_gpu_target_config_filename", b"xla_gpu_target_config_filename", "xla_gpu_temp_buffer_use_separate_color", b"xla_gpu_temp_buffer_use_separate_color", "xla_gpu_threshold_for_windowed_einsum_mib", b"xla_gpu_threshold_for_windowed_einsum_mib", "xla_gpu_triton_fusion_level", b"xla_gpu_triton_fusion_level", "xla_gpu_triton_gemm_any", b"xla_gpu_triton_gemm_any", "xla_gpu_triton_gemm_disable_reduced_precision_reduction", b"xla_gpu_triton_gemm_disable_reduced_precision_reduction", "xla_gpu_unsafe_fallback_to_driver_on_ptxas_not_found", b"xla_gpu_unsafe_fallback_to_driver_on_ptxas_not_found", "xla_gpu_unsafe_pipelined_loop_annotator", b"xla_gpu_unsafe_pipelined_loop_annotator", "xla_gpu_unsupported_enable_triton_gemm", b"xla_gpu_unsupported_enable_triton_gemm", "xla_gpu_use_memcpy_local_p2p", b"xla_gpu_use_memcpy_local_p2p", "xla_gpu_use_runtime_fusion", b"xla_gpu_use_runtime_fusion", "xla_gpu_verify_triton_fusion_numerics", b"xla_gpu_verify_triton_fusion_numerics", "xla_hlo_evaluator_use_fast_path", b"xla_hlo_evaluator_use_fast_path", "xla_hlo_graph_addresses", b"xla_hlo_graph_addresses", "xla_hlo_graph_sharding_color", b"xla_hlo_graph_sharding_color", "xla_hlo_profile", b"xla_hlo_profile", "xla_llvm_disable_expensive_passes", b"xla_llvm_disable_expensive_passes", "xla_llvm_enable_alias_scope_metadata", b"xla_llvm_enable_alias_scope_metadata", "xla_llvm_enable_invariant_load_metadata", b"xla_llvm_enable_invariant_load_metadata", "xla_llvm_enable_noalias_metadata", b"xla_llvm_enable_noalias_metadata", "xla_llvm_force_inline_before_split", b"xla_llvm_force_inline_before_split", "xla_multiheap_size_constraint_per_heap", b"xla_multiheap_size_constraint_per_heap", "xla_partitioning_algorithm", b"xla_partitioning_algorithm", "xla_reduce_window_rewrite_base_length", b"xla_reduce_window_rewrite_base_length", "xla_step_marker_location", b"xla_step_marker_location", "xla_syntax_sugar_async_ops", b"xla_syntax_sugar_async_ops", "xla_test_all_input_layouts", b"xla_test_all_input_layouts", "xla_test_all_output_layouts", b"xla_test_all_output_layouts", "xla_tpu_detect_inf", b"xla_tpu_detect_inf", "xla_tpu_detect_nan", b"xla_tpu_detect_nan"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "legacy_command_buffer_custom_call_targets", + b"legacy_command_buffer_custom_call_targets", + "xla_allow_excess_precision", + b"xla_allow_excess_precision", + "xla_allow_scalar_index_dynamic_ops", + b"xla_allow_scalar_index_dynamic_ops", + "xla_backend_extra_options", + b"xla_backend_extra_options", + "xla_backend_optimization_level", + b"xla_backend_optimization_level", + "xla_cmd_buffer_trace_cache_size", + b"xla_cmd_buffer_trace_cache_size", + "xla_cpu_enable_concurrency_optimized_scheduler", + b"xla_cpu_enable_concurrency_optimized_scheduler", + "xla_cpu_enable_custom_matmul_tiling", + b"xla_cpu_enable_custom_matmul_tiling", + "xla_cpu_enable_experimental_deallocation", + b"xla_cpu_enable_experimental_deallocation", + "xla_cpu_enable_fast_math", + b"xla_cpu_enable_fast_math", + "xla_cpu_enable_fast_min_max", + b"xla_cpu_enable_fast_min_max", + "xla_cpu_enable_mlir_fusion_outlining", + b"xla_cpu_enable_mlir_fusion_outlining", + "xla_cpu_enable_mlir_tiling_and_fusion", + b"xla_cpu_enable_mlir_tiling_and_fusion", + "xla_cpu_enable_xprof_traceme", + b"xla_cpu_enable_xprof_traceme", + "xla_cpu_fast_math_honor_division", + b"xla_cpu_fast_math_honor_division", + "xla_cpu_fast_math_honor_functions", + b"xla_cpu_fast_math_honor_functions", + "xla_cpu_fast_math_honor_infs", + b"xla_cpu_fast_math_honor_infs", + "xla_cpu_fast_math_honor_nans", + b"xla_cpu_fast_math_honor_nans", + "xla_cpu_matmul_tiling_k_dim", + b"xla_cpu_matmul_tiling_k_dim", + "xla_cpu_matmul_tiling_m_dim", + b"xla_cpu_matmul_tiling_m_dim", + "xla_cpu_matmul_tiling_n_dim", + b"xla_cpu_matmul_tiling_n_dim", + "xla_cpu_multi_thread_eigen", + b"xla_cpu_multi_thread_eigen", + "xla_cpu_parallel_codegen_split_count", + b"xla_cpu_parallel_codegen_split_count", + "xla_cpu_prefer_vector_width", + b"xla_cpu_prefer_vector_width", + "xla_cpu_strict_dot_conv_math", + b"xla_cpu_strict_dot_conv_math", + "xla_cpu_use_acl", + b"xla_cpu_use_acl", + "xla_cpu_use_mkl_dnn", + b"xla_cpu_use_mkl_dnn", + "xla_cpu_use_thunk_runtime", + b"xla_cpu_use_thunk_runtime", + "xla_debug_buffer_assignment_show_max", + b"xla_debug_buffer_assignment_show_max", + "xla_detailed_logging", + b"xla_detailed_logging", + "xla_disable_all_hlo_passes", + b"xla_disable_all_hlo_passes", + "xla_disable_hlo_passes", + b"xla_disable_hlo_passes", + "xla_dump_compress_protos", + b"xla_dump_compress_protos", + "xla_dump_disable_metadata", + b"xla_dump_disable_metadata", + "xla_dump_enable_mlir_pretty_form", + b"xla_dump_enable_mlir_pretty_form", + "xla_dump_fusion_visualization", + b"xla_dump_fusion_visualization", + "xla_dump_hlo_as_dot", + b"xla_dump_hlo_as_dot", + "xla_dump_hlo_as_html", + b"xla_dump_hlo_as_html", + "xla_dump_hlo_as_long_text", + b"xla_dump_hlo_as_long_text", + "xla_dump_hlo_as_proto", + b"xla_dump_hlo_as_proto", + "xla_dump_hlo_as_text", + b"xla_dump_hlo_as_text", + "xla_dump_hlo_as_url", + b"xla_dump_hlo_as_url", + "xla_dump_hlo_module_re", + b"xla_dump_hlo_module_re", + "xla_dump_hlo_pass_re", + b"xla_dump_hlo_pass_re", + "xla_dump_hlo_pipeline_re", + b"xla_dump_hlo_pipeline_re", + "xla_dump_hlo_snapshots", + b"xla_dump_hlo_snapshots", + "xla_dump_include_timestamp", + b"xla_dump_include_timestamp", + "xla_dump_large_constants", + b"xla_dump_large_constants", + "xla_dump_latency_hiding_schedule", + b"xla_dump_latency_hiding_schedule", + "xla_dump_max_hlo_modules", + b"xla_dump_max_hlo_modules", + "xla_dump_module_metadata", + b"xla_dump_module_metadata", + "xla_dump_to", + b"xla_dump_to", + "xla_eliminate_hlo_implicit_broadcast", + b"xla_eliminate_hlo_implicit_broadcast", + "xla_embed_ir_in_executable", + b"xla_embed_ir_in_executable", + "xla_enable_command_buffers_during_profiling", + b"xla_enable_command_buffers_during_profiling", + "xla_enable_dumping", + b"xla_enable_dumping", + "xla_enable_hlo_passes_only", + b"xla_enable_hlo_passes_only", + "xla_experimental_ignore_channel_id", + b"xla_experimental_ignore_channel_id", + "xla_force_host_platform_device_count", + b"xla_force_host_platform_device_count", + "xla_gpu_algorithm_denylist_path", + b"xla_gpu_algorithm_denylist_path", + "xla_gpu_all_gather_combine_threshold_bytes", + b"xla_gpu_all_gather_combine_threshold_bytes", + "xla_gpu_all_reduce_blueconnect_num_devices_per_host", + b"xla_gpu_all_reduce_blueconnect_num_devices_per_host", + "xla_gpu_all_reduce_combine_threshold_bytes", + b"xla_gpu_all_reduce_combine_threshold_bytes", + "xla_gpu_asm_extra_flags", + b"xla_gpu_asm_extra_flags", + "xla_gpu_async_dot", + b"xla_gpu_async_dot", + "xla_gpu_auto_spmd_partitioning_memory_budget_gb", + b"xla_gpu_auto_spmd_partitioning_memory_budget_gb", + "xla_gpu_auto_spmd_partitioning_memory_budget_ratio", + b"xla_gpu_auto_spmd_partitioning_memory_budget_ratio", + "xla_gpu_autotune_gemm_rtol", + b"xla_gpu_autotune_gemm_rtol", + "xla_gpu_autotune_level", + b"xla_gpu_autotune_level", + "xla_gpu_autotune_max_solutions", + b"xla_gpu_autotune_max_solutions", + "xla_gpu_collect_cost_model_stats", + b"xla_gpu_collect_cost_model_stats", + "xla_gpu_collective_inflation_factor", + b"xla_gpu_collective_inflation_factor", + "xla_gpu_collective_permute_decomposer_threshold", + b"xla_gpu_collective_permute_decomposer_threshold", + "xla_gpu_copy_insertion_use_region_analysis", + b"xla_gpu_copy_insertion_use_region_analysis", + "xla_gpu_crash_on_verification_failures", + b"xla_gpu_crash_on_verification_failures", + "xla_gpu_cublas_fallback", + b"xla_gpu_cublas_fallback", + "xla_gpu_cuda_data_dir", + b"xla_gpu_cuda_data_dir", + "xla_gpu_cudnn_gemm_fusion_level", + b"xla_gpu_cudnn_gemm_fusion_level", + "xla_gpu_cudnn_gemm_max_plans", + b"xla_gpu_cudnn_gemm_max_plans", + "xla_gpu_deterministic_ops", + b"xla_gpu_deterministic_ops", + "xla_gpu_disable_async_collectives", + b"xla_gpu_disable_async_collectives", + "xla_gpu_disable_gpuasm_optimizations", + b"xla_gpu_disable_gpuasm_optimizations", + "xla_gpu_dump_autotune_logs_to", + b"xla_gpu_dump_autotune_logs_to", + "xla_gpu_dump_autotune_results_to", + b"xla_gpu_dump_autotune_results_to", + "xla_gpu_dump_autotuned_gemm_fusions", + b"xla_gpu_dump_autotuned_gemm_fusions", + "xla_gpu_dump_llvmir", + b"xla_gpu_dump_llvmir", + "xla_gpu_enable_all_gather_combine_by_dim", + b"xla_gpu_enable_all_gather_combine_by_dim", + "xla_gpu_enable_analytical_latency_estimator", + b"xla_gpu_enable_analytical_latency_estimator", + "xla_gpu_enable_approx_costly_collectives", + b"xla_gpu_enable_approx_costly_collectives", + "xla_gpu_enable_bf16_3way_gemm", + b"xla_gpu_enable_bf16_3way_gemm", + "xla_gpu_enable_bf16_6way_gemm", + b"xla_gpu_enable_bf16_6way_gemm", + "xla_gpu_enable_command_buffer", + b"xla_gpu_enable_command_buffer", + "xla_gpu_enable_cub_radix_sort", + b"xla_gpu_enable_cub_radix_sort", + "xla_gpu_enable_cublaslt", + b"xla_gpu_enable_cublaslt", + "xla_gpu_enable_cudnn_fmha", + b"xla_gpu_enable_cudnn_fmha", + "xla_gpu_enable_cudnn_frontend", + b"xla_gpu_enable_cudnn_frontend", + "xla_gpu_enable_cudnn_int8x32_convolution_reordering", + b"xla_gpu_enable_cudnn_int8x32_convolution_reordering", + "xla_gpu_enable_cudnn_layer_norm", + b"xla_gpu_enable_cudnn_layer_norm", + "xla_gpu_enable_custom_fusions", + b"xla_gpu_enable_custom_fusions", + "xla_gpu_enable_custom_fusions_re", + b"xla_gpu_enable_custom_fusions_re", + "xla_gpu_enable_dot_strength_reduction", + b"xla_gpu_enable_dot_strength_reduction", + "xla_gpu_enable_dynamic_slice_fusion", + b"xla_gpu_enable_dynamic_slice_fusion", + "xla_gpu_enable_fast_min_max", + b"xla_gpu_enable_fast_min_max", + "xla_gpu_enable_highest_priority_async_stream", + b"xla_gpu_enable_highest_priority_async_stream", + "xla_gpu_enable_host_memory_offloading", + b"xla_gpu_enable_host_memory_offloading", + "xla_gpu_enable_latency_hiding_scheduler", + b"xla_gpu_enable_latency_hiding_scheduler", + "xla_gpu_enable_libnvjitlink", + b"xla_gpu_enable_libnvjitlink", + "xla_gpu_enable_libnvptxcompiler", + b"xla_gpu_enable_libnvptxcompiler", + "xla_gpu_enable_llvm_module_compilation_parallelism", + b"xla_gpu_enable_llvm_module_compilation_parallelism", + "xla_gpu_enable_nccl_clique_optimization", + b"xla_gpu_enable_nccl_clique_optimization", + "xla_gpu_enable_nccl_comm_splitting", + b"xla_gpu_enable_nccl_comm_splitting", + "xla_gpu_enable_nccl_per_stream_comms", + b"xla_gpu_enable_nccl_per_stream_comms", + "xla_gpu_enable_nccl_user_buffers", + b"xla_gpu_enable_nccl_user_buffers", + "xla_gpu_enable_pgle_accuracy_checker", + b"xla_gpu_enable_pgle_accuracy_checker", + "xla_gpu_enable_pipelined_all_gather", + b"xla_gpu_enable_pipelined_all_gather", + "xla_gpu_enable_pipelined_all_reduce", + b"xla_gpu_enable_pipelined_all_reduce", + "xla_gpu_enable_pipelined_collectives", + b"xla_gpu_enable_pipelined_collectives", + "xla_gpu_enable_pipelined_p2p", + b"xla_gpu_enable_pipelined_p2p", + "xla_gpu_enable_pipelined_reduce_scatter", + b"xla_gpu_enable_pipelined_reduce_scatter", + "xla_gpu_enable_priority_fusion", + b"xla_gpu_enable_priority_fusion", + "xla_gpu_enable_reassociation_for_converted_ar", + b"xla_gpu_enable_reassociation_for_converted_ar", + "xla_gpu_enable_reduce_scatter_combine_by_dim", + b"xla_gpu_enable_reduce_scatter_combine_by_dim", + "xla_gpu_enable_reduction_epilogue_fusion", + b"xla_gpu_enable_reduction_epilogue_fusion", + "xla_gpu_enable_shared_constants", + b"xla_gpu_enable_shared_constants", + "xla_gpu_enable_split_k_autotuning", + b"xla_gpu_enable_split_k_autotuning", + "xla_gpu_enable_triton_gemm", + b"xla_gpu_enable_triton_gemm", + "xla_gpu_enable_triton_gemm_int4", + b"xla_gpu_enable_triton_gemm_int4", + "xla_gpu_enable_triton_hopper", + b"xla_gpu_enable_triton_hopper", + "xla_gpu_enable_while_loop_double_buffering", + b"xla_gpu_enable_while_loop_double_buffering", + "xla_gpu_enable_while_loop_reduce_scatter_code_motion", + b"xla_gpu_enable_while_loop_reduce_scatter_code_motion", + "xla_gpu_enable_while_loop_unrolling", + b"xla_gpu_enable_while_loop_unrolling", + "xla_gpu_ensure_minor_dot_contraction_dims", + b"xla_gpu_ensure_minor_dot_contraction_dims", + "xla_gpu_exclude_nondeterministic_ops", + b"xla_gpu_exclude_nondeterministic_ops", + "xla_gpu_executable_terminate_timeout_seconds", + b"xla_gpu_executable_terminate_timeout_seconds", + "xla_gpu_executable_warn_stuck_timeout_seconds", + b"xla_gpu_executable_warn_stuck_timeout_seconds", + "xla_gpu_exhaustive_tiling_search", + b"xla_gpu_exhaustive_tiling_search", + "xla_gpu_experimental_autotune_cache_mode", + b"xla_gpu_experimental_autotune_cache_mode", + "xla_gpu_experimental_disable_binary_libraries", + b"xla_gpu_experimental_disable_binary_libraries", + "xla_gpu_experimental_enable_triton_softmax_priority_fusion", + b"xla_gpu_experimental_enable_triton_softmax_priority_fusion", + "xla_gpu_filter_kernels_spilling_registers_on_autotuning", + b"xla_gpu_filter_kernels_spilling_registers_on_autotuning", + "xla_gpu_force_compilation_parallelism", + b"xla_gpu_force_compilation_parallelism", + "xla_gpu_force_conv_nchw", + b"xla_gpu_force_conv_nchw", + "xla_gpu_force_conv_nhwc", + b"xla_gpu_force_conv_nhwc", + "xla_gpu_ftz", + b"xla_gpu_ftz", + "xla_gpu_fused_attention_use_cudnn_rng", + b"xla_gpu_fused_attention_use_cudnn_rng", + "xla_gpu_gemm_rewrite_size_threshold", + b"xla_gpu_gemm_rewrite_size_threshold", + "xla_gpu_graph_enable_concurrent_region", + b"xla_gpu_graph_enable_concurrent_region", + "xla_gpu_graph_min_graph_size", + b"xla_gpu_graph_min_graph_size", + "xla_gpu_kernel_cache_file", + b"xla_gpu_kernel_cache_file", + "xla_gpu_lhs_enable_gpu_async_tracker", + b"xla_gpu_lhs_enable_gpu_async_tracker", + "xla_gpu_llvm_ir_file", + b"xla_gpu_llvm_ir_file", + "xla_gpu_llvm_verification_level", + b"xla_gpu_llvm_verification_level", + "xla_gpu_load_autotune_results_from", + b"xla_gpu_load_autotune_results_from", + "xla_gpu_memory_limit_slop_factor", + b"xla_gpu_memory_limit_slop_factor", + "xla_gpu_mlir_emitter_level", + b"xla_gpu_mlir_emitter_level", + "xla_gpu_mock_custom_calls", + b"xla_gpu_mock_custom_calls", + "xla_gpu_multi_streamed_windowed_einsum", + b"xla_gpu_multi_streamed_windowed_einsum", + "xla_gpu_nccl_collective_max_nchannels", + b"xla_gpu_nccl_collective_max_nchannels", + "xla_gpu_nccl_p2p_max_nchannels", + b"xla_gpu_nccl_p2p_max_nchannels", + "xla_gpu_nccl_terminate_on_error", + b"xla_gpu_nccl_terminate_on_error", + "xla_gpu_nccl_termination_timeout_seconds", + b"xla_gpu_nccl_termination_timeout_seconds", + "xla_gpu_override_gemm_autotuner", + b"xla_gpu_override_gemm_autotuner", + "xla_gpu_per_fusion_autotune_cache_dir", + b"xla_gpu_per_fusion_autotune_cache_dir", + "xla_gpu_pgle_profile_file_or_directory_path", + b"xla_gpu_pgle_profile_file_or_directory_path", + "xla_gpu_ptx_file", + b"xla_gpu_ptx_file", + "xla_gpu_reduce_scatter_combine_threshold_bytes", + b"xla_gpu_reduce_scatter_combine_threshold_bytes", + "xla_gpu_redzone_padding_bytes", + b"xla_gpu_redzone_padding_bytes", + "xla_gpu_redzone_scratch_max_megabytes", + b"xla_gpu_redzone_scratch_max_megabytes", + "xla_gpu_require_complete_aot_autotune_results", + b"xla_gpu_require_complete_aot_autotune_results", + "xla_gpu_run_post_layout_collective_pipeliner", + b"xla_gpu_run_post_layout_collective_pipeliner", + "xla_gpu_shape_checks", + b"xla_gpu_shape_checks", + "xla_gpu_shard_autotuning", + b"xla_gpu_shard_autotuning", + "xla_gpu_strict_conv_algorithm_picker", + b"xla_gpu_strict_conv_algorithm_picker", + "xla_gpu_target_config_filename", + b"xla_gpu_target_config_filename", + "xla_gpu_temp_buffer_use_separate_color", + b"xla_gpu_temp_buffer_use_separate_color", + "xla_gpu_threshold_for_windowed_einsum_mib", + b"xla_gpu_threshold_for_windowed_einsum_mib", + "xla_gpu_triton_fusion_level", + b"xla_gpu_triton_fusion_level", + "xla_gpu_triton_gemm_any", + b"xla_gpu_triton_gemm_any", + "xla_gpu_triton_gemm_disable_reduced_precision_reduction", + b"xla_gpu_triton_gemm_disable_reduced_precision_reduction", + "xla_gpu_unsafe_fallback_to_driver_on_ptxas_not_found", + b"xla_gpu_unsafe_fallback_to_driver_on_ptxas_not_found", + "xla_gpu_unsafe_pipelined_loop_annotator", + b"xla_gpu_unsafe_pipelined_loop_annotator", + "xla_gpu_unsupported_enable_triton_gemm", + b"xla_gpu_unsupported_enable_triton_gemm", + "xla_gpu_use_memcpy_local_p2p", + b"xla_gpu_use_memcpy_local_p2p", + "xla_gpu_use_runtime_fusion", + b"xla_gpu_use_runtime_fusion", + "xla_gpu_verify_triton_fusion_numerics", + b"xla_gpu_verify_triton_fusion_numerics", + "xla_hlo_evaluator_use_fast_path", + b"xla_hlo_evaluator_use_fast_path", + "xla_hlo_graph_addresses", + b"xla_hlo_graph_addresses", + "xla_hlo_graph_sharding_color", + b"xla_hlo_graph_sharding_color", + "xla_hlo_profile", + b"xla_hlo_profile", + "xla_llvm_disable_expensive_passes", + b"xla_llvm_disable_expensive_passes", + "xla_llvm_enable_alias_scope_metadata", + b"xla_llvm_enable_alias_scope_metadata", + "xla_llvm_enable_invariant_load_metadata", + b"xla_llvm_enable_invariant_load_metadata", + "xla_llvm_enable_noalias_metadata", + b"xla_llvm_enable_noalias_metadata", + "xla_llvm_force_inline_before_split", + b"xla_llvm_force_inline_before_split", + "xla_multiheap_size_constraint_per_heap", + b"xla_multiheap_size_constraint_per_heap", + "xla_partitioning_algorithm", + b"xla_partitioning_algorithm", + "xla_reduce_window_rewrite_base_length", + b"xla_reduce_window_rewrite_base_length", + "xla_step_marker_location", + b"xla_step_marker_location", + "xla_syntax_sugar_async_ops", + b"xla_syntax_sugar_async_ops", + "xla_test_all_input_layouts", + b"xla_test_all_input_layouts", + "xla_test_all_output_layouts", + b"xla_test_all_output_layouts", + "xla_tpu_detect_inf", + b"xla_tpu_detect_inf", + "xla_tpu_detect_nan", + b"xla_tpu_detect_nan", + ], + ) -> None: ... global___DebugOptions = DebugOptions @@ -1417,11 +1844,7 @@ class GpuCompilationEnvironment(google.protobuf.message.Message): """Temporary dummy flag is added to test the flow. To be removed when we add flags here. """ - def __init__( - self, - *, - dummy_flag: builtins.int | None = ..., - ) -> None: ... + def __init__(self, *, dummy_flag: builtins.int | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["dummy_flag", b"dummy_flag"]) -> None: ... global___GpuCompilationEnvironment = GpuCompilationEnvironment @@ -1445,7 +1868,17 @@ class ShardableValueUpdatePairProto(google.protobuf.message.Message): parameter_shape_index: collections.abc.Iterable[builtins.int] | None = ..., output_shape_index: collections.abc.Iterable[builtins.int] | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["input_parameter_number", b"input_parameter_number", "output_shape_index", b"output_shape_index", "parameter_shape_index", b"parameter_shape_index"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "input_parameter_number", + b"input_parameter_number", + "output_shape_index", + b"output_shape_index", + "parameter_shape_index", + b"parameter_shape_index", + ], + ) -> None: ... global___ShardableValueUpdatePairProto = ShardableValueUpdatePairProto @@ -1541,7 +1974,9 @@ class ExecutionOptions(google.protobuf.message.Message): @property def debug_options(self) -> global___DebugOptions: ... @property - def device_handles(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[tensorflow.compiler.xla.xla_data_pb2.DeviceHandle]: + def device_handles( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[tensorflow.compiler.xla.xla_data_pb2.DeviceHandle]: """This optional field specifies a particular set of devices to run the computation on. The computation will be partitioned across these devices. If not provided, the default device will be chosen. @@ -1566,7 +2001,9 @@ class ExecutionOptions(google.protobuf.message.Message): """ @property - def allow_spmd_sharding_propagation_to_parameters(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.bool]: + def allow_spmd_sharding_propagation_to_parameters( + self, + ) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.bool]: """Allows sharding propagation to propagate to the parameters. This changes the input shape of the computation (which is undesirable), but it can be used to allow to run partial compilation to determine what would be the @@ -1580,7 +2017,9 @@ class ExecutionOptions(google.protobuf.message.Message): """ @property - def allow_spmd_sharding_propagation_to_output(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.bool]: + def allow_spmd_sharding_propagation_to_output( + self, + ) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.bool]: """Allows sharding propagation to propagate to the outputs. This changes the output shape of the computation (which is undesirable), but it can be used to allow to run partial compilation to determine what would be the output @@ -1597,11 +2036,15 @@ class ExecutionOptions(google.protobuf.message.Message): """ @property - def param_requires_broadcast_via_collectives(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.bool]: + def param_requires_broadcast_via_collectives( + self, + ) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.bool]: """Whether to broadcast args across all replicas. One entry per arg.""" @property - def shardable_value_update_pairs(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___ShardableValueUpdatePairProto]: + def shardable_value_update_pairs( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___ShardableValueUpdatePairProto]: """The list of input/output pairs in the main program that could be sharded.""" def __init__( @@ -1630,8 +2073,66 @@ class ExecutionOptions(google.protobuf.message.Message): device_memory_size: builtins.int | None = ..., use_shardy_partitioner: builtins.bool | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["debug_options", b"debug_options", "device_assignment", b"device_assignment", "shape_with_output_layout", b"shape_with_output_layout"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["alias_passthrough_params", b"alias_passthrough_params", "allow_separate_sharding_programs", b"allow_separate_sharding_programs", "allow_spmd_sharding_propagation_to_output", b"allow_spmd_sharding_propagation_to_output", "allow_spmd_sharding_propagation_to_parameters", b"allow_spmd_sharding_propagation_to_parameters", "auto_spmd_partitioning_mesh_ids", b"auto_spmd_partitioning_mesh_ids", "auto_spmd_partitioning_mesh_shape", b"auto_spmd_partitioning_mesh_shape", "debug_options", b"debug_options", "deduplicate_hlo", b"deduplicate_hlo", "device_assignment", b"device_assignment", "device_handles", b"device_handles", "device_memory_size", b"device_memory_size", "fdo_profile", b"fdo_profile", "launch_id", b"launch_id", "num_partitions", b"num_partitions", "num_replicas", b"num_replicas", "param_requires_broadcast_via_collectives", b"param_requires_broadcast_via_collectives", "seed", b"seed", "shape_with_output_layout", b"shape_with_output_layout", "shardable_value_update_pairs", b"shardable_value_update_pairs", "use_auto_spmd_partitioning", b"use_auto_spmd_partitioning", "use_shardy_partitioner", b"use_shardy_partitioner", "use_spmd_partitioning", b"use_spmd_partitioning"]) -> None: ... + def HasField( + self, + field_name: typing.Literal[ + "debug_options", + b"debug_options", + "device_assignment", + b"device_assignment", + "shape_with_output_layout", + b"shape_with_output_layout", + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "alias_passthrough_params", + b"alias_passthrough_params", + "allow_separate_sharding_programs", + b"allow_separate_sharding_programs", + "allow_spmd_sharding_propagation_to_output", + b"allow_spmd_sharding_propagation_to_output", + "allow_spmd_sharding_propagation_to_parameters", + b"allow_spmd_sharding_propagation_to_parameters", + "auto_spmd_partitioning_mesh_ids", + b"auto_spmd_partitioning_mesh_ids", + "auto_spmd_partitioning_mesh_shape", + b"auto_spmd_partitioning_mesh_shape", + "debug_options", + b"debug_options", + "deduplicate_hlo", + b"deduplicate_hlo", + "device_assignment", + b"device_assignment", + "device_handles", + b"device_handles", + "device_memory_size", + b"device_memory_size", + "fdo_profile", + b"fdo_profile", + "launch_id", + b"launch_id", + "num_partitions", + b"num_partitions", + "num_replicas", + b"num_replicas", + "param_requires_broadcast_via_collectives", + b"param_requires_broadcast_via_collectives", + "seed", + b"seed", + "shape_with_output_layout", + b"shape_with_output_layout", + "shardable_value_update_pairs", + b"shardable_value_update_pairs", + "use_auto_spmd_partitioning", + b"use_auto_spmd_partitioning", + "use_shardy_partitioner", + b"use_shardy_partitioner", + "use_spmd_partitioning", + b"use_spmd_partitioning", + ], + ) -> None: ... global___ExecutionOptions = ExecutionOptions @@ -1649,7 +2150,10 @@ class HloModuleConfigProto(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _FusionConfigCollectionEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[HloModuleConfigProto._FusionConfigCollection.ValueType], builtins.type): + class _FusionConfigCollectionEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[HloModuleConfigProto._FusionConfigCollection.ValueType], + builtins.type, + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor OFF: HloModuleConfigProto._FusionConfigCollection.ValueType # 0 """Do not collect configuration.""" @@ -1673,11 +2177,7 @@ class HloModuleConfigProto(google.protobuf.message.Message): VALS_FIELD_NUMBER: builtins.int @property def vals(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.bool]: ... - def __init__( - self, - *, - vals: collections.abc.Iterable[builtins.bool] | None = ..., - ) -> None: ... + def __init__(self, *, vals: collections.abc.Iterable[builtins.bool] | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["vals", b"vals"]) -> None: ... @typing.final @@ -1687,11 +2187,7 @@ class HloModuleConfigProto(google.protobuf.message.Message): VALS_FIELD_NUMBER: builtins.int @property def vals(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.int]: ... - def __init__( - self, - *, - vals: collections.abc.Iterable[builtins.int] | None = ..., - ) -> None: ... + def __init__(self, *, vals: collections.abc.Iterable[builtins.int] | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["vals", b"vals"]) -> None: ... @typing.final @@ -1700,12 +2196,10 @@ class HloModuleConfigProto(google.protobuf.message.Message): LISTS_FIELD_NUMBER: builtins.int @property - def lists(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___HloModuleConfigProto.Int64List]: ... - def __init__( + def lists( self, - *, - lists: collections.abc.Iterable[global___HloModuleConfigProto.Int64List] | None = ..., - ) -> None: ... + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___HloModuleConfigProto.Int64List]: ... + def __init__(self, *, lists: collections.abc.Iterable[global___HloModuleConfigProto.Int64List] | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["lists", b"lists"]) -> None: ... @typing.final @@ -1718,10 +2212,7 @@ class HloModuleConfigProto(google.protobuf.message.Message): @property def value(self) -> global___HloModuleConfigProto.Int64List: ... def __init__( - self, - *, - key: builtins.str | None = ..., - value: global___HloModuleConfigProto.Int64List | None = ..., + self, *, key: builtins.str | None = ..., value: global___HloModuleConfigProto.Int64List | None = ... ) -> None: ... def HasField(self, field_name: typing.Literal["value", b"value"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["key", b"key", "value", b"value"]) -> None: ... @@ -1734,12 +2225,7 @@ class HloModuleConfigProto(google.protobuf.message.Message): VALUE_FIELD_NUMBER: builtins.int key: builtins.str value: builtins.int - def __init__( - self, - *, - key: builtins.str | None = ..., - value: builtins.int | None = ..., - ) -> None: ... + def __init__(self, *, key: builtins.str | None = ..., value: builtins.int | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["key", b"key", "value", b"value"]) -> None: ... ENTRY_COMPUTATION_LAYOUT_FIELD_NUMBER: builtins.int @@ -1797,11 +2283,17 @@ class HloModuleConfigProto(google.protobuf.message.Message): @property def entry_computation_layout(self) -> tensorflow.compiler.xla.xla_data_pb2.ProgramShapeProto: ... @property - def param_requires_broadcast_via_collectives(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.bool]: ... + def param_requires_broadcast_via_collectives( + self, + ) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.bool]: ... @property - def auto_spmd_partitioning_mesh_shape(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.int]: ... + def auto_spmd_partitioning_mesh_shape( + self, + ) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.int]: ... @property - def auto_spmd_partitioning_mesh_ids(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.int]: ... + def auto_spmd_partitioning_mesh_ids( + self, + ) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.int]: ... @property def debug_options(self) -> global___DebugOptions: ... @property @@ -1814,21 +2306,37 @@ class HloModuleConfigProto(google.protobuf.message.Message): """ @property - def shardable_value_update_pairs(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___ShardableValueUpdatePairProto]: ... + def shardable_value_update_pairs( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___ShardableValueUpdatePairProto]: ... @property - def fusion_config(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___HloModuleConfigProto.BoolList]: ... + def fusion_config( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___HloModuleConfigProto.BoolList]: ... @property - def dot_config(self) -> google.protobuf.internal.containers.MessageMap[builtins.str, global___HloModuleConfigProto.Int64List]: ... + def dot_config( + self, + ) -> google.protobuf.internal.containers.MessageMap[builtins.str, global___HloModuleConfigProto.Int64List]: ... @property - def layout_config(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___HloModuleConfigProto.Int64ListList]: ... + def layout_config( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___HloModuleConfigProto.Int64ListList]: ... @property - def memory_space_assignment_config(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.int]: ... + def memory_space_assignment_config( + self, + ) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.int]: ... @property - def phase_ordering_config(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___HloModuleConfigProto.BoolList]: ... + def phase_ordering_config( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___HloModuleConfigProto.BoolList]: ... @property - def allow_spmd_sharding_propagation_to_parameters(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.bool]: ... + def allow_spmd_sharding_propagation_to_parameters( + self, + ) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.bool]: ... @property - def allow_spmd_sharding_propagation_to_output(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.bool]: ... + def allow_spmd_sharding_propagation_to_output( + self, + ) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.bool]: ... @property def analysis_allowance_map(self) -> google.protobuf.internal.containers.ScalarMap[builtins.str, builtins.int]: ... def __init__( @@ -1869,8 +2377,92 @@ class HloModuleConfigProto(google.protobuf.message.Message): device_memory_size: builtins.int | None = ..., use_shardy_partitioner: builtins.bool | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["debug_options", b"debug_options", "entry_computation_layout", b"entry_computation_layout", "pre_simulation_device_assignment", b"pre_simulation_device_assignment", "static_device_assignment", b"static_device_assignment"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["alias_passthrough_params", b"alias_passthrough_params", "allow_separate_sharding_programs", b"allow_separate_sharding_programs", "allow_spmd_sharding_propagation_to_output", b"allow_spmd_sharding_propagation_to_output", "allow_spmd_sharding_propagation_to_parameters", b"allow_spmd_sharding_propagation_to_parameters", "analysis_allowance_map", b"analysis_allowance_map", "auto_spmd_partitioning_mesh_ids", b"auto_spmd_partitioning_mesh_ids", "auto_spmd_partitioning_mesh_shape", b"auto_spmd_partitioning_mesh_shape", "content_aware_computation_sorting", b"content_aware_computation_sorting", "debug_options", b"debug_options", "deduplicate_hlo", b"deduplicate_hlo", "device_memory_size", b"device_memory_size", "device_type", b"device_type", "dot_config", b"dot_config", "entry_computation_layout", b"entry_computation_layout", "fdo_profile", b"fdo_profile", "fusion_config", b"fusion_config", "fusion_config_collection", b"fusion_config_collection", "intra_op_parallelism_threads", b"intra_op_parallelism_threads", "launch_id", b"launch_id", "layout_config", b"layout_config", "matrix_unit_operand_precision", b"matrix_unit_operand_precision", "memory_space_assignment_config", b"memory_space_assignment_config", "num_partitions", b"num_partitions", "param_requires_broadcast_via_collectives", b"param_requires_broadcast_via_collectives", "phase_index", b"phase_index", "phase_ordering_config", b"phase_ordering_config", "pre_simulation_device_assignment", b"pre_simulation_device_assignment", "replica_count", b"replica_count", "seed", b"seed", "shardable_value_update_pairs", b"shardable_value_update_pairs", "static_device_assignment", b"static_device_assignment", "use_auto_spmd_partitioning", b"use_auto_spmd_partitioning", "use_shardy_partitioner", b"use_shardy_partitioner", "use_spmd_partitioning", b"use_spmd_partitioning"]) -> None: ... + def HasField( + self, + field_name: typing.Literal[ + "debug_options", + b"debug_options", + "entry_computation_layout", + b"entry_computation_layout", + "pre_simulation_device_assignment", + b"pre_simulation_device_assignment", + "static_device_assignment", + b"static_device_assignment", + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "alias_passthrough_params", + b"alias_passthrough_params", + "allow_separate_sharding_programs", + b"allow_separate_sharding_programs", + "allow_spmd_sharding_propagation_to_output", + b"allow_spmd_sharding_propagation_to_output", + "allow_spmd_sharding_propagation_to_parameters", + b"allow_spmd_sharding_propagation_to_parameters", + "analysis_allowance_map", + b"analysis_allowance_map", + "auto_spmd_partitioning_mesh_ids", + b"auto_spmd_partitioning_mesh_ids", + "auto_spmd_partitioning_mesh_shape", + b"auto_spmd_partitioning_mesh_shape", + "content_aware_computation_sorting", + b"content_aware_computation_sorting", + "debug_options", + b"debug_options", + "deduplicate_hlo", + b"deduplicate_hlo", + "device_memory_size", + b"device_memory_size", + "device_type", + b"device_type", + "dot_config", + b"dot_config", + "entry_computation_layout", + b"entry_computation_layout", + "fdo_profile", + b"fdo_profile", + "fusion_config", + b"fusion_config", + "fusion_config_collection", + b"fusion_config_collection", + "intra_op_parallelism_threads", + b"intra_op_parallelism_threads", + "launch_id", + b"launch_id", + "layout_config", + b"layout_config", + "matrix_unit_operand_precision", + b"matrix_unit_operand_precision", + "memory_space_assignment_config", + b"memory_space_assignment_config", + "num_partitions", + b"num_partitions", + "param_requires_broadcast_via_collectives", + b"param_requires_broadcast_via_collectives", + "phase_index", + b"phase_index", + "phase_ordering_config", + b"phase_ordering_config", + "pre_simulation_device_assignment", + b"pre_simulation_device_assignment", + "replica_count", + b"replica_count", + "seed", + b"seed", + "shardable_value_update_pairs", + b"shardable_value_update_pairs", + "static_device_assignment", + b"static_device_assignment", + "use_auto_spmd_partitioning", + b"use_auto_spmd_partitioning", + "use_shardy_partitioner", + b"use_shardy_partitioner", + "use_spmd_partitioning", + b"use_spmd_partitioning", + ], + ) -> None: ... global___HloModuleConfigProto = HloModuleConfigProto @@ -1920,7 +2512,12 @@ class ScheduleProto(google.protobuf.message.Message): start_timestamp_cycles: builtins.float | None = ..., end_timestamp_cycles: builtins.float | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["end_timestamp_cycles", b"end_timestamp_cycles", "id", b"id", "start_timestamp_cycles", b"start_timestamp_cycles"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "end_timestamp_cycles", b"end_timestamp_cycles", "id", b"id", "start_timestamp_cycles", b"start_timestamp_cycles" + ], + ) -> None: ... INSTRUCTIONS_FIELD_NUMBER: builtins.int COMPUTATION_ID_FIELD_NUMBER: builtins.int @@ -1930,7 +2527,9 @@ class ScheduleProto(google.protobuf.message.Message): """Computation id (matches the id in HloComputationProto).""" cycles_per_microsecond: builtins.int @property - def instructions(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___ScheduleProto.Instruction]: ... + def instructions( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___ScheduleProto.Instruction]: ... @property def hlo_module(self) -> tensorflow.compiler.xla.service.hlo_pb2.HloModuleProto: ... def __init__( @@ -1942,6 +2541,18 @@ class ScheduleProto(google.protobuf.message.Message): cycles_per_microsecond: builtins.int | None = ..., ) -> None: ... def HasField(self, field_name: typing.Literal["hlo_module", b"hlo_module"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["computation_id", b"computation_id", "cycles_per_microsecond", b"cycles_per_microsecond", "hlo_module", b"hlo_module", "instructions", b"instructions"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "computation_id", + b"computation_id", + "cycles_per_microsecond", + b"cycles_per_microsecond", + "hlo_module", + b"hlo_module", + "instructions", + b"instructions", + ], + ) -> None: ... global___ScheduleProto = ScheduleProto diff --git a/stubs/tensorflow/tensorflow/core/example/example_parser_configuration_pb2.pyi b/stubs/tensorflow/tensorflow/core/example/example_parser_configuration_pb2.pyi index 6b5c255da3b4..9ffaef3d17a0 100644 --- a/stubs/tensorflow/tensorflow/core/example/example_parser_configuration_pb2.pyi +++ b/stubs/tensorflow/tensorflow/core/example/example_parser_configuration_pb2.pyi @@ -36,7 +36,19 @@ class VarLenFeatureProto(google.protobuf.message.Message): indices_output_tensor_name: builtins.str | None = ..., shapes_output_tensor_name: builtins.str | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["dtype", b"dtype", "indices_output_tensor_name", b"indices_output_tensor_name", "shapes_output_tensor_name", b"shapes_output_tensor_name", "values_output_tensor_name", b"values_output_tensor_name"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "dtype", + b"dtype", + "indices_output_tensor_name", + b"indices_output_tensor_name", + "shapes_output_tensor_name", + b"shapes_output_tensor_name", + "values_output_tensor_name", + b"values_output_tensor_name", + ], + ) -> None: ... global___VarLenFeatureProto = VarLenFeatureProto @@ -63,7 +75,19 @@ class FixedLenFeatureProto(google.protobuf.message.Message): values_output_tensor_name: builtins.str | None = ..., ) -> None: ... def HasField(self, field_name: typing.Literal["default_value", b"default_value", "shape", b"shape"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["default_value", b"default_value", "dtype", b"dtype", "shape", b"shape", "values_output_tensor_name", b"values_output_tensor_name"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "default_value", + b"default_value", + "dtype", + b"dtype", + "shape", + b"shape", + "values_output_tensor_name", + b"values_output_tensor_name", + ], + ) -> None: ... global___FixedLenFeatureProto = FixedLenFeatureProto @@ -83,9 +107,21 @@ class FeatureConfiguration(google.protobuf.message.Message): fixed_len_feature: global___FixedLenFeatureProto | None = ..., var_len_feature: global___VarLenFeatureProto | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["config", b"config", "fixed_len_feature", b"fixed_len_feature", "var_len_feature", b"var_len_feature"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["config", b"config", "fixed_len_feature", b"fixed_len_feature", "var_len_feature", b"var_len_feature"]) -> None: ... - def WhichOneof(self, oneof_group: typing.Literal["config", b"config"]) -> typing.Literal["fixed_len_feature", "var_len_feature"] | None: ... + def HasField( + self, + field_name: typing.Literal[ + "config", b"config", "fixed_len_feature", b"fixed_len_feature", "var_len_feature", b"var_len_feature" + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "config", b"config", "fixed_len_feature", b"fixed_len_feature", "var_len_feature", b"var_len_feature" + ], + ) -> None: ... + def WhichOneof( + self, oneof_group: typing.Literal["config", b"config"] + ) -> typing.Literal["fixed_len_feature", "var_len_feature"] | None: ... global___FeatureConfiguration = FeatureConfiguration @@ -102,12 +138,7 @@ class ExampleParserConfiguration(google.protobuf.message.Message): key: builtins.str @property def value(self) -> global___FeatureConfiguration: ... - def __init__( - self, - *, - key: builtins.str | None = ..., - value: global___FeatureConfiguration | None = ..., - ) -> None: ... + def __init__(self, *, key: builtins.str | None = ..., value: global___FeatureConfiguration | None = ...) -> None: ... def HasField(self, field_name: typing.Literal["value", b"value"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["key", b"key", "value", b"value"]) -> None: ... @@ -115,9 +146,7 @@ class ExampleParserConfiguration(google.protobuf.message.Message): @property def feature_map(self) -> google.protobuf.internal.containers.MessageMap[builtins.str, global___FeatureConfiguration]: ... def __init__( - self, - *, - feature_map: collections.abc.Mapping[builtins.str, global___FeatureConfiguration] | None = ..., + self, *, feature_map: collections.abc.Mapping[builtins.str, global___FeatureConfiguration] | None = ... ) -> None: ... def ClearField(self, field_name: typing.Literal["feature_map", b"feature_map"]) -> None: ... diff --git a/stubs/tensorflow/tensorflow/core/example/example_pb2.pyi b/stubs/tensorflow/tensorflow/core/example/example_pb2.pyi index 562a6c44cf55..63cf2c14f17a 100644 --- a/stubs/tensorflow/tensorflow/core/example/example_pb2.pyi +++ b/stubs/tensorflow/tensorflow/core/example/example_pb2.pyi @@ -95,11 +95,7 @@ class Example(google.protobuf.message.Message): FEATURES_FIELD_NUMBER: builtins.int @property def features(self) -> tensorflow.core.example.feature_pb2.Features: ... - def __init__( - self, - *, - features: tensorflow.core.example.feature_pb2.Features | None = ..., - ) -> None: ... + def __init__(self, *, features: tensorflow.core.example.feature_pb2.Features | None = ...) -> None: ... def HasField(self, field_name: typing.Literal["features", b"features"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["features", b"features"]) -> None: ... diff --git a/stubs/tensorflow/tensorflow/core/example/feature_pb2.pyi b/stubs/tensorflow/tensorflow/core/example/feature_pb2.pyi index 3fbc881cbaa2..9bec4dd165e0 100644 --- a/stubs/tensorflow/tensorflow/core/example/feature_pb2.pyi +++ b/stubs/tensorflow/tensorflow/core/example/feature_pb2.pyi @@ -77,11 +77,7 @@ class BytesList(google.protobuf.message.Message): VALUE_FIELD_NUMBER: builtins.int @property def value(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.bytes]: ... - def __init__( - self, - *, - value: collections.abc.Iterable[builtins.bytes] | None = ..., - ) -> None: ... + def __init__(self, *, value: collections.abc.Iterable[builtins.bytes] | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["value", b"value"]) -> None: ... global___BytesList = BytesList @@ -93,11 +89,7 @@ class FloatList(google.protobuf.message.Message): VALUE_FIELD_NUMBER: builtins.int @property def value(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.float]: ... - def __init__( - self, - *, - value: collections.abc.Iterable[builtins.float] | None = ..., - ) -> None: ... + def __init__(self, *, value: collections.abc.Iterable[builtins.float] | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["value", b"value"]) -> None: ... global___FloatList = FloatList @@ -109,11 +101,7 @@ class Int64List(google.protobuf.message.Message): VALUE_FIELD_NUMBER: builtins.int @property def value(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.int]: ... - def __init__( - self, - *, - value: collections.abc.Iterable[builtins.int] | None = ..., - ) -> None: ... + def __init__(self, *, value: collections.abc.Iterable[builtins.int] | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["value", b"value"]) -> None: ... global___Int64List = Int64List @@ -140,9 +128,21 @@ class Feature(google.protobuf.message.Message): float_list: global___FloatList | None = ..., int64_list: global___Int64List | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["bytes_list", b"bytes_list", "float_list", b"float_list", "int64_list", b"int64_list", "kind", b"kind"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["bytes_list", b"bytes_list", "float_list", b"float_list", "int64_list", b"int64_list", "kind", b"kind"]) -> None: ... - def WhichOneof(self, oneof_group: typing.Literal["kind", b"kind"]) -> typing.Literal["bytes_list", "float_list", "int64_list"] | None: ... + def HasField( + self, + field_name: typing.Literal[ + "bytes_list", b"bytes_list", "float_list", b"float_list", "int64_list", b"int64_list", "kind", b"kind" + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "bytes_list", b"bytes_list", "float_list", b"float_list", "int64_list", b"int64_list", "kind", b"kind" + ], + ) -> None: ... + def WhichOneof( + self, oneof_group: typing.Literal["kind", b"kind"] + ) -> typing.Literal["bytes_list", "float_list", "int64_list"] | None: ... global___Feature = Feature @@ -159,12 +159,7 @@ class Features(google.protobuf.message.Message): key: builtins.str @property def value(self) -> global___Feature: ... - def __init__( - self, - *, - key: builtins.str | None = ..., - value: global___Feature | None = ..., - ) -> None: ... + def __init__(self, *, key: builtins.str | None = ..., value: global___Feature | None = ...) -> None: ... def HasField(self, field_name: typing.Literal["value", b"value"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["key", b"key", "value", b"value"]) -> None: ... @@ -173,11 +168,7 @@ class Features(google.protobuf.message.Message): def feature(self) -> google.protobuf.internal.containers.MessageMap[builtins.str, global___Feature]: """Map from feature name to feature.""" - def __init__( - self, - *, - feature: collections.abc.Mapping[builtins.str, global___Feature] | None = ..., - ) -> None: ... + def __init__(self, *, feature: collections.abc.Mapping[builtins.str, global___Feature] | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["feature", b"feature"]) -> None: ... global___Features = Features @@ -198,11 +189,7 @@ class FeatureList(google.protobuf.message.Message): FEATURE_FIELD_NUMBER: builtins.int @property def feature(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___Feature]: ... - def __init__( - self, - *, - feature: collections.abc.Iterable[global___Feature] | None = ..., - ) -> None: ... + def __init__(self, *, feature: collections.abc.Iterable[global___Feature] | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["feature", b"feature"]) -> None: ... global___FeatureList = FeatureList @@ -220,12 +207,7 @@ class FeatureLists(google.protobuf.message.Message): key: builtins.str @property def value(self) -> global___FeatureList: ... - def __init__( - self, - *, - key: builtins.str | None = ..., - value: global___FeatureList | None = ..., - ) -> None: ... + def __init__(self, *, key: builtins.str | None = ..., value: global___FeatureList | None = ...) -> None: ... def HasField(self, field_name: typing.Literal["value", b"value"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["key", b"key", "value", b"value"]) -> None: ... @@ -234,11 +216,7 @@ class FeatureLists(google.protobuf.message.Message): def feature_list(self) -> google.protobuf.internal.containers.MessageMap[builtins.str, global___FeatureList]: """Map from feature name to feature list.""" - def __init__( - self, - *, - feature_list: collections.abc.Mapping[builtins.str, global___FeatureList] | None = ..., - ) -> None: ... + def __init__(self, *, feature_list: collections.abc.Mapping[builtins.str, global___FeatureList] | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["feature_list", b"feature_list"]) -> None: ... global___FeatureLists = FeatureLists diff --git a/stubs/tensorflow/tensorflow/core/framework/allocation_description_pb2.pyi b/stubs/tensorflow/tensorflow/core/framework/allocation_description_pb2.pyi index 6e0f0cb22534..9f4e541f2299 100644 --- a/stubs/tensorflow/tensorflow/core/framework/allocation_description_pb2.pyi +++ b/stubs/tensorflow/tensorflow/core/framework/allocation_description_pb2.pyi @@ -43,6 +43,22 @@ class AllocationDescription(google.protobuf.message.Message): has_single_reference: builtins.bool | None = ..., ptr: builtins.int | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["allocated_bytes", b"allocated_bytes", "allocation_id", b"allocation_id", "allocator_name", b"allocator_name", "has_single_reference", b"has_single_reference", "ptr", b"ptr", "requested_bytes", b"requested_bytes"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "allocated_bytes", + b"allocated_bytes", + "allocation_id", + b"allocation_id", + "allocator_name", + b"allocator_name", + "has_single_reference", + b"has_single_reference", + "ptr", + b"ptr", + "requested_bytes", + b"requested_bytes", + ], + ) -> None: ... global___AllocationDescription = AllocationDescription diff --git a/stubs/tensorflow/tensorflow/core/framework/api_def_pb2.pyi b/stubs/tensorflow/tensorflow/core/framework/api_def_pb2.pyi index 30f81032a88d..b3d8c424234f 100644 --- a/stubs/tensorflow/tensorflow/core/framework/api_def_pb2.pyi +++ b/stubs/tensorflow/tensorflow/core/framework/api_def_pb2.pyi @@ -49,7 +49,9 @@ class ApiDef(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _VisibilityEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[ApiDef._Visibility.ValueType], builtins.type): + class _VisibilityEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[ApiDef._Visibility.ValueType], builtins.type + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor DEFAULT_VISIBILITY: ApiDef._Visibility.ValueType # 0 """Normally this is "VISIBLE" unless you are inheriting a @@ -117,7 +119,12 @@ class ApiDef(google.protobuf.message.Message): deprecated: builtins.bool | None = ..., deprecation_version: builtins.int | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["deprecated", b"deprecated", "deprecation_version", b"deprecation_version", "name", b"name"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "deprecated", b"deprecated", "deprecation_version", b"deprecation_version", "name", b"name" + ], + ) -> None: ... @typing.final class Arg(google.protobuf.message.Message): @@ -138,13 +145,11 @@ class ApiDef(google.protobuf.message.Message): them entirely) as can be done with op descriptions. """ def __init__( - self, - *, - name: builtins.str | None = ..., - rename_to: builtins.str | None = ..., - description: builtins.str | None = ..., + self, *, name: builtins.str | None = ..., rename_to: builtins.str | None = ..., description: builtins.str | None = ... + ) -> None: ... + def ClearField( + self, field_name: typing.Literal["description", b"description", "name", b"name", "rename_to", b"rename_to"] ) -> None: ... - def ClearField(self, field_name: typing.Literal["description", b"description", "name", b"name", "rename_to", b"rename_to"]) -> None: ... @typing.final class Attr(google.protobuf.message.Message): @@ -186,7 +191,12 @@ class ApiDef(google.protobuf.message.Message): description: builtins.str | None = ..., ) -> None: ... def HasField(self, field_name: typing.Literal["default_value", b"default_value"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["default_value", b"default_value", "description", b"description", "name", b"name", "rename_to", b"rename_to"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "default_value", b"default_value", "description", b"description", "name", b"name", "rename_to", b"rename_to" + ], + ) -> None: ... GRAPH_OP_NAME_FIELD_NUMBER: builtins.int DEPRECATION_MESSAGE_FIELD_NUMBER: builtins.int @@ -255,7 +265,37 @@ class ApiDef(google.protobuf.message.Message): description_prefix: builtins.str | None = ..., description_suffix: builtins.str | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["arg_order", b"arg_order", "attr", b"attr", "deprecation_message", b"deprecation_message", "deprecation_version", b"deprecation_version", "description", b"description", "description_prefix", b"description_prefix", "description_suffix", b"description_suffix", "endpoint", b"endpoint", "graph_op_name", b"graph_op_name", "in_arg", b"in_arg", "out_arg", b"out_arg", "summary", b"summary", "visibility", b"visibility"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "arg_order", + b"arg_order", + "attr", + b"attr", + "deprecation_message", + b"deprecation_message", + "deprecation_version", + b"deprecation_version", + "description", + b"description", + "description_prefix", + b"description_prefix", + "description_suffix", + b"description_suffix", + "endpoint", + b"endpoint", + "graph_op_name", + b"graph_op_name", + "in_arg", + b"in_arg", + "out_arg", + b"out_arg", + "summary", + b"summary", + "visibility", + b"visibility", + ], + ) -> None: ... global___ApiDef = ApiDef @@ -266,11 +306,7 @@ class ApiDefs(google.protobuf.message.Message): OP_FIELD_NUMBER: builtins.int @property def op(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___ApiDef]: ... - def __init__( - self, - *, - op: collections.abc.Iterable[global___ApiDef] | None = ..., - ) -> None: ... + def __init__(self, *, op: collections.abc.Iterable[global___ApiDef] | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["op", b"op"]) -> None: ... global___ApiDefs = ApiDefs diff --git a/stubs/tensorflow/tensorflow/core/framework/attr_value_pb2.pyi b/stubs/tensorflow/tensorflow/core/framework/attr_value_pb2.pyi index 80bdde459e0f..bc36030c511d 100644 --- a/stubs/tensorflow/tensorflow/core/framework/attr_value_pb2.pyi +++ b/stubs/tensorflow/tensorflow/core/framework/attr_value_pb2.pyi @@ -41,35 +41,47 @@ class AttrValue(google.protobuf.message.Message): FUNC_FIELD_NUMBER: builtins.int @property def s(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.bytes]: - """"list(string)" """ + """ "list(string)" """ @property def i(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.int]: - """"list(int)" """ + """ "list(int)" """ @property def f(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.float]: - """"list(float)" """ + """ "list(float)" """ @property def b(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.bool]: - """"list(bool)" """ + """ "list(bool)" """ @property - def type(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[tensorflow.core.framework.types_pb2.DataType.ValueType]: - """"list(type)" """ + def type( + self, + ) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[ + tensorflow.core.framework.types_pb2.DataType.ValueType + ]: + """ "list(type)" """ @property - def shape(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[tensorflow.core.framework.tensor_shape_pb2.TensorShapeProto]: - """"list(shape)" """ + def shape( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[ + tensorflow.core.framework.tensor_shape_pb2.TensorShapeProto + ]: + """ "list(shape)" """ @property - def tensor(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[tensorflow.core.framework.tensor_pb2.TensorProto]: - """"list(tensor)" """ + def tensor( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[ + tensorflow.core.framework.tensor_pb2.TensorProto + ]: + """ "list(tensor)" """ @property def func(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___NameAttrList]: - """"list(attr)" """ + """ "list(attr)" """ def __init__( self, @@ -83,7 +95,27 @@ class AttrValue(google.protobuf.message.Message): tensor: collections.abc.Iterable[tensorflow.core.framework.tensor_pb2.TensorProto] | None = ..., func: collections.abc.Iterable[global___NameAttrList] | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["b", b"b", "f", b"f", "func", b"func", "i", b"i", "s", b"s", "shape", b"shape", "tensor", b"tensor", "type", b"type"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "b", + b"b", + "f", + b"f", + "func", + b"func", + "i", + b"i", + "s", + b"s", + "shape", + b"shape", + "tensor", + b"tensor", + "type", + b"type", + ], + ) -> None: ... S_FIELD_NUMBER: builtins.int I_FIELD_NUMBER: builtins.int @@ -116,11 +148,11 @@ class AttrValue(google.protobuf.message.Message): """ @property def shape(self) -> tensorflow.core.framework.tensor_shape_pb2.TensorShapeProto: - """"shape" """ + """ "shape" """ @property def tensor(self) -> tensorflow.core.framework.tensor_pb2.TensorProto: - """"tensor" """ + """ "tensor" """ @property def list(self) -> global___AttrValue.ListValue: @@ -128,7 +160,7 @@ class AttrValue(google.protobuf.message.Message): @property def func(self) -> global___NameAttrList: - """"func" represents a function. func.name is a function's name or + """ "func" represents a function. func.name is a function's name or a primitive op's name. func.attr.first is the name of an attr defined for that function. func.attr.second is the value for that attr in the instantiation. @@ -148,9 +180,63 @@ class AttrValue(google.protobuf.message.Message): func: global___NameAttrList | None = ..., placeholder: builtins.str | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["b", b"b", "f", b"f", "func", b"func", "i", b"i", "list", b"list", "placeholder", b"placeholder", "s", b"s", "shape", b"shape", "tensor", b"tensor", "type", b"type", "value", b"value"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["b", b"b", "f", b"f", "func", b"func", "i", b"i", "list", b"list", "placeholder", b"placeholder", "s", b"s", "shape", b"shape", "tensor", b"tensor", "type", b"type", "value", b"value"]) -> None: ... - def WhichOneof(self, oneof_group: typing.Literal["value", b"value"]) -> typing.Literal["s", "i", "f", "b", "type", "shape", "tensor", "list", "func", "placeholder"] | None: ... + def HasField( + self, + field_name: typing.Literal[ + "b", + b"b", + "f", + b"f", + "func", + b"func", + "i", + b"i", + "list", + b"list", + "placeholder", + b"placeholder", + "s", + b"s", + "shape", + b"shape", + "tensor", + b"tensor", + "type", + b"type", + "value", + b"value", + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "b", + b"b", + "f", + b"f", + "func", + b"func", + "i", + b"i", + "list", + b"list", + "placeholder", + b"placeholder", + "s", + b"s", + "shape", + b"shape", + "tensor", + b"tensor", + "type", + b"type", + "value", + b"value", + ], + ) -> None: ... + def WhichOneof( + self, oneof_group: typing.Literal["value", b"value"] + ) -> typing.Literal["s", "i", "f", "b", "type", "shape", "tensor", "list", "func", "placeholder"] | None: ... global___AttrValue = AttrValue @@ -171,12 +257,7 @@ class NameAttrList(google.protobuf.message.Message): key: builtins.str @property def value(self) -> global___AttrValue: ... - def __init__( - self, - *, - key: builtins.str | None = ..., - value: global___AttrValue | None = ..., - ) -> None: ... + def __init__(self, *, key: builtins.str | None = ..., value: global___AttrValue | None = ...) -> None: ... def HasField(self, field_name: typing.Literal["value", b"value"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["key", b"key", "value", b"value"]) -> None: ... @@ -186,10 +267,7 @@ class NameAttrList(google.protobuf.message.Message): @property def attr(self) -> google.protobuf.internal.containers.MessageMap[builtins.str, global___AttrValue]: ... def __init__( - self, - *, - name: builtins.str | None = ..., - attr: collections.abc.Mapping[builtins.str, global___AttrValue] | None = ..., + self, *, name: builtins.str | None = ..., attr: collections.abc.Mapping[builtins.str, global___AttrValue] | None = ... ) -> None: ... def ClearField(self, field_name: typing.Literal["attr", b"attr", "name", b"name"]) -> None: ... diff --git a/stubs/tensorflow/tensorflow/core/framework/cost_graph_pb2.pyi b/stubs/tensorflow/tensorflow/core/framework/cost_graph_pb2.pyi index d54c0196320a..3ba7d01628bf 100644 --- a/stubs/tensorflow/tensorflow/core/framework/cost_graph_pb2.pyi +++ b/stubs/tensorflow/tensorflow/core/framework/cost_graph_pb2.pyi @@ -37,12 +37,11 @@ class CostGraphDef(google.protobuf.message.Message): preceding_node: builtins.int preceding_port: builtins.int def __init__( - self, - *, - preceding_node: builtins.int | None = ..., - preceding_port: builtins.int | None = ..., + self, *, preceding_node: builtins.int | None = ..., preceding_port: builtins.int | None = ... + ) -> None: ... + def ClearField( + self, field_name: typing.Literal["preceding_node", b"preceding_node", "preceding_port", b"preceding_port"] ) -> None: ... - def ClearField(self, field_name: typing.Literal["preceding_node", b"preceding_node", "preceding_port", b"preceding_port"]) -> None: ... @typing.final class OutputInfo(google.protobuf.message.Message): @@ -72,7 +71,12 @@ class CostGraphDef(google.protobuf.message.Message): dtype: tensorflow.core.framework.types_pb2.DataType.ValueType | None = ..., ) -> None: ... def HasField(self, field_name: typing.Literal["shape", b"shape"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["alias_input_port", b"alias_input_port", "dtype", b"dtype", "shape", b"shape", "size", b"size"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "alias_input_port", b"alias_input_port", "dtype", b"dtype", "shape", b"shape", "size", b"size" + ], + ) -> None: ... NAME_FIELD_NUMBER: builtins.int DEVICE_FIELD_NUMBER: builtins.int @@ -122,9 +126,13 @@ class CostGraphDef(google.protobuf.message.Message): inaccurate: builtins.bool """Are the costs inaccurate?""" @property - def input_info(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___CostGraphDef.Node.InputInfo]: ... + def input_info( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___CostGraphDef.Node.InputInfo]: ... @property - def output_info(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___CostGraphDef.Node.OutputInfo]: ... + def output_info( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___CostGraphDef.Node.OutputInfo]: ... @property def control_input(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.int]: """Ids of the control inputs for this node.""" @@ -149,7 +157,43 @@ class CostGraphDef(google.protobuf.message.Message): control_input: collections.abc.Iterable[builtins.int] | None = ..., inaccurate: builtins.bool | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["compute_cost", b"compute_cost", "compute_time", b"compute_time", "control_input", b"control_input", "device", b"device", "device_persistent_memory_size", b"device_persistent_memory_size", "device_temp_memory_size", b"device_temp_memory_size", "host_temp_memory_size", b"host_temp_memory_size", "id", b"id", "inaccurate", b"inaccurate", "input_info", b"input_info", "is_final", b"is_final", "memory_time", b"memory_time", "name", b"name", "output_info", b"output_info", "persistent_memory_size", b"persistent_memory_size", "temporary_memory_size", b"temporary_memory_size"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "compute_cost", + b"compute_cost", + "compute_time", + b"compute_time", + "control_input", + b"control_input", + "device", + b"device", + "device_persistent_memory_size", + b"device_persistent_memory_size", + "device_temp_memory_size", + b"device_temp_memory_size", + "host_temp_memory_size", + b"host_temp_memory_size", + "id", + b"id", + "inaccurate", + b"inaccurate", + "input_info", + b"input_info", + "is_final", + b"is_final", + "memory_time", + b"memory_time", + "name", + b"name", + "output_info", + b"output_info", + "persistent_memory_size", + b"persistent_memory_size", + "temporary_memory_size", + b"temporary_memory_size", + ], + ) -> None: ... @typing.final class AggregatedCost(google.protobuf.message.Message): @@ -163,12 +207,7 @@ class CostGraphDef(google.protobuf.message.Message): """Aggregated cost value.""" dimension: builtins.str """Aggregated cost dimension (e.g. 'memory', 'compute', 'network').""" - def __init__( - self, - *, - cost: builtins.float | None = ..., - dimension: builtins.str | None = ..., - ) -> None: ... + def __init__(self, *, cost: builtins.float | None = ..., dimension: builtins.str | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["cost", b"cost", "dimension", b"dimension"]) -> None: ... NODE_FIELD_NUMBER: builtins.int @@ -176,7 +215,9 @@ class CostGraphDef(google.protobuf.message.Message): @property def node(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___CostGraphDef.Node]: ... @property - def cost(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___CostGraphDef.AggregatedCost]: ... + def cost( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___CostGraphDef.AggregatedCost]: ... def __init__( self, *, diff --git a/stubs/tensorflow/tensorflow/core/framework/cpp_shape_inference_pb2.pyi b/stubs/tensorflow/tensorflow/core/framework/cpp_shape_inference_pb2.pyi index 070c7c3cdb63..1bf1c41641cc 100644 --- a/stubs/tensorflow/tensorflow/core/framework/cpp_shape_inference_pb2.pyi +++ b/stubs/tensorflow/tensorflow/core/framework/cpp_shape_inference_pb2.pyi @@ -50,7 +50,11 @@ class CppShapeInferenceResult(google.protobuf.message.Message): SHAPE_AND_TYPE_FIELD_NUMBER: builtins.int is_set: builtins.bool @property - def shape_and_type(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___CppShapeInferenceResult.HandleShapeAndType]: + def shape_and_type( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[ + global___CppShapeInferenceResult.HandleShapeAndType + ]: """Only valid if .""" def __init__( @@ -87,13 +91,20 @@ class CppShapeInferenceInputsNeeded(google.protobuf.message.Message): @property def input_tensors_needed(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.int]: ... @property - def input_tensors_as_shapes_needed(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.int]: ... + def input_tensors_as_shapes_needed( + self, + ) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.int]: ... def __init__( self, *, input_tensors_needed: collections.abc.Iterable[builtins.int] | None = ..., input_tensors_as_shapes_needed: collections.abc.Iterable[builtins.int] | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["input_tensors_as_shapes_needed", b"input_tensors_as_shapes_needed", "input_tensors_needed", b"input_tensors_needed"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "input_tensors_as_shapes_needed", b"input_tensors_as_shapes_needed", "input_tensors_needed", b"input_tensors_needed" + ], + ) -> None: ... global___CppShapeInferenceInputsNeeded = CppShapeInferenceInputsNeeded diff --git a/stubs/tensorflow/tensorflow/core/framework/dataset_metadata_pb2.pyi b/stubs/tensorflow/tensorflow/core/framework/dataset_metadata_pb2.pyi index a1c7b19143c5..afde78f967f6 100644 --- a/stubs/tensorflow/tensorflow/core/framework/dataset_metadata_pb2.pyi +++ b/stubs/tensorflow/tensorflow/core/framework/dataset_metadata_pb2.pyi @@ -19,11 +19,7 @@ class Metadata(google.protobuf.message.Message): NAME_FIELD_NUMBER: builtins.int name: builtins.bytes - def __init__( - self, - *, - name: builtins.bytes | None = ..., - ) -> None: ... + def __init__(self, *, name: builtins.bytes | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["name", b"name"]) -> None: ... global___Metadata = Metadata diff --git a/stubs/tensorflow/tensorflow/core/framework/dataset_options_pb2.pyi b/stubs/tensorflow/tensorflow/core/framework/dataset_options_pb2.pyi index 5837446a1e77..724249f3c540 100644 --- a/stubs/tensorflow/tensorflow/core/framework/dataset_options_pb2.pyi +++ b/stubs/tensorflow/tensorflow/core/framework/dataset_options_pb2.pyi @@ -25,7 +25,9 @@ class _AutoShardPolicy: ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType -class _AutoShardPolicyEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_AutoShardPolicy.ValueType], builtins.type): +class _AutoShardPolicyEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_AutoShardPolicy.ValueType], builtins.type +): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor AUTO: _AutoShardPolicy.ValueType # 0 """AUTO: Attempts FILE-based sharding, falling back to DATA-based sharding.""" @@ -77,7 +79,9 @@ class _ExternalStatePolicy: ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType -class _ExternalStatePolicyEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_ExternalStatePolicy.ValueType], builtins.type): +class _ExternalStatePolicyEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_ExternalStatePolicy.ValueType], builtins.type +): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor POLICY_WARN: _ExternalStatePolicy.ValueType # 0 POLICY_IGNORE: _ExternalStatePolicy.ValueType # 1 @@ -116,18 +120,76 @@ class AutotuneOptions(google.protobuf.message.Message): autotune_algorithm: tensorflow.core.framework.model_pb2.AutotuneAlgorithm.ValueType | None = ..., initial_parallelism: builtins.int | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["autotune_algorithm", b"autotune_algorithm", "cpu_budget", b"cpu_budget", "enabled", b"enabled", "initial_parallelism", b"initial_parallelism", "optional_autotune_algorithm", b"optional_autotune_algorithm", "optional_cpu_budget", b"optional_cpu_budget", "optional_enabled", b"optional_enabled", "optional_initial_parallelism", b"optional_initial_parallelism", "optional_ram_budget", b"optional_ram_budget", "ram_budget", b"ram_budget"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["autotune_algorithm", b"autotune_algorithm", "cpu_budget", b"cpu_budget", "enabled", b"enabled", "initial_parallelism", b"initial_parallelism", "optional_autotune_algorithm", b"optional_autotune_algorithm", "optional_cpu_budget", b"optional_cpu_budget", "optional_enabled", b"optional_enabled", "optional_initial_parallelism", b"optional_initial_parallelism", "optional_ram_budget", b"optional_ram_budget", "ram_budget", b"ram_budget"]) -> None: ... + def HasField( + self, + field_name: typing.Literal[ + "autotune_algorithm", + b"autotune_algorithm", + "cpu_budget", + b"cpu_budget", + "enabled", + b"enabled", + "initial_parallelism", + b"initial_parallelism", + "optional_autotune_algorithm", + b"optional_autotune_algorithm", + "optional_cpu_budget", + b"optional_cpu_budget", + "optional_enabled", + b"optional_enabled", + "optional_initial_parallelism", + b"optional_initial_parallelism", + "optional_ram_budget", + b"optional_ram_budget", + "ram_budget", + b"ram_budget", + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "autotune_algorithm", + b"autotune_algorithm", + "cpu_budget", + b"cpu_budget", + "enabled", + b"enabled", + "initial_parallelism", + b"initial_parallelism", + "optional_autotune_algorithm", + b"optional_autotune_algorithm", + "optional_cpu_budget", + b"optional_cpu_budget", + "optional_enabled", + b"optional_enabled", + "optional_initial_parallelism", + b"optional_initial_parallelism", + "optional_ram_budget", + b"optional_ram_budget", + "ram_budget", + b"ram_budget", + ], + ) -> None: ... @typing.overload - def WhichOneof(self, oneof_group: typing.Literal["optional_autotune_algorithm", b"optional_autotune_algorithm"]) -> typing.Literal["autotune_algorithm"] | None: ... + def WhichOneof( + self, oneof_group: typing.Literal["optional_autotune_algorithm", b"optional_autotune_algorithm"] + ) -> typing.Literal["autotune_algorithm"] | None: ... @typing.overload - def WhichOneof(self, oneof_group: typing.Literal["optional_cpu_budget", b"optional_cpu_budget"]) -> typing.Literal["cpu_budget"] | None: ... + def WhichOneof( + self, oneof_group: typing.Literal["optional_cpu_budget", b"optional_cpu_budget"] + ) -> typing.Literal["cpu_budget"] | None: ... @typing.overload - def WhichOneof(self, oneof_group: typing.Literal["optional_enabled", b"optional_enabled"]) -> typing.Literal["enabled"] | None: ... + def WhichOneof( + self, oneof_group: typing.Literal["optional_enabled", b"optional_enabled"] + ) -> typing.Literal["enabled"] | None: ... @typing.overload - def WhichOneof(self, oneof_group: typing.Literal["optional_initial_parallelism", b"optional_initial_parallelism"]) -> typing.Literal["initial_parallelism"] | None: ... + def WhichOneof( + self, oneof_group: typing.Literal["optional_initial_parallelism", b"optional_initial_parallelism"] + ) -> typing.Literal["initial_parallelism"] | None: ... @typing.overload - def WhichOneof(self, oneof_group: typing.Literal["optional_ram_budget", b"optional_ram_budget"]) -> typing.Literal["ram_budget"] | None: ... + def WhichOneof( + self, oneof_group: typing.Literal["optional_ram_budget", b"optional_ram_budget"] + ) -> typing.Literal["ram_budget"] | None: ... global___AutotuneOptions = AutotuneOptions @@ -141,7 +203,9 @@ class CardinalityOptions(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _ComputeLevelEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[CardinalityOptions._ComputeLevel.ValueType], builtins.type): + class _ComputeLevelEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[CardinalityOptions._ComputeLevel.ValueType], builtins.type + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor CARDINALITY_COMPUTE_UNSPECIFIED: CardinalityOptions._ComputeLevel.ValueType # 0 CARDINALITY_COMPUTE_LOW: CardinalityOptions._ComputeLevel.ValueType # 1 @@ -172,11 +236,7 @@ class CardinalityOptions(google.protobuf.message.Message): COMPUTE_LEVEL_FIELD_NUMBER: builtins.int compute_level: global___CardinalityOptions.ComputeLevel.ValueType - def __init__( - self, - *, - compute_level: global___CardinalityOptions.ComputeLevel.ValueType | None = ..., - ) -> None: ... + def __init__(self, *, compute_level: global___CardinalityOptions.ComputeLevel.ValueType | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["compute_level", b"compute_level"]) -> None: ... global___CardinalityOptions = CardinalityOptions @@ -192,14 +252,25 @@ class DistributeOptions(google.protobuf.message.Message): auto_shard_policy: global___AutoShardPolicy.ValueType num_devices: builtins.int def __init__( + self, *, auto_shard_policy: global___AutoShardPolicy.ValueType | None = ..., num_devices: builtins.int | None = ... + ) -> None: ... + def HasField( + self, field_name: typing.Literal["num_devices", b"num_devices", "optional_num_devices", b"optional_num_devices"] + ) -> builtins.bool: ... + def ClearField( self, - *, - auto_shard_policy: global___AutoShardPolicy.ValueType | None = ..., - num_devices: builtins.int | None = ..., + field_name: typing.Literal[ + "auto_shard_policy", + b"auto_shard_policy", + "num_devices", + b"num_devices", + "optional_num_devices", + b"optional_num_devices", + ], ) -> None: ... - def HasField(self, field_name: typing.Literal["num_devices", b"num_devices", "optional_num_devices", b"optional_num_devices"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["auto_shard_policy", b"auto_shard_policy", "num_devices", b"num_devices", "optional_num_devices", b"optional_num_devices"]) -> None: ... - def WhichOneof(self, oneof_group: typing.Literal["optional_num_devices", b"optional_num_devices"]) -> typing.Literal["num_devices"] | None: ... + def WhichOneof( + self, oneof_group: typing.Literal["optional_num_devices", b"optional_num_devices"] + ) -> typing.Literal["num_devices"] | None: ... global___DistributeOptions = DistributeOptions @@ -249,32 +320,160 @@ class OptimizationOptions(google.protobuf.message.Message): inject_prefetch: builtins.bool | None = ..., seq_interleave_prefetch: builtins.bool | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["apply_default_optimizations", b"apply_default_optimizations", "filter_fusion", b"filter_fusion", "filter_parallelization", b"filter_parallelization", "inject_prefetch", b"inject_prefetch", "map_and_batch_fusion", b"map_and_batch_fusion", "map_and_filter_fusion", b"map_and_filter_fusion", "map_fusion", b"map_fusion", "map_parallelization", b"map_parallelization", "noop_elimination", b"noop_elimination", "optional_apply_default_optimizations", b"optional_apply_default_optimizations", "optional_filter_fusion", b"optional_filter_fusion", "optional_filter_parallelization", b"optional_filter_parallelization", "optional_inject_prefetch", b"optional_inject_prefetch", "optional_map_and_batch_fusion", b"optional_map_and_batch_fusion", "optional_map_and_filter_fusion", b"optional_map_and_filter_fusion", "optional_map_fusion", b"optional_map_fusion", "optional_map_parallelization", b"optional_map_parallelization", "optional_noop_elimination", b"optional_noop_elimination", "optional_parallel_batch", b"optional_parallel_batch", "optional_seq_interleave_prefetch", b"optional_seq_interleave_prefetch", "optional_shuffle_and_repeat_fusion", b"optional_shuffle_and_repeat_fusion", "parallel_batch", b"parallel_batch", "seq_interleave_prefetch", b"seq_interleave_prefetch", "shuffle_and_repeat_fusion", b"shuffle_and_repeat_fusion"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["apply_default_optimizations", b"apply_default_optimizations", "filter_fusion", b"filter_fusion", "filter_parallelization", b"filter_parallelization", "inject_prefetch", b"inject_prefetch", "map_and_batch_fusion", b"map_and_batch_fusion", "map_and_filter_fusion", b"map_and_filter_fusion", "map_fusion", b"map_fusion", "map_parallelization", b"map_parallelization", "noop_elimination", b"noop_elimination", "optional_apply_default_optimizations", b"optional_apply_default_optimizations", "optional_filter_fusion", b"optional_filter_fusion", "optional_filter_parallelization", b"optional_filter_parallelization", "optional_inject_prefetch", b"optional_inject_prefetch", "optional_map_and_batch_fusion", b"optional_map_and_batch_fusion", "optional_map_and_filter_fusion", b"optional_map_and_filter_fusion", "optional_map_fusion", b"optional_map_fusion", "optional_map_parallelization", b"optional_map_parallelization", "optional_noop_elimination", b"optional_noop_elimination", "optional_parallel_batch", b"optional_parallel_batch", "optional_seq_interleave_prefetch", b"optional_seq_interleave_prefetch", "optional_shuffle_and_repeat_fusion", b"optional_shuffle_and_repeat_fusion", "parallel_batch", b"parallel_batch", "seq_interleave_prefetch", b"seq_interleave_prefetch", "shuffle_and_repeat_fusion", b"shuffle_and_repeat_fusion"]) -> None: ... + def HasField( + self, + field_name: typing.Literal[ + "apply_default_optimizations", + b"apply_default_optimizations", + "filter_fusion", + b"filter_fusion", + "filter_parallelization", + b"filter_parallelization", + "inject_prefetch", + b"inject_prefetch", + "map_and_batch_fusion", + b"map_and_batch_fusion", + "map_and_filter_fusion", + b"map_and_filter_fusion", + "map_fusion", + b"map_fusion", + "map_parallelization", + b"map_parallelization", + "noop_elimination", + b"noop_elimination", + "optional_apply_default_optimizations", + b"optional_apply_default_optimizations", + "optional_filter_fusion", + b"optional_filter_fusion", + "optional_filter_parallelization", + b"optional_filter_parallelization", + "optional_inject_prefetch", + b"optional_inject_prefetch", + "optional_map_and_batch_fusion", + b"optional_map_and_batch_fusion", + "optional_map_and_filter_fusion", + b"optional_map_and_filter_fusion", + "optional_map_fusion", + b"optional_map_fusion", + "optional_map_parallelization", + b"optional_map_parallelization", + "optional_noop_elimination", + b"optional_noop_elimination", + "optional_parallel_batch", + b"optional_parallel_batch", + "optional_seq_interleave_prefetch", + b"optional_seq_interleave_prefetch", + "optional_shuffle_and_repeat_fusion", + b"optional_shuffle_and_repeat_fusion", + "parallel_batch", + b"parallel_batch", + "seq_interleave_prefetch", + b"seq_interleave_prefetch", + "shuffle_and_repeat_fusion", + b"shuffle_and_repeat_fusion", + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "apply_default_optimizations", + b"apply_default_optimizations", + "filter_fusion", + b"filter_fusion", + "filter_parallelization", + b"filter_parallelization", + "inject_prefetch", + b"inject_prefetch", + "map_and_batch_fusion", + b"map_and_batch_fusion", + "map_and_filter_fusion", + b"map_and_filter_fusion", + "map_fusion", + b"map_fusion", + "map_parallelization", + b"map_parallelization", + "noop_elimination", + b"noop_elimination", + "optional_apply_default_optimizations", + b"optional_apply_default_optimizations", + "optional_filter_fusion", + b"optional_filter_fusion", + "optional_filter_parallelization", + b"optional_filter_parallelization", + "optional_inject_prefetch", + b"optional_inject_prefetch", + "optional_map_and_batch_fusion", + b"optional_map_and_batch_fusion", + "optional_map_and_filter_fusion", + b"optional_map_and_filter_fusion", + "optional_map_fusion", + b"optional_map_fusion", + "optional_map_parallelization", + b"optional_map_parallelization", + "optional_noop_elimination", + b"optional_noop_elimination", + "optional_parallel_batch", + b"optional_parallel_batch", + "optional_seq_interleave_prefetch", + b"optional_seq_interleave_prefetch", + "optional_shuffle_and_repeat_fusion", + b"optional_shuffle_and_repeat_fusion", + "parallel_batch", + b"parallel_batch", + "seq_interleave_prefetch", + b"seq_interleave_prefetch", + "shuffle_and_repeat_fusion", + b"shuffle_and_repeat_fusion", + ], + ) -> None: ... @typing.overload - def WhichOneof(self, oneof_group: typing.Literal["optional_apply_default_optimizations", b"optional_apply_default_optimizations"]) -> typing.Literal["apply_default_optimizations"] | None: ... + def WhichOneof( + self, oneof_group: typing.Literal["optional_apply_default_optimizations", b"optional_apply_default_optimizations"] + ) -> typing.Literal["apply_default_optimizations"] | None: ... @typing.overload - def WhichOneof(self, oneof_group: typing.Literal["optional_filter_fusion", b"optional_filter_fusion"]) -> typing.Literal["filter_fusion"] | None: ... + def WhichOneof( + self, oneof_group: typing.Literal["optional_filter_fusion", b"optional_filter_fusion"] + ) -> typing.Literal["filter_fusion"] | None: ... @typing.overload - def WhichOneof(self, oneof_group: typing.Literal["optional_filter_parallelization", b"optional_filter_parallelization"]) -> typing.Literal["filter_parallelization"] | None: ... + def WhichOneof( + self, oneof_group: typing.Literal["optional_filter_parallelization", b"optional_filter_parallelization"] + ) -> typing.Literal["filter_parallelization"] | None: ... @typing.overload - def WhichOneof(self, oneof_group: typing.Literal["optional_inject_prefetch", b"optional_inject_prefetch"]) -> typing.Literal["inject_prefetch"] | None: ... + def WhichOneof( + self, oneof_group: typing.Literal["optional_inject_prefetch", b"optional_inject_prefetch"] + ) -> typing.Literal["inject_prefetch"] | None: ... @typing.overload - def WhichOneof(self, oneof_group: typing.Literal["optional_map_and_batch_fusion", b"optional_map_and_batch_fusion"]) -> typing.Literal["map_and_batch_fusion"] | None: ... + def WhichOneof( + self, oneof_group: typing.Literal["optional_map_and_batch_fusion", b"optional_map_and_batch_fusion"] + ) -> typing.Literal["map_and_batch_fusion"] | None: ... @typing.overload - def WhichOneof(self, oneof_group: typing.Literal["optional_map_and_filter_fusion", b"optional_map_and_filter_fusion"]) -> typing.Literal["map_and_filter_fusion"] | None: ... + def WhichOneof( + self, oneof_group: typing.Literal["optional_map_and_filter_fusion", b"optional_map_and_filter_fusion"] + ) -> typing.Literal["map_and_filter_fusion"] | None: ... @typing.overload - def WhichOneof(self, oneof_group: typing.Literal["optional_map_fusion", b"optional_map_fusion"]) -> typing.Literal["map_fusion"] | None: ... + def WhichOneof( + self, oneof_group: typing.Literal["optional_map_fusion", b"optional_map_fusion"] + ) -> typing.Literal["map_fusion"] | None: ... @typing.overload - def WhichOneof(self, oneof_group: typing.Literal["optional_map_parallelization", b"optional_map_parallelization"]) -> typing.Literal["map_parallelization"] | None: ... + def WhichOneof( + self, oneof_group: typing.Literal["optional_map_parallelization", b"optional_map_parallelization"] + ) -> typing.Literal["map_parallelization"] | None: ... @typing.overload - def WhichOneof(self, oneof_group: typing.Literal["optional_noop_elimination", b"optional_noop_elimination"]) -> typing.Literal["noop_elimination"] | None: ... + def WhichOneof( + self, oneof_group: typing.Literal["optional_noop_elimination", b"optional_noop_elimination"] + ) -> typing.Literal["noop_elimination"] | None: ... @typing.overload - def WhichOneof(self, oneof_group: typing.Literal["optional_parallel_batch", b"optional_parallel_batch"]) -> typing.Literal["parallel_batch"] | None: ... + def WhichOneof( + self, oneof_group: typing.Literal["optional_parallel_batch", b"optional_parallel_batch"] + ) -> typing.Literal["parallel_batch"] | None: ... @typing.overload - def WhichOneof(self, oneof_group: typing.Literal["optional_seq_interleave_prefetch", b"optional_seq_interleave_prefetch"]) -> typing.Literal["seq_interleave_prefetch"] | None: ... + def WhichOneof( + self, oneof_group: typing.Literal["optional_seq_interleave_prefetch", b"optional_seq_interleave_prefetch"] + ) -> typing.Literal["seq_interleave_prefetch"] | None: ... @typing.overload - def WhichOneof(self, oneof_group: typing.Literal["optional_shuffle_and_repeat_fusion", b"optional_shuffle_and_repeat_fusion"]) -> typing.Literal["shuffle_and_repeat_fusion"] | None: ... + def WhichOneof( + self, oneof_group: typing.Literal["optional_shuffle_and_repeat_fusion", b"optional_shuffle_and_repeat_fusion"] + ) -> typing.Literal["shuffle_and_repeat_fusion"] | None: ... global___OptimizationOptions = OptimizationOptions @@ -286,14 +485,14 @@ class ServiceOptions(google.protobuf.message.Message): PINNED_FIELD_NUMBER: builtins.int pinned: builtins.bool - def __init__( - self, - *, - pinned: builtins.bool | None = ..., - ) -> None: ... - def HasField(self, field_name: typing.Literal["optional_pinned", b"optional_pinned", "pinned", b"pinned"]) -> builtins.bool: ... + def __init__(self, *, pinned: builtins.bool | None = ...) -> None: ... + def HasField( + self, field_name: typing.Literal["optional_pinned", b"optional_pinned", "pinned", b"pinned"] + ) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["optional_pinned", b"optional_pinned", "pinned", b"pinned"]) -> None: ... - def WhichOneof(self, oneof_group: typing.Literal["optional_pinned", b"optional_pinned"]) -> typing.Literal["pinned"] | None: ... + def WhichOneof( + self, oneof_group: typing.Literal["optional_pinned", b"optional_pinned"] + ) -> typing.Literal["pinned"] | None: ... global___ServiceOptions = ServiceOptions @@ -308,17 +507,42 @@ class ThreadingOptions(google.protobuf.message.Message): max_intra_op_parallelism: builtins.int private_threadpool_size: builtins.int def __init__( + self, *, max_intra_op_parallelism: builtins.int | None = ..., private_threadpool_size: builtins.int | None = ... + ) -> None: ... + def HasField( self, - *, - max_intra_op_parallelism: builtins.int | None = ..., - private_threadpool_size: builtins.int | None = ..., + field_name: typing.Literal[ + "max_intra_op_parallelism", + b"max_intra_op_parallelism", + "optional_max_intra_op_parallelism", + b"optional_max_intra_op_parallelism", + "optional_private_threadpool_size", + b"optional_private_threadpool_size", + "private_threadpool_size", + b"private_threadpool_size", + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "max_intra_op_parallelism", + b"max_intra_op_parallelism", + "optional_max_intra_op_parallelism", + b"optional_max_intra_op_parallelism", + "optional_private_threadpool_size", + b"optional_private_threadpool_size", + "private_threadpool_size", + b"private_threadpool_size", + ], ) -> None: ... - def HasField(self, field_name: typing.Literal["max_intra_op_parallelism", b"max_intra_op_parallelism", "optional_max_intra_op_parallelism", b"optional_max_intra_op_parallelism", "optional_private_threadpool_size", b"optional_private_threadpool_size", "private_threadpool_size", b"private_threadpool_size"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["max_intra_op_parallelism", b"max_intra_op_parallelism", "optional_max_intra_op_parallelism", b"optional_max_intra_op_parallelism", "optional_private_threadpool_size", b"optional_private_threadpool_size", "private_threadpool_size", b"private_threadpool_size"]) -> None: ... @typing.overload - def WhichOneof(self, oneof_group: typing.Literal["optional_max_intra_op_parallelism", b"optional_max_intra_op_parallelism"]) -> typing.Literal["max_intra_op_parallelism"] | None: ... + def WhichOneof( + self, oneof_group: typing.Literal["optional_max_intra_op_parallelism", b"optional_max_intra_op_parallelism"] + ) -> typing.Literal["max_intra_op_parallelism"] | None: ... @typing.overload - def WhichOneof(self, oneof_group: typing.Literal["optional_private_threadpool_size", b"optional_private_threadpool_size"]) -> typing.Literal["private_threadpool_size"] | None: ... + def WhichOneof( + self, oneof_group: typing.Literal["optional_private_threadpool_size", b"optional_private_threadpool_size"] + ) -> typing.Literal["private_threadpool_size"] | None: ... global___ThreadingOptions = ThreadingOptions @@ -390,19 +614,107 @@ class Options(google.protobuf.message.Message): symbolic_checkpoint: builtins.bool | None = ..., warm_start: builtins.bool | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["autotune_options", b"autotune_options", "dataset_name", b"dataset_name", "deterministic", b"deterministic", "distribute_options", b"distribute_options", "external_state_policy", b"external_state_policy", "optimization_options", b"optimization_options", "optional_dataset_name", b"optional_dataset_name", "optional_deterministic", b"optional_deterministic", "optional_external_state_policy", b"optional_external_state_policy", "optional_slack", b"optional_slack", "optional_symbolic_checkpoint", b"optional_symbolic_checkpoint", "optional_warm_start", b"optional_warm_start", "service_options", b"service_options", "slack", b"slack", "symbolic_checkpoint", b"symbolic_checkpoint", "threading_options", b"threading_options", "warm_start", b"warm_start"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["autotune_options", b"autotune_options", "dataset_name", b"dataset_name", "deterministic", b"deterministic", "distribute_options", b"distribute_options", "external_state_policy", b"external_state_policy", "framework_type", b"framework_type", "optimization_options", b"optimization_options", "optional_dataset_name", b"optional_dataset_name", "optional_deterministic", b"optional_deterministic", "optional_external_state_policy", b"optional_external_state_policy", "optional_slack", b"optional_slack", "optional_symbolic_checkpoint", b"optional_symbolic_checkpoint", "optional_warm_start", b"optional_warm_start", "service_options", b"service_options", "slack", b"slack", "symbolic_checkpoint", b"symbolic_checkpoint", "threading_options", b"threading_options", "warm_start", b"warm_start"]) -> None: ... + def HasField( + self, + field_name: typing.Literal[ + "autotune_options", + b"autotune_options", + "dataset_name", + b"dataset_name", + "deterministic", + b"deterministic", + "distribute_options", + b"distribute_options", + "external_state_policy", + b"external_state_policy", + "optimization_options", + b"optimization_options", + "optional_dataset_name", + b"optional_dataset_name", + "optional_deterministic", + b"optional_deterministic", + "optional_external_state_policy", + b"optional_external_state_policy", + "optional_slack", + b"optional_slack", + "optional_symbolic_checkpoint", + b"optional_symbolic_checkpoint", + "optional_warm_start", + b"optional_warm_start", + "service_options", + b"service_options", + "slack", + b"slack", + "symbolic_checkpoint", + b"symbolic_checkpoint", + "threading_options", + b"threading_options", + "warm_start", + b"warm_start", + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "autotune_options", + b"autotune_options", + "dataset_name", + b"dataset_name", + "deterministic", + b"deterministic", + "distribute_options", + b"distribute_options", + "external_state_policy", + b"external_state_policy", + "framework_type", + b"framework_type", + "optimization_options", + b"optimization_options", + "optional_dataset_name", + b"optional_dataset_name", + "optional_deterministic", + b"optional_deterministic", + "optional_external_state_policy", + b"optional_external_state_policy", + "optional_slack", + b"optional_slack", + "optional_symbolic_checkpoint", + b"optional_symbolic_checkpoint", + "optional_warm_start", + b"optional_warm_start", + "service_options", + b"service_options", + "slack", + b"slack", + "symbolic_checkpoint", + b"symbolic_checkpoint", + "threading_options", + b"threading_options", + "warm_start", + b"warm_start", + ], + ) -> None: ... @typing.overload - def WhichOneof(self, oneof_group: typing.Literal["optional_dataset_name", b"optional_dataset_name"]) -> typing.Literal["dataset_name"] | None: ... + def WhichOneof( + self, oneof_group: typing.Literal["optional_dataset_name", b"optional_dataset_name"] + ) -> typing.Literal["dataset_name"] | None: ... @typing.overload - def WhichOneof(self, oneof_group: typing.Literal["optional_deterministic", b"optional_deterministic"]) -> typing.Literal["deterministic"] | None: ... + def WhichOneof( + self, oneof_group: typing.Literal["optional_deterministic", b"optional_deterministic"] + ) -> typing.Literal["deterministic"] | None: ... @typing.overload - def WhichOneof(self, oneof_group: typing.Literal["optional_external_state_policy", b"optional_external_state_policy"]) -> typing.Literal["external_state_policy"] | None: ... + def WhichOneof( + self, oneof_group: typing.Literal["optional_external_state_policy", b"optional_external_state_policy"] + ) -> typing.Literal["external_state_policy"] | None: ... @typing.overload def WhichOneof(self, oneof_group: typing.Literal["optional_slack", b"optional_slack"]) -> typing.Literal["slack"] | None: ... @typing.overload - def WhichOneof(self, oneof_group: typing.Literal["optional_symbolic_checkpoint", b"optional_symbolic_checkpoint"]) -> typing.Literal["symbolic_checkpoint"] | None: ... + def WhichOneof( + self, oneof_group: typing.Literal["optional_symbolic_checkpoint", b"optional_symbolic_checkpoint"] + ) -> typing.Literal["symbolic_checkpoint"] | None: ... @typing.overload - def WhichOneof(self, oneof_group: typing.Literal["optional_warm_start", b"optional_warm_start"]) -> typing.Literal["warm_start"] | None: ... + def WhichOneof( + self, oneof_group: typing.Literal["optional_warm_start", b"optional_warm_start"] + ) -> typing.Literal["warm_start"] | None: ... global___Options = Options diff --git a/stubs/tensorflow/tensorflow/core/framework/dataset_pb2.pyi b/stubs/tensorflow/tensorflow/core/framework/dataset_pb2.pyi index 72392b2b7698..0cfb9bb85928 100644 --- a/stubs/tensorflow/tensorflow/core/framework/dataset_pb2.pyi +++ b/stubs/tensorflow/tensorflow/core/framework/dataset_pb2.pyi @@ -51,7 +51,12 @@ class CompressedComponentMetadata(google.protobuf.message.Message): uncompressed_bytes: collections.abc.Iterable[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["dtype", b"dtype", "tensor_shape", b"tensor_shape", "uncompressed_bytes", b"uncompressed_bytes"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "dtype", b"dtype", "tensor_shape", b"tensor_shape", "uncompressed_bytes", b"uncompressed_bytes" + ], + ) -> None: ... global___CompressedComponentMetadata = CompressedComponentMetadata @@ -72,7 +77,9 @@ class CompressedElement(google.protobuf.message.Message): tensorflow/core/data/compression_utils.cc. """ @property - def component_metadata(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___CompressedComponentMetadata]: + def component_metadata( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___CompressedComponentMetadata]: """Metadata for the components of the element.""" def __init__( @@ -82,7 +89,9 @@ class CompressedElement(google.protobuf.message.Message): component_metadata: collections.abc.Iterable[global___CompressedComponentMetadata] | None = ..., version: builtins.int | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["component_metadata", b"component_metadata", "data", b"data", "version", b"version"]) -> None: ... + def ClearField( + self, field_name: typing.Literal["component_metadata", b"component_metadata", "data", b"data", "version", b"version"] + ) -> None: ... global___CompressedElement = CompressedElement @@ -94,11 +103,13 @@ class UncompressedElement(google.protobuf.message.Message): COMPONENTS_FIELD_NUMBER: builtins.int @property - def components(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[tensorflow.core.framework.tensor_pb2.TensorProto]: ... - def __init__( + def components( self, - *, - components: collections.abc.Iterable[tensorflow.core.framework.tensor_pb2.TensorProto] | None = ..., + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[ + tensorflow.core.framework.tensor_pb2.TensorProto + ]: ... + def __init__( + self, *, components: collections.abc.Iterable[tensorflow.core.framework.tensor_pb2.TensorProto] | None = ... ) -> None: ... def ClearField(self, field_name: typing.Literal["components", b"components"]) -> None: ... diff --git a/stubs/tensorflow/tensorflow/core/framework/device_attributes_pb2.pyi b/stubs/tensorflow/tensorflow/core/framework/device_attributes_pb2.pyi index aafdc2c8a51e..41abb3a43b72 100644 --- a/stubs/tensorflow/tensorflow/core/framework/device_attributes_pb2.pyi +++ b/stubs/tensorflow/tensorflow/core/framework/device_attributes_pb2.pyi @@ -24,13 +24,11 @@ class InterconnectLink(google.protobuf.message.Message): type: builtins.str strength: builtins.int def __init__( - self, - *, - device_id: builtins.int | None = ..., - type: builtins.str | None = ..., - strength: builtins.int | None = ..., + self, *, device_id: builtins.int | None = ..., type: builtins.str | None = ..., strength: builtins.int | None = ... + ) -> None: ... + def ClearField( + self, field_name: typing.Literal["device_id", b"device_id", "strength", b"strength", "type", b"type"] ) -> None: ... - def ClearField(self, field_name: typing.Literal["device_id", b"device_id", "strength", b"strength", "type", b"type"]) -> None: ... global___InterconnectLink = InterconnectLink @@ -41,11 +39,7 @@ class LocalLinks(google.protobuf.message.Message): LINK_FIELD_NUMBER: builtins.int @property def link(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___InterconnectLink]: ... - def __init__( - self, - *, - link: collections.abc.Iterable[global___InterconnectLink] | None = ..., - ) -> None: ... + def __init__(self, *, link: collections.abc.Iterable[global___InterconnectLink] | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["link", b"link"]) -> None: ... global___LocalLinks = LocalLinks @@ -68,14 +62,12 @@ class DeviceLocality(google.protobuf.message.Message): """Optional local interconnect links to other devices.""" def __init__( - self, - *, - bus_id: builtins.int | None = ..., - numa_node: builtins.int | None = ..., - links: global___LocalLinks | None = ..., + self, *, bus_id: builtins.int | None = ..., numa_node: builtins.int | None = ..., links: global___LocalLinks | None = ... ) -> None: ... def HasField(self, field_name: typing.Literal["links", b"links"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["bus_id", b"bus_id", "links", b"links", "numa_node", b"numa_node"]) -> None: ... + def ClearField( + self, field_name: typing.Literal["bus_id", b"bus_id", "links", b"links", "numa_node", b"numa_node"] + ) -> None: ... global___DeviceLocality = DeviceLocality @@ -125,6 +117,24 @@ class DeviceAttributes(google.protobuf.message.Message): xla_global_id: builtins.int | None = ..., ) -> None: ... def HasField(self, field_name: typing.Literal["locality", b"locality"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["device_type", b"device_type", "incarnation", b"incarnation", "locality", b"locality", "memory_limit", b"memory_limit", "name", b"name", "physical_device_desc", b"physical_device_desc", "xla_global_id", b"xla_global_id"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "device_type", + b"device_type", + "incarnation", + b"incarnation", + "locality", + b"locality", + "memory_limit", + b"memory_limit", + "name", + b"name", + "physical_device_desc", + b"physical_device_desc", + "xla_global_id", + b"xla_global_id", + ], + ) -> None: ... global___DeviceAttributes = DeviceAttributes diff --git a/stubs/tensorflow/tensorflow/core/framework/full_type_pb2.pyi b/stubs/tensorflow/tensorflow/core/framework/full_type_pb2.pyi index e37ac0ff0495..8025593122d5 100644 --- a/stubs/tensorflow/tensorflow/core/framework/full_type_pb2.pyi +++ b/stubs/tensorflow/tensorflow/core/framework/full_type_pb2.pyi @@ -24,7 +24,9 @@ class _FullTypeId: ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType -class _FullTypeIdEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_FullTypeId.ValueType], builtins.type): +class _FullTypeIdEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_FullTypeId.ValueType], builtins.type +): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor TFT_UNSET: _FullTypeId.ValueType # 0 """The default represents an uninitialized values.""" @@ -607,7 +609,9 @@ class FullTypeDef(google.protobuf.message.Message): i: builtins.int | None = ..., ) -> None: ... def HasField(self, field_name: typing.Literal["attr", b"attr", "i", b"i", "s", b"s"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["args", b"args", "attr", b"attr", "i", b"i", "s", b"s", "type_id", b"type_id"]) -> None: ... + def ClearField( + self, field_name: typing.Literal["args", b"args", "attr", b"attr", "i", b"i", "s", b"s", "type_id", b"type_id"] + ) -> None: ... def WhichOneof(self, oneof_group: typing.Literal["attr", b"attr"]) -> typing.Literal["s", "i"] | None: ... global___FullTypeDef = FullTypeDef diff --git a/stubs/tensorflow/tensorflow/core/framework/function_pb2.pyi b/stubs/tensorflow/tensorflow/core/framework/function_pb2.pyi index c0908173a958..8506f933b2c0 100644 --- a/stubs/tensorflow/tensorflow/core/framework/function_pb2.pyi +++ b/stubs/tensorflow/tensorflow/core/framework/function_pb2.pyi @@ -30,7 +30,9 @@ class FunctionDefLibrary(google.protobuf.message.Message): @property def gradient(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___GradientDef]: ... @property - def registered_gradients(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___RegisteredGradient]: ... + def registered_gradients( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___RegisteredGradient]: ... def __init__( self, *, @@ -38,7 +40,12 @@ class FunctionDefLibrary(google.protobuf.message.Message): gradient: collections.abc.Iterable[global___GradientDef] | None = ..., registered_gradients: collections.abc.Iterable[global___RegisteredGradient] | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["function", b"function", "gradient", b"gradient", "registered_gradients", b"registered_gradients"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "function", b"function", "gradient", b"gradient", "registered_gradients", b"registered_gradients" + ], + ) -> None: ... global___FunctionDefLibrary = FunctionDefLibrary @@ -64,10 +71,7 @@ class FunctionDef(google.protobuf.message.Message): @property def value(self) -> tensorflow.core.framework.attr_value_pb2.AttrValue: ... def __init__( - self, - *, - key: builtins.str | None = ..., - value: tensorflow.core.framework.attr_value_pb2.AttrValue | None = ..., + self, *, key: builtins.str | None = ..., value: tensorflow.core.framework.attr_value_pb2.AttrValue | None = ... ) -> None: ... def HasField(self, field_name: typing.Literal["value", b"value"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["key", b"key", "value", b"value"]) -> None: ... @@ -90,21 +94,18 @@ class FunctionDef(google.protobuf.message.Message): @property def value(self) -> tensorflow.core.framework.attr_value_pb2.AttrValue: ... def __init__( - self, - *, - key: builtins.str | None = ..., - value: tensorflow.core.framework.attr_value_pb2.AttrValue | None = ..., + self, *, key: builtins.str | None = ..., value: tensorflow.core.framework.attr_value_pb2.AttrValue | None = ... ) -> None: ... def HasField(self, field_name: typing.Literal["value", b"value"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["key", b"key", "value", b"value"]) -> None: ... ATTR_FIELD_NUMBER: builtins.int @property - def attr(self) -> google.protobuf.internal.containers.MessageMap[builtins.str, tensorflow.core.framework.attr_value_pb2.AttrValue]: ... - def __init__( + def attr( self, - *, - attr: collections.abc.Mapping[builtins.str, tensorflow.core.framework.attr_value_pb2.AttrValue] | None = ..., + ) -> google.protobuf.internal.containers.MessageMap[builtins.str, tensorflow.core.framework.attr_value_pb2.AttrValue]: ... + def __init__( + self, *, attr: collections.abc.Mapping[builtins.str, tensorflow.core.framework.attr_value_pb2.AttrValue] | None = ... ) -> None: ... def ClearField(self, field_name: typing.Literal["attr", b"attr"]) -> None: ... @@ -117,12 +118,7 @@ class FunctionDef(google.protobuf.message.Message): key: builtins.int @property def value(self) -> global___FunctionDef.ArgAttrs: ... - def __init__( - self, - *, - key: builtins.int | None = ..., - value: global___FunctionDef.ArgAttrs | None = ..., - ) -> None: ... + def __init__(self, *, key: builtins.int | None = ..., value: global___FunctionDef.ArgAttrs | None = ...) -> None: ... def HasField(self, field_name: typing.Literal["value", b"value"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["key", b"key", "value", b"value"]) -> None: ... @@ -134,12 +130,7 @@ class FunctionDef(google.protobuf.message.Message): VALUE_FIELD_NUMBER: builtins.int key: builtins.int value: builtins.int - def __init__( - self, - *, - key: builtins.int | None = ..., - value: builtins.int | None = ..., - ) -> None: ... + def __init__(self, *, key: builtins.int | None = ..., value: builtins.int | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["key", b"key", "value", b"value"]) -> None: ... @typing.final @@ -150,12 +141,7 @@ class FunctionDef(google.protobuf.message.Message): VALUE_FIELD_NUMBER: builtins.int key: builtins.str value: builtins.str - def __init__( - self, - *, - key: builtins.str | None = ..., - value: builtins.str | None = ..., - ) -> None: ... + def __init__(self, *, key: builtins.str | None = ..., value: builtins.str | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["key", b"key", "value", b"value"]) -> None: ... @typing.final @@ -166,12 +152,7 @@ class FunctionDef(google.protobuf.message.Message): VALUE_FIELD_NUMBER: builtins.int key: builtins.str value: builtins.str - def __init__( - self, - *, - key: builtins.str | None = ..., - value: builtins.str | None = ..., - ) -> None: ... + def __init__(self, *, key: builtins.str | None = ..., value: builtins.str | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["key", b"key", "value", b"value"]) -> None: ... SIGNATURE_FIELD_NUMBER: builtins.int @@ -188,7 +169,9 @@ class FunctionDef(google.protobuf.message.Message): """ @property - def attr(self) -> google.protobuf.internal.containers.MessageMap[builtins.str, tensorflow.core.framework.attr_value_pb2.AttrValue]: + def attr( + self, + ) -> google.protobuf.internal.containers.MessageMap[builtins.str, tensorflow.core.framework.attr_value_pb2.AttrValue]: """Attributes specific to this function definition.""" @property @@ -207,7 +190,9 @@ class FunctionDef(google.protobuf.message.Message): """ @property - def node_def(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[tensorflow.core.framework.node_def_pb2.NodeDef]: + def node_def( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[tensorflow.core.framework.node_def_pb2.NodeDef]: """The body of the function. Unlike the NodeDefs in a GraphDef, attrs may have values of type `placeholder` and the `input` field uses the "output" format above. @@ -241,7 +226,25 @@ class FunctionDef(google.protobuf.message.Message): control_ret: collections.abc.Mapping[builtins.str, builtins.str] | None = ..., ) -> None: ... def HasField(self, field_name: typing.Literal["signature", b"signature"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["arg_attr", b"arg_attr", "attr", b"attr", "control_ret", b"control_ret", "node_def", b"node_def", "resource_arg_unique_id", b"resource_arg_unique_id", "ret", b"ret", "signature", b"signature"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "arg_attr", + b"arg_attr", + "attr", + b"attr", + "control_ret", + b"control_ret", + "node_def", + b"node_def", + "resource_arg_unique_id", + b"resource_arg_unique_id", + "ret", + b"ret", + "signature", + b"signature", + ], + ) -> None: ... global___FunctionDef = FunctionDef @@ -275,13 +278,10 @@ class GradientDef(google.protobuf.message.Message): """The function name.""" gradient_func: builtins.str """The gradient function's name.""" - def __init__( - self, - *, - function_name: builtins.str | None = ..., - gradient_func: builtins.str | None = ..., + def __init__(self, *, function_name: builtins.str | None = ..., gradient_func: builtins.str | None = ...) -> None: ... + def ClearField( + self, field_name: typing.Literal["function_name", b"function_name", "gradient_func", b"gradient_func"] ) -> None: ... - def ClearField(self, field_name: typing.Literal["function_name", b"function_name", "gradient_func", b"gradient_func"]) -> None: ... global___GradientDef = GradientDef @@ -301,12 +301,9 @@ class RegisteredGradient(google.protobuf.message.Message): """The gradient function's name.""" registered_op_type: builtins.str """The gradient function's registered op type.""" - def __init__( - self, - *, - gradient_func: builtins.str | None = ..., - registered_op_type: builtins.str | None = ..., + def __init__(self, *, gradient_func: builtins.str | None = ..., registered_op_type: builtins.str | None = ...) -> None: ... + def ClearField( + self, field_name: typing.Literal["gradient_func", b"gradient_func", "registered_op_type", b"registered_op_type"] ) -> None: ... - def ClearField(self, field_name: typing.Literal["gradient_func", b"gradient_func", "registered_op_type", b"registered_op_type"]) -> None: ... global___RegisteredGradient = RegisteredGradient diff --git a/stubs/tensorflow/tensorflow/core/framework/graph_debug_info_pb2.pyi b/stubs/tensorflow/tensorflow/core/framework/graph_debug_info_pb2.pyi index 739c2b5bd300..2c3f6761f1ef 100644 --- a/stubs/tensorflow/tensorflow/core/framework/graph_debug_info_pb2.pyi +++ b/stubs/tensorflow/tensorflow/core/framework/graph_debug_info_pb2.pyi @@ -49,8 +49,18 @@ class GraphDebugInfo(google.protobuf.message.Message): func: builtins.str | None = ..., code: builtins.str | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["code", b"code", "col", b"col", "file_index", b"file_index", "func", b"func", "line", b"line"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["code", b"code", "col", b"col", "file_index", b"file_index", "func", b"func", "line", b"line"]) -> None: ... + def HasField( + self, + field_name: typing.Literal[ + "code", b"code", "col", b"col", "file_index", b"file_index", "func", b"func", "line", b"line" + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "code", b"code", "col", b"col", "file_index", b"file_index", "func", b"func", "line", b"line" + ], + ) -> None: ... @typing.final class StackTrace(google.protobuf.message.Message): @@ -61,7 +71,9 @@ class GraphDebugInfo(google.protobuf.message.Message): FILE_LINE_COLS_FIELD_NUMBER: builtins.int FRAME_ID_FIELD_NUMBER: builtins.int @property - def file_line_cols(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___GraphDebugInfo.FileLineCol]: + def file_line_cols( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___GraphDebugInfo.FileLineCol]: """Deprecated.""" @property @@ -72,7 +84,9 @@ class GraphDebugInfo(google.protobuf.message.Message): file_line_cols: collections.abc.Iterable[global___GraphDebugInfo.FileLineCol] | None = ..., frame_id: collections.abc.Iterable[builtins.int] | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["file_line_cols", b"file_line_cols", "frame_id", b"frame_id"]) -> None: ... + def ClearField( + self, field_name: typing.Literal["file_line_cols", b"file_line_cols", "frame_id", b"frame_id"] + ) -> None: ... @typing.final class FramesByIdEntry(google.protobuf.message.Message): @@ -84,10 +98,7 @@ class GraphDebugInfo(google.protobuf.message.Message): @property def value(self) -> global___GraphDebugInfo.FileLineCol: ... def __init__( - self, - *, - key: builtins.int | None = ..., - value: global___GraphDebugInfo.FileLineCol | None = ..., + self, *, key: builtins.int | None = ..., value: global___GraphDebugInfo.FileLineCol | None = ... ) -> None: ... def HasField(self, field_name: typing.Literal["key", b"key", "value", b"value"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["key", b"key", "value", b"value"]) -> None: ... @@ -101,12 +112,7 @@ class GraphDebugInfo(google.protobuf.message.Message): key: builtins.int @property def value(self) -> global___GraphDebugInfo.StackTrace: ... - def __init__( - self, - *, - key: builtins.int | None = ..., - value: global___GraphDebugInfo.StackTrace | None = ..., - ) -> None: ... + def __init__(self, *, key: builtins.int | None = ..., value: global___GraphDebugInfo.StackTrace | None = ...) -> None: ... def HasField(self, field_name: typing.Literal["key", b"key", "value", b"value"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["key", b"key", "value", b"value"]) -> None: ... @@ -119,12 +125,7 @@ class GraphDebugInfo(google.protobuf.message.Message): key: builtins.str @property def value(self) -> global___GraphDebugInfo.StackTrace: ... - def __init__( - self, - *, - key: builtins.str | None = ..., - value: global___GraphDebugInfo.StackTrace | None = ..., - ) -> None: ... + def __init__(self, *, key: builtins.str | None = ..., value: global___GraphDebugInfo.StackTrace | None = ...) -> None: ... def HasField(self, field_name: typing.Literal["key", b"key", "value", b"value"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["key", b"key", "value", b"value"]) -> None: ... @@ -136,12 +137,7 @@ class GraphDebugInfo(google.protobuf.message.Message): VALUE_FIELD_NUMBER: builtins.int key: builtins.str value: builtins.int - def __init__( - self, - *, - key: builtins.str | None = ..., - value: builtins.int | None = ..., - ) -> None: ... + def __init__(self, *, key: builtins.str | None = ..., value: builtins.int | None = ...) -> None: ... def HasField(self, field_name: typing.Literal["key", b"key", "value", b"value"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["key", b"key", "value", b"value"]) -> None: ... @@ -163,7 +159,9 @@ class GraphDebugInfo(google.protobuf.message.Message): """ @property - def traces_by_id(self) -> google.protobuf.internal.containers.MessageMap[builtins.int, global___GraphDebugInfo.StackTrace]: ... + def traces_by_id( + self, + ) -> google.protobuf.internal.containers.MessageMap[builtins.int, global___GraphDebugInfo.StackTrace]: ... @property def traces(self) -> google.protobuf.internal.containers.MessageMap[builtins.str, global___GraphDebugInfo.StackTrace]: """Deprecated.""" @@ -193,6 +191,20 @@ class GraphDebugInfo(google.protobuf.message.Message): traces: collections.abc.Mapping[builtins.str, global___GraphDebugInfo.StackTrace] | None = ..., name_to_trace_id: collections.abc.Mapping[builtins.str, builtins.int] | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["files", b"files", "frames_by_id", b"frames_by_id", "name_to_trace_id", b"name_to_trace_id", "traces", b"traces", "traces_by_id", b"traces_by_id"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "files", + b"files", + "frames_by_id", + b"frames_by_id", + "name_to_trace_id", + b"name_to_trace_id", + "traces", + b"traces", + "traces_by_id", + b"traces_by_id", + ], + ) -> None: ... global___GraphDebugInfo = GraphDebugInfo diff --git a/stubs/tensorflow/tensorflow/core/framework/graph_pb2.pyi b/stubs/tensorflow/tensorflow/core/framework/graph_pb2.pyi index 1158f045aba3..3a14bc413e52 100644 --- a/stubs/tensorflow/tensorflow/core/framework/graph_pb2.pyi +++ b/stubs/tensorflow/tensorflow/core/framework/graph_pb2.pyi @@ -34,7 +34,9 @@ class GraphDef(google.protobuf.message.Message): compatible, this field is entirely ignored. """ @property - def node(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[tensorflow.core.framework.node_def_pb2.NodeDef]: ... + def node( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[tensorflow.core.framework.node_def_pb2.NodeDef]: ... @property def versions(self) -> tensorflow.core.framework.versions_pb2.VersionDef: """Compatibility versions of the graph. See core/public/version.h for version @@ -44,7 +46,7 @@ class GraphDef(google.protobuf.message.Message): @property def library(self) -> tensorflow.core.framework.function_pb2.FunctionDefLibrary: - """"library" provides user-defined functions. + """ "library" provides user-defined functions. Naming: * library.function.name are in a flat namespace. @@ -85,7 +87,14 @@ class GraphDef(google.protobuf.message.Message): library: tensorflow.core.framework.function_pb2.FunctionDefLibrary | None = ..., debug_info: tensorflow.core.framework.graph_debug_info_pb2.GraphDebugInfo | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["debug_info", b"debug_info", "library", b"library", "versions", b"versions"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["debug_info", b"debug_info", "library", b"library", "node", b"node", "version", b"version", "versions", b"versions"]) -> None: ... + def HasField( + self, field_name: typing.Literal["debug_info", b"debug_info", "library", b"library", "versions", b"versions"] + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "debug_info", b"debug_info", "library", b"library", "node", b"node", "version", b"version", "versions", b"versions" + ], + ) -> None: ... global___GraphDef = GraphDef diff --git a/stubs/tensorflow/tensorflow/core/framework/graph_transfer_info_pb2.pyi b/stubs/tensorflow/tensorflow/core/framework/graph_transfer_info_pb2.pyi index ab4dd51228cd..bba887ced518 100644 --- a/stubs/tensorflow/tensorflow/core/framework/graph_transfer_info_pb2.pyi +++ b/stubs/tensorflow/tensorflow/core/framework/graph_transfer_info_pb2.pyi @@ -29,12 +29,7 @@ class GraphTransferNodeInput(google.protobuf.message.Message): OUTPUT_PORT_FIELD_NUMBER: builtins.int node_id: builtins.int output_port: builtins.int - def __init__( - self, - *, - node_id: builtins.int | None = ..., - output_port: builtins.int | None = ..., - ) -> None: ... + def __init__(self, *, node_id: builtins.int | None = ..., output_port: builtins.int | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["node_id", b"node_id", "output_port", b"output_port"]) -> None: ... global___GraphTransferNodeInput = GraphTransferNodeInput @@ -68,7 +63,25 @@ class GraphTransferNodeInfo(google.protobuf.message.Message): input_count: builtins.int | None = ..., output_count: builtins.int | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["input_count", b"input_count", "name", b"name", "node_id", b"node_id", "output_count", b"output_count", "padding_id", b"padding_id", "soc_op_id", b"soc_op_id", "type_name", b"type_name"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "input_count", + b"input_count", + "name", + b"name", + "node_id", + b"node_id", + "output_count", + b"output_count", + "padding_id", + b"padding_id", + "soc_op_id", + b"soc_op_id", + "type_name", + b"type_name", + ], + ) -> None: ... global___GraphTransferNodeInfo = GraphTransferNodeInfo @@ -96,7 +109,10 @@ class GraphTransferConstNodeInfo(google.protobuf.message.Message): data: builtins.bytes | None = ..., dtype: tensorflow.core.framework.types_pb2.DataType.ValueType | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["data", b"data", "dtype", b"dtype", "name", b"name", "node_id", b"node_id", "shape", b"shape"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal["data", b"data", "dtype", b"dtype", "name", b"name", "node_id", b"node_id", "shape", b"shape"], + ) -> None: ... global___GraphTransferConstNodeInfo = GraphTransferConstNodeInfo @@ -108,7 +124,9 @@ class GraphTransferNodeInputInfo(google.protobuf.message.Message): NODE_INPUT_FIELD_NUMBER: builtins.int node_id: builtins.int @property - def node_input(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___GraphTransferNodeInput]: ... + def node_input( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___GraphTransferNodeInput]: ... def __init__( self, *, @@ -129,10 +147,7 @@ class GraphTransferNodeOutputInfo(google.protobuf.message.Message): @property def max_byte_size(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.int]: ... def __init__( - self, - *, - node_id: builtins.int | None = ..., - max_byte_size: collections.abc.Iterable[builtins.int] | None = ..., + self, *, node_id: builtins.int | None = ..., max_byte_size: collections.abc.Iterable[builtins.int] | None = ... ) -> None: ... def ClearField(self, field_name: typing.Literal["max_byte_size", b"max_byte_size", "node_id", b"node_id"]) -> None: ... @@ -195,7 +210,9 @@ class GraphTransferInfo(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _DestinationEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[GraphTransferInfo._Destination.ValueType], builtins.type): + class _DestinationEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[GraphTransferInfo._Destination.ValueType], builtins.type + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor NOP: GraphTransferInfo._Destination.ValueType # 0 HEXAGON: GraphTransferInfo._Destination.ValueType # 1 @@ -214,19 +231,31 @@ class GraphTransferInfo(google.protobuf.message.Message): destination: global___GraphTransferInfo.Destination.ValueType """Destination of graph transfer""" @property - def node_info(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___GraphTransferNodeInfo]: ... + def node_info( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___GraphTransferNodeInfo]: ... @property - def const_node_info(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___GraphTransferConstNodeInfo]: ... + def const_node_info( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___GraphTransferConstNodeInfo]: ... @property - def node_input_info(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___GraphTransferNodeInputInfo]: ... + def node_input_info( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___GraphTransferNodeInputInfo]: ... @property - def node_output_info(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___GraphTransferNodeOutputInfo]: ... + def node_output_info( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___GraphTransferNodeOutputInfo]: ... @property - def graph_input_node_info(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___GraphTransferGraphInputNodeInfo]: + def graph_input_node_info( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___GraphTransferGraphInputNodeInfo]: """Input Node parameters of transferred graph""" @property - def graph_output_node_info(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___GraphTransferGraphOutputNodeInfo]: ... + def graph_output_node_info( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___GraphTransferGraphOutputNodeInfo]: ... def __init__( self, *, @@ -238,6 +267,24 @@ class GraphTransferInfo(google.protobuf.message.Message): graph_output_node_info: collections.abc.Iterable[global___GraphTransferGraphOutputNodeInfo] | None = ..., destination: global___GraphTransferInfo.Destination.ValueType | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["const_node_info", b"const_node_info", "destination", b"destination", "graph_input_node_info", b"graph_input_node_info", "graph_output_node_info", b"graph_output_node_info", "node_info", b"node_info", "node_input_info", b"node_input_info", "node_output_info", b"node_output_info"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "const_node_info", + b"const_node_info", + "destination", + b"destination", + "graph_input_node_info", + b"graph_input_node_info", + "graph_output_node_info", + b"graph_output_node_info", + "node_info", + b"node_info", + "node_input_info", + b"node_input_info", + "node_output_info", + b"node_output_info", + ], + ) -> None: ... global___GraphTransferInfo = GraphTransferInfo diff --git a/stubs/tensorflow/tensorflow/core/framework/kernel_def_pb2.pyi b/stubs/tensorflow/tensorflow/core/framework/kernel_def_pb2.pyi index ea1844b41fbc..c0f0f52e5cbf 100644 --- a/stubs/tensorflow/tensorflow/core/framework/kernel_def_pb2.pyi +++ b/stubs/tensorflow/tensorflow/core/framework/kernel_def_pb2.pyi @@ -62,7 +62,9 @@ class KernelDef(google.protobuf.message.Message): this is not set), we prefer GPU kernels over CPU. """ @property - def constraint(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___KernelDef.AttrConstraint]: ... + def constraint( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___KernelDef.AttrConstraint]: ... @property def host_memory_arg(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]: """Names of the Op's input_/output_args that reside in host memory @@ -79,7 +81,23 @@ class KernelDef(google.protobuf.message.Message): label: builtins.str | None = ..., priority: builtins.int | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["constraint", b"constraint", "device_type", b"device_type", "host_memory_arg", b"host_memory_arg", "label", b"label", "op", b"op", "priority", b"priority"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "constraint", + b"constraint", + "device_type", + b"device_type", + "host_memory_arg", + b"host_memory_arg", + "label", + b"label", + "op", + b"op", + "priority", + b"priority", + ], + ) -> None: ... global___KernelDef = KernelDef @@ -92,11 +110,7 @@ class KernelList(google.protobuf.message.Message): KERNEL_FIELD_NUMBER: builtins.int @property def kernel(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___KernelDef]: ... - def __init__( - self, - *, - kernel: collections.abc.Iterable[global___KernelDef] | None = ..., - ) -> None: ... + def __init__(self, *, kernel: collections.abc.Iterable[global___KernelDef] | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["kernel", b"kernel"]) -> None: ... global___KernelList = KernelList diff --git a/stubs/tensorflow/tensorflow/core/framework/log_memory_pb2.pyi b/stubs/tensorflow/tensorflow/core/framework/log_memory_pb2.pyi index ba0e1700fc9f..7b8d91953e77 100644 --- a/stubs/tensorflow/tensorflow/core/framework/log_memory_pb2.pyi +++ b/stubs/tensorflow/tensorflow/core/framework/log_memory_pb2.pyi @@ -22,12 +22,7 @@ class MemoryLogStep(google.protobuf.message.Message): """Process-unique step id.""" handle: builtins.str """Handle describing the feeds and fetches of the step.""" - def __init__( - self, - *, - step_id: builtins.int | None = ..., - handle: builtins.str | None = ..., - ) -> None: ... + def __init__(self, *, step_id: builtins.int | None = ..., handle: builtins.str | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["handle", b"handle", "step_id", b"step_id"]) -> None: ... global___MemoryLogStep = MemoryLogStep @@ -57,7 +52,9 @@ class MemoryLogTensorAllocation(google.protobuf.message.Message): tensor: tensorflow.core.framework.tensor_description_pb2.TensorDescription | None = ..., ) -> None: ... def HasField(self, field_name: typing.Literal["tensor", b"tensor"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["kernel_name", b"kernel_name", "step_id", b"step_id", "tensor", b"tensor"]) -> None: ... + def ClearField( + self, field_name: typing.Literal["kernel_name", b"kernel_name", "step_id", b"step_id", "tensor", b"tensor"] + ) -> None: ... global___MemoryLogTensorAllocation = MemoryLogTensorAllocation @@ -73,13 +70,10 @@ class MemoryLogTensorDeallocation(google.protobuf.message.Message): """ allocator_name: builtins.str """Name of the allocator used.""" - def __init__( - self, - *, - allocation_id: builtins.int | None = ..., - allocator_name: builtins.str | None = ..., + def __init__(self, *, allocation_id: builtins.int | None = ..., allocator_name: builtins.str | None = ...) -> None: ... + def ClearField( + self, field_name: typing.Literal["allocation_id", b"allocation_id", "allocator_name", b"allocator_name"] ) -> None: ... - def ClearField(self, field_name: typing.Literal["allocation_id", b"allocation_id", "allocator_name", b"allocator_name"]) -> None: ... global___MemoryLogTensorDeallocation = MemoryLogTensorDeallocation @@ -112,7 +106,10 @@ class MemoryLogTensorOutput(google.protobuf.message.Message): tensor: tensorflow.core.framework.tensor_description_pb2.TensorDescription | None = ..., ) -> None: ... def HasField(self, field_name: typing.Literal["tensor", b"tensor"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["index", b"index", "kernel_name", b"kernel_name", "step_id", b"step_id", "tensor", b"tensor"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal["index", b"index", "kernel_name", b"kernel_name", "step_id", b"step_id", "tensor", b"tensor"], + ) -> None: ... global___MemoryLogTensorOutput = MemoryLogTensorOutput @@ -150,7 +147,23 @@ class MemoryLogRawAllocation(google.protobuf.message.Message): allocation_id: builtins.int | None = ..., allocator_name: builtins.str | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["allocation_id", b"allocation_id", "allocator_name", b"allocator_name", "num_bytes", b"num_bytes", "operation", b"operation", "ptr", b"ptr", "step_id", b"step_id"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "allocation_id", + b"allocation_id", + "allocator_name", + b"allocator_name", + "num_bytes", + b"num_bytes", + "operation", + b"operation", + "ptr", + b"ptr", + "step_id", + b"step_id", + ], + ) -> None: ... global___MemoryLogRawAllocation = MemoryLogRawAllocation @@ -186,6 +199,20 @@ class MemoryLogRawDeallocation(google.protobuf.message.Message): allocator_name: builtins.str | None = ..., deferred: builtins.bool | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["allocation_id", b"allocation_id", "allocator_name", b"allocator_name", "deferred", b"deferred", "operation", b"operation", "step_id", b"step_id"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "allocation_id", + b"allocation_id", + "allocator_name", + b"allocator_name", + "deferred", + b"deferred", + "operation", + b"operation", + "step_id", + b"step_id", + ], + ) -> None: ... global___MemoryLogRawDeallocation = MemoryLogRawDeallocation diff --git a/stubs/tensorflow/tensorflow/core/framework/model_pb2.pyi b/stubs/tensorflow/tensorflow/core/framework/model_pb2.pyi index 1fc78e2d7b8b..5f80b951a386 100644 --- a/stubs/tensorflow/tensorflow/core/framework/model_pb2.pyi +++ b/stubs/tensorflow/tensorflow/core/framework/model_pb2.pyi @@ -50,7 +50,9 @@ class _AutotuneAlgorithm: ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType -class _AutotuneAlgorithmEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_AutotuneAlgorithm.ValueType], builtins.type): +class _AutotuneAlgorithmEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_AutotuneAlgorithm.ValueType], builtins.type +): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor DEFAULT: _AutotuneAlgorithm.ValueType # 0 HILL_CLIMB: _AutotuneAlgorithm.ValueType # 1 @@ -118,7 +120,23 @@ class ModelProto(google.protobuf.message.Message): max: builtins.float | None = ..., tunable: builtins.bool | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["max", b"max", "min", b"min", "name", b"name", "state_value", b"state_value", "tunable", b"tunable", "value", b"value"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "max", + b"max", + "min", + b"min", + "name", + b"name", + "state_value", + b"state_value", + "tunable", + b"tunable", + "value", + b"value", + ], + ) -> None: ... ID_FIELD_NUMBER: builtins.int NAME_FIELD_NUMBER: builtins.int @@ -173,7 +191,9 @@ class ModelProto(google.protobuf.message.Message): buffered element. This is only used by ASYNC_KNOWN_RATIO nodes. """ @property - def parameters(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___ModelProto.Node.Parameter]: + def parameters( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___ModelProto.Node.Parameter]: """Parameters of this node.""" @property @@ -201,7 +221,45 @@ class ModelProto(google.protobuf.message.Message): ratio: builtins.float | None = ..., memory_ratio: builtins.float | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["autotune", b"autotune", "buffered_bytes", b"buffered_bytes", "buffered_elements", b"buffered_elements", "bytes_consumed", b"bytes_consumed", "bytes_produced", b"bytes_produced", "id", b"id", "input_processing_time_count", b"input_processing_time_count", "input_processing_time_sum", b"input_processing_time_sum", "inputs", b"inputs", "memory_ratio", b"memory_ratio", "name", b"name", "node_class", b"node_class", "num_elements", b"num_elements", "parameters", b"parameters", "processing_time", b"processing_time", "ratio", b"ratio", "record_metrics", b"record_metrics"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "autotune", + b"autotune", + "buffered_bytes", + b"buffered_bytes", + "buffered_elements", + b"buffered_elements", + "bytes_consumed", + b"bytes_consumed", + "bytes_produced", + b"bytes_produced", + "id", + b"id", + "input_processing_time_count", + b"input_processing_time_count", + "input_processing_time_sum", + b"input_processing_time_sum", + "inputs", + b"inputs", + "memory_ratio", + b"memory_ratio", + "name", + b"name", + "node_class", + b"node_class", + "num_elements", + b"num_elements", + "parameters", + b"parameters", + "processing_time", + b"processing_time", + "ratio", + b"ratio", + "record_metrics", + b"record_metrics", + ], + ) -> None: ... @typing.final class NodesEntry(google.protobuf.message.Message): @@ -212,12 +270,7 @@ class ModelProto(google.protobuf.message.Message): key: builtins.int @property def value(self) -> global___ModelProto.Node: ... - def __init__( - self, - *, - key: builtins.int | None = ..., - value: global___ModelProto.Node | None = ..., - ) -> None: ... + def __init__(self, *, key: builtins.int | None = ..., value: global___ModelProto.Node | None = ...) -> None: ... def HasField(self, field_name: typing.Literal["value", b"value"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["key", b"key", "value", b"value"]) -> None: ... @@ -249,7 +302,19 @@ class ModelProto(google.protobuf.message.Message): ram_budget: builtins.int | None = ..., model_input_time: builtins.float | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["algorithm", b"algorithm", "cpu_budget", b"cpu_budget", "model_input_time", b"model_input_time", "ram_budget", b"ram_budget"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "algorithm", + b"algorithm", + "cpu_budget", + b"cpu_budget", + "model_input_time", + b"model_input_time", + "ram_budget", + b"ram_budget", + ], + ) -> None: ... DATASET_NAME_FIELD_NUMBER: builtins.int NODES_FIELD_NUMBER: builtins.int @@ -282,6 +347,22 @@ class ModelProto(google.protobuf.message.Message): gap_times: collections.abc.Iterable[builtins.int] | None = ..., ) -> None: ... def HasField(self, field_name: typing.Literal["optimization_params", b"optimization_params"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["dataset_name", b"dataset_name", "gap_times", b"gap_times", "id_counter", b"id_counter", "nodes", b"nodes", "optimization_params", b"optimization_params", "output", b"output"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "dataset_name", + b"dataset_name", + "gap_times", + b"gap_times", + "id_counter", + b"id_counter", + "nodes", + b"nodes", + "optimization_params", + b"optimization_params", + "output", + b"output", + ], + ) -> None: ... global___ModelProto = ModelProto diff --git a/stubs/tensorflow/tensorflow/core/framework/node_def_pb2.pyi b/stubs/tensorflow/tensorflow/core/framework/node_def_pb2.pyi index 663e4d216ec9..510af1b62e6b 100644 --- a/stubs/tensorflow/tensorflow/core/framework/node_def_pb2.pyi +++ b/stubs/tensorflow/tensorflow/core/framework/node_def_pb2.pyi @@ -29,10 +29,7 @@ class NodeDef(google.protobuf.message.Message): @property def value(self) -> tensorflow.core.framework.attr_value_pb2.AttrValue: ... def __init__( - self, - *, - key: builtins.str | None = ..., - value: tensorflow.core.framework.attr_value_pb2.AttrValue | None = ..., + self, *, key: builtins.str | None = ..., value: tensorflow.core.framework.attr_value_pb2.AttrValue | None = ... ) -> None: ... def HasField(self, field_name: typing.Literal["value", b"value"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["key", b"key", "value", b"value"]) -> None: ... @@ -71,7 +68,12 @@ class NodeDef(google.protobuf.message.Message): original_node_names: collections.abc.Iterable[builtins.str] | None = ..., original_func_names: collections.abc.Iterable[builtins.str] | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["original_func_names", b"original_func_names", "original_node_names", b"original_node_names"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "original_func_names", b"original_func_names", "original_node_names", b"original_node_names" + ], + ) -> None: ... NAME_FIELD_NUMBER: builtins.int OP_FIELD_NUMBER: builtins.int @@ -121,7 +123,9 @@ class NodeDef(google.protobuf.message.Message): """ @property - def attr(self) -> google.protobuf.internal.containers.MessageMap[builtins.str, tensorflow.core.framework.attr_value_pb2.AttrValue]: + def attr( + self, + ) -> google.protobuf.internal.containers.MessageMap[builtins.str, tensorflow.core.framework.attr_value_pb2.AttrValue]: """Operation-specific graph-construction-time configuration. Note that this should include all attrs defined in the corresponding OpDef, including those with a value matching @@ -159,7 +163,30 @@ class NodeDef(google.protobuf.message.Message): experimental_debug_info: global___NodeDef.ExperimentalDebugInfo | None = ..., experimental_type: tensorflow.core.framework.full_type_pb2.FullTypeDef | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["experimental_debug_info", b"experimental_debug_info", "experimental_type", b"experimental_type"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["attr", b"attr", "device", b"device", "experimental_debug_info", b"experimental_debug_info", "experimental_type", b"experimental_type", "input", b"input", "name", b"name", "op", b"op"]) -> None: ... + def HasField( + self, + field_name: typing.Literal[ + "experimental_debug_info", b"experimental_debug_info", "experimental_type", b"experimental_type" + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "attr", + b"attr", + "device", + b"device", + "experimental_debug_info", + b"experimental_debug_info", + "experimental_type", + b"experimental_type", + "input", + b"input", + "name", + b"name", + "op", + b"op", + ], + ) -> None: ... global___NodeDef = NodeDef diff --git a/stubs/tensorflow/tensorflow/core/framework/op_def_pb2.pyi b/stubs/tensorflow/tensorflow/core/framework/op_def_pb2.pyi index 6656ddcadf3f..ee978cd9fac6 100644 --- a/stubs/tensorflow/tensorflow/core/framework/op_def_pb2.pyi +++ b/stubs/tensorflow/tensorflow/core/framework/op_def_pb2.pyi @@ -71,7 +71,11 @@ class OpDef(google.protobuf.message.Message): For outputs: if true, outputs are refs, otherwise they are not. """ @property - def handle_data(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[tensorflow.core.framework.resource_handle_pb2.ResourceHandleProto.DtypeAndShape]: + def handle_data( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[ + tensorflow.core.framework.resource_handle_pb2.ResourceHandleProto.DtypeAndShape + ]: """The handle data for resource inputs.""" @property @@ -96,12 +100,36 @@ class OpDef(google.protobuf.message.Message): type_attr: builtins.str | None = ..., number_attr: builtins.str | None = ..., type_list_attr: builtins.str | None = ..., - handle_data: collections.abc.Iterable[tensorflow.core.framework.resource_handle_pb2.ResourceHandleProto.DtypeAndShape] | None = ..., + handle_data: ( + collections.abc.Iterable[tensorflow.core.framework.resource_handle_pb2.ResourceHandleProto.DtypeAndShape] | None + ) = ..., is_ref: builtins.bool | None = ..., experimental_full_type: tensorflow.core.framework.full_type_pb2.FullTypeDef | None = ..., ) -> None: ... def HasField(self, field_name: typing.Literal["experimental_full_type", b"experimental_full_type"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["description", b"description", "experimental_full_type", b"experimental_full_type", "handle_data", b"handle_data", "is_ref", b"is_ref", "name", b"name", "number_attr", b"number_attr", "type", b"type", "type_attr", b"type_attr", "type_list_attr", b"type_list_attr"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "description", + b"description", + "experimental_full_type", + b"experimental_full_type", + "handle_data", + b"handle_data", + "is_ref", + b"is_ref", + "name", + b"name", + "number_attr", + b"number_attr", + "type", + b"type", + "type_attr", + b"type_attr", + "type_list_attr", + b"type_list_attr", + ], + ) -> None: ... @typing.final class AttrDef(google.protobuf.message.Message): @@ -166,8 +194,28 @@ class OpDef(google.protobuf.message.Message): minimum: builtins.int | None = ..., allowed_values: tensorflow.core.framework.attr_value_pb2.AttrValue | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["allowed_values", b"allowed_values", "default_value", b"default_value"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["allowed_values", b"allowed_values", "default_value", b"default_value", "description", b"description", "has_minimum", b"has_minimum", "minimum", b"minimum", "name", b"name", "type", b"type"]) -> None: ... + def HasField( + self, field_name: typing.Literal["allowed_values", b"allowed_values", "default_value", b"default_value"] + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "allowed_values", + b"allowed_values", + "default_value", + b"default_value", + "description", + b"description", + "has_minimum", + b"has_minimum", + "minimum", + b"minimum", + "name", + b"name", + "type", + b"type", + ], + ) -> None: ... NAME_FIELD_NUMBER: builtins.int INPUT_ARG_FIELD_NUMBER: builtins.int @@ -277,7 +325,37 @@ class OpDef(google.protobuf.message.Message): is_distributed_communication: builtins.bool | None = ..., ) -> None: ... def HasField(self, field_name: typing.Literal["deprecation", b"deprecation"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["allows_uninitialized_input", b"allows_uninitialized_input", "attr", b"attr", "control_output", b"control_output", "deprecation", b"deprecation", "description", b"description", "input_arg", b"input_arg", "is_aggregate", b"is_aggregate", "is_commutative", b"is_commutative", "is_distributed_communication", b"is_distributed_communication", "is_stateful", b"is_stateful", "name", b"name", "output_arg", b"output_arg", "summary", b"summary"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "allows_uninitialized_input", + b"allows_uninitialized_input", + "attr", + b"attr", + "control_output", + b"control_output", + "deprecation", + b"deprecation", + "description", + b"description", + "input_arg", + b"input_arg", + "is_aggregate", + b"is_aggregate", + "is_commutative", + b"is_commutative", + "is_distributed_communication", + b"is_distributed_communication", + "is_stateful", + b"is_stateful", + "name", + b"name", + "output_arg", + b"output_arg", + "summary", + b"summary", + ], + ) -> None: ... global___OpDef = OpDef @@ -293,12 +371,7 @@ class OpDeprecation(google.protobuf.message.Message): """First GraphDef version at which the op is disallowed.""" explanation: builtins.str """Explanation of why it was deprecated and what to use instead.""" - def __init__( - self, - *, - version: builtins.int | None = ..., - explanation: builtins.str | None = ..., - ) -> None: ... + def __init__(self, *, version: builtins.int | None = ..., explanation: builtins.str | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["explanation", b"explanation", "version", b"version"]) -> None: ... global___OpDeprecation = OpDeprecation @@ -312,11 +385,7 @@ class OpList(google.protobuf.message.Message): OP_FIELD_NUMBER: builtins.int @property def op(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___OpDef]: ... - def __init__( - self, - *, - op: collections.abc.Iterable[global___OpDef] | None = ..., - ) -> None: ... + def __init__(self, *, op: collections.abc.Iterable[global___OpDef] | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["op", b"op"]) -> None: ... global___OpList = OpList diff --git a/stubs/tensorflow/tensorflow/core/framework/optimized_function_graph_pb2.pyi b/stubs/tensorflow/tensorflow/core/framework/optimized_function_graph_pb2.pyi index 7f48d2d4b519..152ce34664ac 100644 --- a/stubs/tensorflow/tensorflow/core/framework/optimized_function_graph_pb2.pyi +++ b/stubs/tensorflow/tensorflow/core/framework/optimized_function_graph_pb2.pyi @@ -36,7 +36,10 @@ class OptimizedFunctionGraph(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _OptimizationSourceEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[OptimizedFunctionGraph._OptimizationSource.ValueType], builtins.type): + class _OptimizationSourceEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[OptimizedFunctionGraph._OptimizationSource.ValueType], + builtins.type, + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor SOURCE_UNSPECIFIED: OptimizedFunctionGraph._OptimizationSource.ValueType # 0 AOT: OptimizedFunctionGraph._OptimizationSource.ValueType # 1 @@ -64,12 +67,7 @@ class OptimizedFunctionGraph(google.protobuf.message.Message): VALUE_FIELD_NUMBER: builtins.int key: builtins.str value: builtins.str - def __init__( - self, - *, - key: builtins.str | None = ..., - value: builtins.str | None = ..., - ) -> None: ... + def __init__(self, *, key: builtins.str | None = ..., value: builtins.str | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["key", b"key", "value", b"value"]) -> None: ... NAME_FIELD_NUMBER: builtins.int @@ -102,7 +100,9 @@ class OptimizedFunctionGraph(google.protobuf.message.Message): """ @property - def ret_types(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[tensorflow.core.framework.types_pb2.DataType.ValueType]: + def ret_types( + self, + ) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[tensorflow.core.framework.types_pb2.DataType.ValueType]: """Return node types of the function. This is an output of graph preprocessing. """ @@ -118,10 +118,48 @@ class OptimizedFunctionGraph(google.protobuf.message.Message): source: global___OptimizedFunctionGraph.OptimizationSource.ValueType | None = ..., optimization_time_usecs: builtins.int | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["_optimization_time_usecs", b"_optimization_time_usecs", "_source", b"_source", "function_graph", b"function_graph", "optimization_time_usecs", b"optimization_time_usecs", "source", b"source"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["_optimization_time_usecs", b"_optimization_time_usecs", "_source", b"_source", "function_graph", b"function_graph", "name", b"name", "node_name_to_control_ret", b"node_name_to_control_ret", "num_return_nodes", b"num_return_nodes", "optimization_time_usecs", b"optimization_time_usecs", "ret_types", b"ret_types", "source", b"source"]) -> None: ... + def HasField( + self, + field_name: typing.Literal[ + "_optimization_time_usecs", + b"_optimization_time_usecs", + "_source", + b"_source", + "function_graph", + b"function_graph", + "optimization_time_usecs", + b"optimization_time_usecs", + "source", + b"source", + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "_optimization_time_usecs", + b"_optimization_time_usecs", + "_source", + b"_source", + "function_graph", + b"function_graph", + "name", + b"name", + "node_name_to_control_ret", + b"node_name_to_control_ret", + "num_return_nodes", + b"num_return_nodes", + "optimization_time_usecs", + b"optimization_time_usecs", + "ret_types", + b"ret_types", + "source", + b"source", + ], + ) -> None: ... @typing.overload - def WhichOneof(self, oneof_group: typing.Literal["_optimization_time_usecs", b"_optimization_time_usecs"]) -> typing.Literal["optimization_time_usecs"] | None: ... + def WhichOneof( + self, oneof_group: typing.Literal["_optimization_time_usecs", b"_optimization_time_usecs"] + ) -> typing.Literal["optimization_time_usecs"] | None: ... @typing.overload def WhichOneof(self, oneof_group: typing.Literal["_source", b"_source"]) -> typing.Literal["source"] | None: ... diff --git a/stubs/tensorflow/tensorflow/core/framework/reader_base_pb2.pyi b/stubs/tensorflow/tensorflow/core/framework/reader_base_pb2.pyi index 7e6e6ea31932..0ae5cb8b33fe 100644 --- a/stubs/tensorflow/tensorflow/core/framework/reader_base_pb2.pyi +++ b/stubs/tensorflow/tensorflow/core/framework/reader_base_pb2.pyi @@ -35,6 +35,18 @@ class ReaderBaseState(google.protobuf.message.Message): num_records_produced: builtins.int | None = ..., current_work: builtins.bytes | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["current_work", b"current_work", "num_records_produced", b"num_records_produced", "work_finished", b"work_finished", "work_started", b"work_started"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "current_work", + b"current_work", + "num_records_produced", + b"num_records_produced", + "work_finished", + b"work_finished", + "work_started", + b"work_started", + ], + ) -> None: ... global___ReaderBaseState = ReaderBaseState diff --git a/stubs/tensorflow/tensorflow/core/framework/resource_handle_pb2.pyi b/stubs/tensorflow/tensorflow/core/framework/resource_handle_pb2.pyi index 8767e1768bce..aeb126eae7c6 100644 --- a/stubs/tensorflow/tensorflow/core/framework/resource_handle_pb2.pyi +++ b/stubs/tensorflow/tensorflow/core/framework/resource_handle_pb2.pyi @@ -68,7 +68,9 @@ class ResourceHandleProto(google.protobuf.message.Message): available. """ @property - def dtypes_and_shapes(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___ResourceHandleProto.DtypeAndShape]: + def dtypes_and_shapes( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___ResourceHandleProto.DtypeAndShape]: """Data types and shapes for the underlying resource.""" def __init__( @@ -81,6 +83,22 @@ class ResourceHandleProto(google.protobuf.message.Message): maybe_type_name: builtins.str | None = ..., dtypes_and_shapes: collections.abc.Iterable[global___ResourceHandleProto.DtypeAndShape] | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["container", b"container", "device", b"device", "dtypes_and_shapes", b"dtypes_and_shapes", "hash_code", b"hash_code", "maybe_type_name", b"maybe_type_name", "name", b"name"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "container", + b"container", + "device", + b"device", + "dtypes_and_shapes", + b"dtypes_and_shapes", + "hash_code", + b"hash_code", + "maybe_type_name", + b"maybe_type_name", + "name", + b"name", + ], + ) -> None: ... global___ResourceHandleProto = ResourceHandleProto diff --git a/stubs/tensorflow/tensorflow/core/framework/step_stats_pb2.pyi b/stubs/tensorflow/tensorflow/core/framework/step_stats_pb2.pyi index 947cbdb41683..1bcf58a8e510 100644 --- a/stubs/tensorflow/tensorflow/core/framework/step_stats_pb2.pyi +++ b/stubs/tensorflow/tensorflow/core/framework/step_stats_pb2.pyi @@ -27,12 +27,7 @@ class AllocationRecord(google.protobuf.message.Message): """The timestamp of the operation.""" alloc_bytes: builtins.int """Number of bytes allocated, or de-allocated if negative.""" - def __init__( - self, - *, - alloc_micros: builtins.int | None = ..., - alloc_bytes: builtins.int | None = ..., - ) -> None: ... + def __init__(self, *, alloc_micros: builtins.int | None = ..., alloc_bytes: builtins.int | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["alloc_bytes", b"alloc_bytes", "alloc_micros", b"alloc_micros"]) -> None: ... global___AllocationRecord = AllocationRecord @@ -58,7 +53,9 @@ class AllocatorMemoryUsed(google.protobuf.message.Message): The number of live bytes currently allocated by the allocator. """ @property - def allocation_records(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___AllocationRecord]: + def allocation_records( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___AllocationRecord]: """The allocation and deallocation timeline.""" def __init__( @@ -71,7 +68,23 @@ class AllocatorMemoryUsed(google.protobuf.message.Message): allocation_records: collections.abc.Iterable[global___AllocationRecord] | None = ..., allocator_bytes_in_use: builtins.int | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["allocation_records", b"allocation_records", "allocator_bytes_in_use", b"allocator_bytes_in_use", "allocator_name", b"allocator_name", "live_bytes", b"live_bytes", "peak_bytes", b"peak_bytes", "total_bytes", b"total_bytes"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "allocation_records", + b"allocation_records", + "allocator_bytes_in_use", + b"allocator_bytes_in_use", + "allocator_name", + b"allocator_name", + "live_bytes", + b"live_bytes", + "peak_bytes", + b"peak_bytes", + "total_bytes", + b"total_bytes", + ], + ) -> None: ... global___AllocatorMemoryUsed = AllocatorMemoryUsed @@ -116,7 +129,9 @@ class MemoryStats(google.protobuf.message.Message): @property def persistent_tensor_alloc_ids(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.int]: ... @property - def device_persistent_tensor_alloc_ids(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.int]: ... + def device_persistent_tensor_alloc_ids( + self, + ) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.int]: ... def __init__( self, *, @@ -127,7 +142,23 @@ class MemoryStats(google.protobuf.message.Message): device_persistent_memory_size: builtins.int | None = ..., device_persistent_tensor_alloc_ids: collections.abc.Iterable[builtins.int] | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["device_persistent_memory_size", b"device_persistent_memory_size", "device_persistent_tensor_alloc_ids", b"device_persistent_tensor_alloc_ids", "device_temp_memory_size", b"device_temp_memory_size", "persistent_memory_size", b"persistent_memory_size", "persistent_tensor_alloc_ids", b"persistent_tensor_alloc_ids", "temp_memory_size", b"temp_memory_size"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "device_persistent_memory_size", + b"device_persistent_memory_size", + "device_persistent_tensor_alloc_ids", + b"device_persistent_tensor_alloc_ids", + "device_temp_memory_size", + b"device_temp_memory_size", + "persistent_memory_size", + b"persistent_memory_size", + "persistent_tensor_alloc_ids", + b"persistent_tensor_alloc_ids", + "temp_memory_size", + b"temp_memory_size", + ], + ) -> None: ... global___MemoryStats = MemoryStats @@ -177,7 +208,11 @@ class NodeExecStats(google.protobuf.message.Message): @property def output(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___NodeOutput]: ... @property - def referenced_tensor(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[tensorflow.core.framework.allocation_description_pb2.AllocationDescription]: ... + def referenced_tensor( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[ + tensorflow.core.framework.allocation_description_pb2.AllocationDescription + ]: ... @property def memory_stats(self) -> global___MemoryStats: ... def __init__( @@ -193,7 +228,9 @@ class NodeExecStats(google.protobuf.message.Message): timeline_label: builtins.str | None = ..., scheduled_micros: builtins.int | None = ..., thread_id: builtins.int | None = ..., - referenced_tensor: collections.abc.Iterable[tensorflow.core.framework.allocation_description_pb2.AllocationDescription] | None = ..., + referenced_tensor: ( + collections.abc.Iterable[tensorflow.core.framework.allocation_description_pb2.AllocationDescription] | None + ) = ..., memory_stats: global___MemoryStats | None = ..., all_start_nanos: builtins.int | None = ..., op_start_rel_nanos: builtins.int | None = ..., @@ -202,7 +239,45 @@ class NodeExecStats(google.protobuf.message.Message): scheduled_nanos: builtins.int | None = ..., ) -> None: ... def HasField(self, field_name: typing.Literal["memory_stats", b"memory_stats"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["all_end_rel_micros", b"all_end_rel_micros", "all_end_rel_nanos", b"all_end_rel_nanos", "all_start_micros", b"all_start_micros", "all_start_nanos", b"all_start_nanos", "memory", b"memory", "memory_stats", b"memory_stats", "node_name", b"node_name", "op_end_rel_micros", b"op_end_rel_micros", "op_end_rel_nanos", b"op_end_rel_nanos", "op_start_rel_micros", b"op_start_rel_micros", "op_start_rel_nanos", b"op_start_rel_nanos", "output", b"output", "referenced_tensor", b"referenced_tensor", "scheduled_micros", b"scheduled_micros", "scheduled_nanos", b"scheduled_nanos", "thread_id", b"thread_id", "timeline_label", b"timeline_label"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "all_end_rel_micros", + b"all_end_rel_micros", + "all_end_rel_nanos", + b"all_end_rel_nanos", + "all_start_micros", + b"all_start_micros", + "all_start_nanos", + b"all_start_nanos", + "memory", + b"memory", + "memory_stats", + b"memory_stats", + "node_name", + b"node_name", + "op_end_rel_micros", + b"op_end_rel_micros", + "op_end_rel_nanos", + b"op_end_rel_nanos", + "op_start_rel_micros", + b"op_start_rel_micros", + "op_start_rel_nanos", + b"op_start_rel_nanos", + "output", + b"output", + "referenced_tensor", + b"referenced_tensor", + "scheduled_micros", + b"scheduled_micros", + "scheduled_nanos", + b"scheduled_nanos", + "thread_id", + b"thread_id", + "timeline_label", + b"timeline_label", + ], + ) -> None: ... global___NodeExecStats = NodeExecStats @@ -218,12 +293,7 @@ class DeviceStepStats(google.protobuf.message.Message): VALUE_FIELD_NUMBER: builtins.int key: builtins.int value: builtins.str - def __init__( - self, - *, - key: builtins.int | None = ..., - value: builtins.str | None = ..., - ) -> None: ... + def __init__(self, *, key: builtins.int | None = ..., value: builtins.str | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["key", b"key", "value", b"value"]) -> None: ... DEVICE_FIELD_NUMBER: builtins.int @@ -243,7 +313,9 @@ class DeviceStepStats(google.protobuf.message.Message): node_stats: collections.abc.Iterable[global___NodeExecStats] | None = ..., thread_names: collections.abc.Mapping[builtins.int, builtins.str] | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["device", b"device", "node_stats", b"node_stats", "thread_names", b"thread_names"]) -> None: ... + def ClearField( + self, field_name: typing.Literal["device", b"device", "node_stats", b"node_stats", "thread_names", b"thread_names"] + ) -> None: ... global___DeviceStepStats = DeviceStepStats @@ -254,11 +326,7 @@ class StepStats(google.protobuf.message.Message): DEV_STATS_FIELD_NUMBER: builtins.int @property def dev_stats(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___DeviceStepStats]: ... - def __init__( - self, - *, - dev_stats: collections.abc.Iterable[global___DeviceStepStats] | None = ..., - ) -> None: ... + def __init__(self, *, dev_stats: collections.abc.Iterable[global___DeviceStepStats] | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["dev_stats", b"dev_stats"]) -> None: ... global___StepStats = StepStats diff --git a/stubs/tensorflow/tensorflow/core/framework/summary_pb2.pyi b/stubs/tensorflow/tensorflow/core/framework/summary_pb2.pyi index 5df66b7e7978..848d802496f8 100644 --- a/stubs/tensorflow/tensorflow/core/framework/summary_pb2.pyi +++ b/stubs/tensorflow/tensorflow/core/framework/summary_pb2.pyi @@ -81,11 +81,7 @@ class SummaryDescription(google.protobuf.message.Message): """Hint on how plugins should process the data in this series. Supported values include "scalar", "histogram", "image", "audio" """ - def __init__( - self, - *, - type_hint: builtins.str | None = ..., - ) -> None: ... + def __init__(self, *, type_hint: builtins.str | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["type_hint", b"type_hint"]) -> None: ... global___SummaryDescription = SummaryDescription @@ -110,12 +106,7 @@ class SummaryMetadata(google.protobuf.message.Message): """The content to store for the plugin. The best practice is for this to be a binary serialized protocol buffer. """ - def __init__( - self, - *, - plugin_name: builtins.str | None = ..., - content: builtins.bytes | None = ..., - ) -> None: ... + def __init__(self, *, plugin_name: builtins.str | None = ..., content: builtins.bytes | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["content", b"content", "plugin_name", b"plugin_name"]) -> None: ... PLUGIN_DATA_FIELD_NUMBER: builtins.int @@ -145,7 +136,19 @@ class SummaryMetadata(google.protobuf.message.Message): data_class: global___DataClass.ValueType | None = ..., ) -> None: ... def HasField(self, field_name: typing.Literal["plugin_data", b"plugin_data"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["data_class", b"data_class", "display_name", b"display_name", "plugin_data", b"plugin_data", "summary_description", b"summary_description"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "data_class", + b"data_class", + "display_name", + b"display_name", + "plugin_data", + b"plugin_data", + "summary_description", + b"summary_description", + ], + ) -> None: ... global___SummaryMetadata = SummaryMetadata @@ -193,7 +196,19 @@ class Summary(google.protobuf.message.Message): colorspace: builtins.int | None = ..., encoded_image_string: builtins.bytes | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["colorspace", b"colorspace", "encoded_image_string", b"encoded_image_string", "height", b"height", "width", b"width"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "colorspace", + b"colorspace", + "encoded_image_string", + b"encoded_image_string", + "height", + b"height", + "width", + b"width", + ], + ) -> None: ... @typing.final class Audio(google.protobuf.message.Message): @@ -224,7 +239,21 @@ class Summary(google.protobuf.message.Message): encoded_audio_string: builtins.bytes | None = ..., content_type: builtins.str | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["content_type", b"content_type", "encoded_audio_string", b"encoded_audio_string", "length_frames", b"length_frames", "num_channels", b"num_channels", "sample_rate", b"sample_rate"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "content_type", + b"content_type", + "encoded_audio_string", + b"encoded_audio_string", + "length_frames", + b"length_frames", + "num_channels", + b"num_channels", + "sample_rate", + b"sample_rate", + ], + ) -> None: ... @typing.final class Value(google.protobuf.message.Message): @@ -278,20 +307,62 @@ class Summary(google.protobuf.message.Message): audio: global___Summary.Audio | None = ..., tensor: tensorflow.core.framework.tensor_pb2.TensorProto | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["audio", b"audio", "histo", b"histo", "image", b"image", "metadata", b"metadata", "obsolete_old_style_histogram", b"obsolete_old_style_histogram", "simple_value", b"simple_value", "tensor", b"tensor", "value", b"value"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["audio", b"audio", "histo", b"histo", "image", b"image", "metadata", b"metadata", "node_name", b"node_name", "obsolete_old_style_histogram", b"obsolete_old_style_histogram", "simple_value", b"simple_value", "tag", b"tag", "tensor", b"tensor", "value", b"value"]) -> None: ... - def WhichOneof(self, oneof_group: typing.Literal["value", b"value"]) -> typing.Literal["simple_value", "obsolete_old_style_histogram", "image", "histo", "audio", "tensor"] | None: ... + def HasField( + self, + field_name: typing.Literal[ + "audio", + b"audio", + "histo", + b"histo", + "image", + b"image", + "metadata", + b"metadata", + "obsolete_old_style_histogram", + b"obsolete_old_style_histogram", + "simple_value", + b"simple_value", + "tensor", + b"tensor", + "value", + b"value", + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "audio", + b"audio", + "histo", + b"histo", + "image", + b"image", + "metadata", + b"metadata", + "node_name", + b"node_name", + "obsolete_old_style_histogram", + b"obsolete_old_style_histogram", + "simple_value", + b"simple_value", + "tag", + b"tag", + "tensor", + b"tensor", + "value", + b"value", + ], + ) -> None: ... + def WhichOneof( + self, oneof_group: typing.Literal["value", b"value"] + ) -> typing.Literal["simple_value", "obsolete_old_style_histogram", "image", "histo", "audio", "tensor"] | None: ... VALUE_FIELD_NUMBER: builtins.int @property def value(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___Summary.Value]: """Set of values for the summary.""" - def __init__( - self, - *, - value: collections.abc.Iterable[global___Summary.Value] | None = ..., - ) -> None: ... + def __init__(self, *, value: collections.abc.Iterable[global___Summary.Value] | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["value", b"value"]) -> None: ... global___Summary = Summary diff --git a/stubs/tensorflow/tensorflow/core/framework/tensor_description_pb2.pyi b/stubs/tensorflow/tensorflow/core/framework/tensor_description_pb2.pyi index 71a5df5340a2..3062402cc3c5 100644 --- a/stubs/tensorflow/tensorflow/core/framework/tensor_description_pb2.pyi +++ b/stubs/tensorflow/tensorflow/core/framework/tensor_description_pb2.pyi @@ -38,7 +38,12 @@ class TensorDescription(google.protobuf.message.Message): shape: tensorflow.core.framework.tensor_shape_pb2.TensorShapeProto | None = ..., allocation_description: tensorflow.core.framework.allocation_description_pb2.AllocationDescription | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["allocation_description", b"allocation_description", "shape", b"shape"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["allocation_description", b"allocation_description", "dtype", b"dtype", "shape", b"shape"]) -> None: ... + def HasField( + self, field_name: typing.Literal["allocation_description", b"allocation_description", "shape", b"shape"] + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal["allocation_description", b"allocation_description", "dtype", b"dtype", "shape", b"shape"], + ) -> None: ... global___TensorDescription = TensorDescription diff --git a/stubs/tensorflow/tensorflow/core/framework/tensor_pb2.pyi b/stubs/tensorflow/tensorflow/core/framework/tensor_pb2.pyi index 8e1a764d7021..40edf767d049 100644 --- a/stubs/tensorflow/tensorflow/core/framework/tensor_pb2.pyi +++ b/stubs/tensorflow/tensorflow/core/framework/tensor_pb2.pyi @@ -116,7 +116,11 @@ class TensorProto(google.protobuf.message.Message): """ @property - def resource_handle_val(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[tensorflow.core.framework.resource_handle_pb2.ResourceHandleProto]: + def resource_handle_val( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[ + tensorflow.core.framework.resource_handle_pb2.ResourceHandleProto + ]: """DT_RESOURCE""" @property @@ -147,14 +151,56 @@ class TensorProto(google.protobuf.message.Message): int64_val: collections.abc.Iterable[builtins.int] | None = ..., bool_val: collections.abc.Iterable[builtins.bool] | None = ..., dcomplex_val: collections.abc.Iterable[builtins.float] | None = ..., - resource_handle_val: collections.abc.Iterable[tensorflow.core.framework.resource_handle_pb2.ResourceHandleProto] | None = ..., + resource_handle_val: ( + collections.abc.Iterable[tensorflow.core.framework.resource_handle_pb2.ResourceHandleProto] | None + ) = ..., variant_val: collections.abc.Iterable[global___VariantTensorDataProto] | None = ..., uint32_val: collections.abc.Iterable[builtins.int] | None = ..., uint64_val: collections.abc.Iterable[builtins.int] | None = ..., float8_val: builtins.bytes | None = ..., ) -> None: ... def HasField(self, field_name: typing.Literal["tensor_shape", b"tensor_shape"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["bool_val", b"bool_val", "dcomplex_val", b"dcomplex_val", "double_val", b"double_val", "dtype", b"dtype", "float8_val", b"float8_val", "float_val", b"float_val", "half_val", b"half_val", "int64_val", b"int64_val", "int_val", b"int_val", "resource_handle_val", b"resource_handle_val", "scomplex_val", b"scomplex_val", "string_val", b"string_val", "tensor_content", b"tensor_content", "tensor_shape", b"tensor_shape", "uint32_val", b"uint32_val", "uint64_val", b"uint64_val", "variant_val", b"variant_val", "version_number", b"version_number"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "bool_val", + b"bool_val", + "dcomplex_val", + b"dcomplex_val", + "double_val", + b"double_val", + "dtype", + b"dtype", + "float8_val", + b"float8_val", + "float_val", + b"float_val", + "half_val", + b"half_val", + "int64_val", + b"int64_val", + "int_val", + b"int_val", + "resource_handle_val", + b"resource_handle_val", + "scomplex_val", + b"scomplex_val", + "string_val", + b"string_val", + "tensor_content", + b"tensor_content", + "tensor_shape", + b"tensor_shape", + "uint32_val", + b"uint32_val", + "uint64_val", + b"uint64_val", + "variant_val", + b"variant_val", + "version_number", + b"version_number", + ], + ) -> None: ... global___TensorProto = TensorProto @@ -182,6 +228,8 @@ class VariantTensorDataProto(google.protobuf.message.Message): metadata: builtins.bytes | None = ..., tensors: collections.abc.Iterable[global___TensorProto] | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["metadata", b"metadata", "tensors", b"tensors", "type_name", b"type_name"]) -> None: ... + def ClearField( + self, field_name: typing.Literal["metadata", b"metadata", "tensors", b"tensors", "type_name", b"type_name"] + ) -> None: ... global___VariantTensorDataProto = VariantTensorDataProto diff --git a/stubs/tensorflow/tensorflow/core/framework/tensor_shape_pb2.pyi b/stubs/tensorflow/tensorflow/core/framework/tensor_shape_pb2.pyi index 2d852a920271..908b1159b8e2 100644 --- a/stubs/tensorflow/tensorflow/core/framework/tensor_shape_pb2.pyi +++ b/stubs/tensorflow/tensorflow/core/framework/tensor_shape_pb2.pyi @@ -36,12 +36,7 @@ class TensorShapeProto(google.protobuf.message.Message): """ name: builtins.str """Optional name of the tensor dimension.""" - def __init__( - self, - *, - size: builtins.int | None = ..., - name: builtins.str | None = ..., - ) -> None: ... + def __init__(self, *, size: builtins.int | None = ..., name: builtins.str | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["name", b"name", "size", b"size"]) -> None: ... DIM_FIELD_NUMBER: builtins.int diff --git a/stubs/tensorflow/tensorflow/core/framework/tensor_slice_pb2.pyi b/stubs/tensorflow/tensorflow/core/framework/tensor_slice_pb2.pyi index 949fd0f1d6a3..1ad03f3c5859 100644 --- a/stubs/tensorflow/tensorflow/core/framework/tensor_slice_pb2.pyi +++ b/stubs/tensorflow/tensorflow/core/framework/tensor_slice_pb2.pyi @@ -33,14 +33,11 @@ class TensorSliceProto(google.protobuf.message.Message): start: builtins.int """Start index of the slice, starting at 0.""" length: builtins.int - def __init__( - self, - *, - start: builtins.int | None = ..., - length: builtins.int | None = ..., - ) -> None: ... + def __init__(self, *, start: builtins.int | None = ..., length: builtins.int | None = ...) -> None: ... def HasField(self, field_name: typing.Literal["has_length", b"has_length", "length", b"length"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["has_length", b"has_length", "length", b"length", "start", b"start"]) -> None: ... + def ClearField( + self, field_name: typing.Literal["has_length", b"has_length", "length", b"length", "start", b"start"] + ) -> None: ... def WhichOneof(self, oneof_group: typing.Literal["has_length", b"has_length"]) -> typing.Literal["length"] | None: ... EXTENT_FIELD_NUMBER: builtins.int @@ -53,11 +50,7 @@ class TensorSliceProto(google.protobuf.message.Message): dimensions in the TensorShape. """ - def __init__( - self, - *, - extent: collections.abc.Iterable[global___TensorSliceProto.Extent] | None = ..., - ) -> None: ... + def __init__(self, *, extent: collections.abc.Iterable[global___TensorSliceProto.Extent] | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["extent", b"extent"]) -> None: ... global___TensorSliceProto = TensorSliceProto diff --git a/stubs/tensorflow/tensorflow/core/framework/types_pb2.pyi b/stubs/tensorflow/tensorflow/core/framework/types_pb2.pyi index a07d25c40e0e..18c74011593b 100644 --- a/stubs/tensorflow/tensorflow/core/framework/types_pb2.pyi +++ b/stubs/tensorflow/tensorflow/core/framework/types_pb2.pyi @@ -214,11 +214,7 @@ class SerializedDType(google.protobuf.message.Message): DATATYPE_FIELD_NUMBER: builtins.int datatype: global___DataType.ValueType - def __init__( - self, - *, - datatype: global___DataType.ValueType | None = ..., - ) -> None: ... + def __init__(self, *, datatype: global___DataType.ValueType | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["datatype", b"datatype"]) -> None: ... global___SerializedDType = SerializedDType diff --git a/stubs/tensorflow/tensorflow/core/framework/variable_pb2.pyi b/stubs/tensorflow/tensorflow/core/framework/variable_pb2.pyi index 2d2320096e66..1341eee711e5 100644 --- a/stubs/tensorflow/tensorflow/core/framework/variable_pb2.pyi +++ b/stubs/tensorflow/tensorflow/core/framework/variable_pb2.pyi @@ -24,7 +24,9 @@ class _VariableSynchronization: ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType -class _VariableSynchronizationEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_VariableSynchronization.ValueType], builtins.type): +class _VariableSynchronizationEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_VariableSynchronization.ValueType], builtins.type +): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor VARIABLE_SYNCHRONIZATION_AUTO: _VariableSynchronization.ValueType # 0 """`AUTO`: Indicates that the synchronization will be determined by the @@ -72,7 +74,9 @@ class _VariableAggregation: ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType -class _VariableAggregationEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_VariableAggregation.ValueType], builtins.type): +class _VariableAggregationEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_VariableAggregation.ValueType], builtins.type +): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor VARIABLE_AGGREGATION_NONE: _VariableAggregation.ValueType # 0 """`NONE`: This is the default, giving an error if you use a @@ -159,7 +163,29 @@ class VariableDef(google.protobuf.message.Message): aggregation: global___VariableAggregation.ValueType | None = ..., ) -> None: ... def HasField(self, field_name: typing.Literal["save_slice_info_def", b"save_slice_info_def"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["aggregation", b"aggregation", "initial_value_name", b"initial_value_name", "initializer_name", b"initializer_name", "is_resource", b"is_resource", "save_slice_info_def", b"save_slice_info_def", "snapshot_name", b"snapshot_name", "synchronization", b"synchronization", "trainable", b"trainable", "variable_name", b"variable_name"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "aggregation", + b"aggregation", + "initial_value_name", + b"initial_value_name", + "initializer_name", + b"initializer_name", + "is_resource", + b"is_resource", + "save_slice_info_def", + b"save_slice_info_def", + "snapshot_name", + b"snapshot_name", + "synchronization", + b"synchronization", + "trainable", + b"trainable", + "variable_name", + b"variable_name", + ], + ) -> None: ... global___VariableDef = VariableDef @@ -193,6 +219,11 @@ class SaveSliceInfoDef(google.protobuf.message.Message): var_offset: collections.abc.Iterable[builtins.int] | None = ..., var_shape: collections.abc.Iterable[builtins.int] | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["full_name", b"full_name", "full_shape", b"full_shape", "var_offset", b"var_offset", "var_shape", b"var_shape"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "full_name", b"full_name", "full_shape", b"full_shape", "var_offset", b"var_offset", "var_shape", b"var_shape" + ], + ) -> None: ... global___SaveSliceInfoDef = SaveSliceInfoDef diff --git a/stubs/tensorflow/tensorflow/core/framework/versions_pb2.pyi b/stubs/tensorflow/tensorflow/core/framework/versions_pb2.pyi index 7ca17c36bdfd..040b4932af13 100644 --- a/stubs/tensorflow/tensorflow/core/framework/versions_pb2.pyi +++ b/stubs/tensorflow/tensorflow/core/framework/versions_pb2.pyi @@ -49,6 +49,9 @@ class VersionDef(google.protobuf.message.Message): min_consumer: builtins.int | None = ..., bad_consumers: collections.abc.Iterable[builtins.int] | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["bad_consumers", b"bad_consumers", "min_consumer", b"min_consumer", "producer", b"producer"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal["bad_consumers", b"bad_consumers", "min_consumer", b"min_consumer", "producer", b"producer"], + ) -> None: ... global___VersionDef = VersionDef diff --git a/stubs/tensorflow/tensorflow/core/protobuf/cluster_pb2.pyi b/stubs/tensorflow/tensorflow/core/protobuf/cluster_pb2.pyi index ae19698d703d..0b4f68cbf42b 100644 --- a/stubs/tensorflow/tensorflow/core/protobuf/cluster_pb2.pyi +++ b/stubs/tensorflow/tensorflow/core/protobuf/cluster_pb2.pyi @@ -83,12 +83,7 @@ class JobDef(google.protobuf.message.Message): VALUE_FIELD_NUMBER: builtins.int key: builtins.int value: builtins.str - def __init__( - self, - *, - key: builtins.int | None = ..., - value: builtins.str | None = ..., - ) -> None: ... + def __init__(self, *, key: builtins.int | None = ..., value: builtins.str | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["key", b"key", "value", b"value"]) -> None: ... NAME_FIELD_NUMBER: builtins.int @@ -108,10 +103,7 @@ class JobDef(google.protobuf.message.Message): """ def __init__( - self, - *, - name: builtins.str | None = ..., - tasks: collections.abc.Mapping[builtins.int, builtins.str] | None = ..., + self, *, name: builtins.str | None = ..., tasks: collections.abc.Mapping[builtins.int, builtins.str] | None = ... ) -> None: ... def ClearField(self, field_name: typing.Literal["name", b"name", "tasks", b"tasks"]) -> None: ... @@ -128,11 +120,7 @@ class ClusterDef(google.protobuf.message.Message): def job(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___JobDef]: """The jobs that comprise the cluster.""" - def __init__( - self, - *, - job: collections.abc.Iterable[global___JobDef] | None = ..., - ) -> None: ... + def __init__(self, *, job: collections.abc.Iterable[global___JobDef] | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["job", b"job"]) -> None: ... global___ClusterDef = ClusterDef diff --git a/stubs/tensorflow/tensorflow/core/protobuf/composite_tensor_variant_pb2.pyi b/stubs/tensorflow/tensorflow/core/protobuf/composite_tensor_variant_pb2.pyi index 1dbe6e5573b5..e17278f1946a 100644 --- a/stubs/tensorflow/tensorflow/core/protobuf/composite_tensor_variant_pb2.pyi +++ b/stubs/tensorflow/tensorflow/core/protobuf/composite_tensor_variant_pb2.pyi @@ -26,11 +26,7 @@ class CompositeTensorVariantMetadata(google.protobuf.message.Message): TYPE_SPEC_PROTO_FIELD_NUMBER: builtins.int @property def type_spec_proto(self) -> tensorflow.core.protobuf.struct_pb2.TypeSpecProto: ... - def __init__( - self, - *, - type_spec_proto: tensorflow.core.protobuf.struct_pb2.TypeSpecProto | None = ..., - ) -> None: ... + def __init__(self, *, type_spec_proto: tensorflow.core.protobuf.struct_pb2.TypeSpecProto | None = ...) -> None: ... def HasField(self, field_name: typing.Literal["type_spec_proto", b"type_spec_proto"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["type_spec_proto", b"type_spec_proto"]) -> None: ... diff --git a/stubs/tensorflow/tensorflow/core/protobuf/config_pb2.pyi b/stubs/tensorflow/tensorflow/core/protobuf/config_pb2.pyi index a655ef204b81..1930db2b79dd 100644 --- a/stubs/tensorflow/tensorflow/core/protobuf/config_pb2.pyi +++ b/stubs/tensorflow/tensorflow/core/protobuf/config_pb2.pyi @@ -88,7 +88,12 @@ class GPUOptions(google.protobuf.message.Message): priority: collections.abc.Iterable[builtins.int] | None = ..., device_ordinal: collections.abc.Iterable[builtins.int] | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["device_ordinal", b"device_ordinal", "memory_limit_mb", b"memory_limit_mb", "priority", b"priority"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "device_ordinal", b"device_ordinal", "memory_limit_mb", b"memory_limit_mb", "priority", b"priority" + ], + ) -> None: ... @typing.final class StreamMergeOptions(google.protobuf.message.Message): @@ -131,7 +136,17 @@ class GPUOptions(google.protobuf.message.Message): merge_device_to_host_stream: builtins.bool | None = ..., merge_device_to_device_stream: builtins.bool | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["merge_device_to_device_stream", b"merge_device_to_device_stream", "merge_device_to_host_stream", b"merge_device_to_host_stream", "merge_host_to_device_stream", b"merge_host_to_device_stream"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "merge_device_to_device_stream", + b"merge_device_to_device_stream", + "merge_device_to_host_stream", + b"merge_device_to_host_stream", + "merge_host_to_device_stream", + b"merge_host_to_device_stream", + ], + ) -> None: ... VIRTUAL_DEVICES_FIELD_NUMBER: builtins.int NUM_VIRTUAL_DEVICES_PER_GPU_FIELD_NUMBER: builtins.int @@ -255,7 +270,9 @@ class GPUOptions(google.protobuf.message.Message): which enumerates jobs*tasks from a ServerDef. """ @property - def virtual_devices(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___GPUOptions.Experimental.VirtualDevices]: + def virtual_devices( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___GPUOptions.Experimental.VirtualDevices]: """The multi virtual device settings. If empty (not set), it will create single virtual device on each visible GPU, according to the settings in "visible_device_list" above. Otherwise, the number of elements in the @@ -321,7 +338,47 @@ class GPUOptions(google.protobuf.message.Message): stream_merge_options: global___GPUOptions.Experimental.StreamMergeOptions | None = ..., ) -> None: ... def HasField(self, field_name: typing.Literal["stream_merge_options", b"stream_merge_options"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["collective_ring_order", b"collective_ring_order", "disallow_retry_on_allocation_failure", b"disallow_retry_on_allocation_failure", "gpu_host_mem_disallow_growth", b"gpu_host_mem_disallow_growth", "gpu_host_mem_limit_in_mb", b"gpu_host_mem_limit_in_mb", "gpu_system_memory_size_in_mb", b"gpu_system_memory_size_in_mb", "internal_fragmentation_fraction", b"internal_fragmentation_fraction", "kernel_tracker_max_bytes", b"kernel_tracker_max_bytes", "kernel_tracker_max_interval", b"kernel_tracker_max_interval", "kernel_tracker_max_pending", b"kernel_tracker_max_pending", "node_id", b"node_id", "num_dev_to_dev_copy_streams", b"num_dev_to_dev_copy_streams", "num_virtual_devices_per_gpu", b"num_virtual_devices_per_gpu", "populate_pjrt_gpu_client_creation_info", b"populate_pjrt_gpu_client_creation_info", "stream_merge_options", b"stream_merge_options", "timestamped_allocator", b"timestamped_allocator", "use_cuda_malloc_async", b"use_cuda_malloc_async", "use_unified_memory", b"use_unified_memory", "virtual_devices", b"virtual_devices"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "collective_ring_order", + b"collective_ring_order", + "disallow_retry_on_allocation_failure", + b"disallow_retry_on_allocation_failure", + "gpu_host_mem_disallow_growth", + b"gpu_host_mem_disallow_growth", + "gpu_host_mem_limit_in_mb", + b"gpu_host_mem_limit_in_mb", + "gpu_system_memory_size_in_mb", + b"gpu_system_memory_size_in_mb", + "internal_fragmentation_fraction", + b"internal_fragmentation_fraction", + "kernel_tracker_max_bytes", + b"kernel_tracker_max_bytes", + "kernel_tracker_max_interval", + b"kernel_tracker_max_interval", + "kernel_tracker_max_pending", + b"kernel_tracker_max_pending", + "node_id", + b"node_id", + "num_dev_to_dev_copy_streams", + b"num_dev_to_dev_copy_streams", + "num_virtual_devices_per_gpu", + b"num_virtual_devices_per_gpu", + "populate_pjrt_gpu_client_creation_info", + b"populate_pjrt_gpu_client_creation_info", + "stream_merge_options", + b"stream_merge_options", + "timestamped_allocator", + b"timestamped_allocator", + "use_cuda_malloc_async", + b"use_cuda_malloc_async", + "use_unified_memory", + b"use_unified_memory", + "virtual_devices", + b"virtual_devices", + ], + ) -> None: ... PER_PROCESS_GPU_MEMORY_FRACTION_FIELD_NUMBER: builtins.int ALLOW_GROWTH_FIELD_NUMBER: builtins.int @@ -433,7 +490,29 @@ class GPUOptions(google.protobuf.message.Message): experimental: global___GPUOptions.Experimental | None = ..., ) -> None: ... def HasField(self, field_name: typing.Literal["experimental", b"experimental"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["allocator_type", b"allocator_type", "allow_growth", b"allow_growth", "deferred_deletion_bytes", b"deferred_deletion_bytes", "experimental", b"experimental", "force_gpu_compatible", b"force_gpu_compatible", "per_process_gpu_memory_fraction", b"per_process_gpu_memory_fraction", "polling_active_delay_usecs", b"polling_active_delay_usecs", "polling_inactive_delay_msecs", b"polling_inactive_delay_msecs", "visible_device_list", b"visible_device_list"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "allocator_type", + b"allocator_type", + "allow_growth", + b"allow_growth", + "deferred_deletion_bytes", + b"deferred_deletion_bytes", + "experimental", + b"experimental", + "force_gpu_compatible", + b"force_gpu_compatible", + "per_process_gpu_memory_fraction", + b"per_process_gpu_memory_fraction", + "polling_active_delay_usecs", + b"polling_active_delay_usecs", + "polling_inactive_delay_msecs", + b"polling_inactive_delay_msecs", + "visible_device_list", + b"visible_device_list", + ], + ) -> None: ... global___GPUOptions = GPUOptions @@ -447,7 +526,9 @@ class OptimizerOptions(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _LevelEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[OptimizerOptions._Level.ValueType], builtins.type): + class _LevelEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[OptimizerOptions._Level.ValueType], builtins.type + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor L1: OptimizerOptions._Level.ValueType # 0 """L1 is the default level. @@ -474,7 +555,9 @@ class OptimizerOptions(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _GlobalJitLevelEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[OptimizerOptions._GlobalJitLevel.ValueType], builtins.type): + class _GlobalJitLevelEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[OptimizerOptions._GlobalJitLevel.ValueType], builtins.type + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor DEFAULT: OptimizerOptions._GlobalJitLevel.ValueType # 0 """Default setting ("off" now, but later expected to be "on")""" @@ -549,7 +632,25 @@ class OptimizerOptions(google.protobuf.message.Message): global_jit_level: global___OptimizerOptions.GlobalJitLevel.ValueType | None = ..., cpu_global_jit: builtins.bool | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["cpu_global_jit", b"cpu_global_jit", "do_common_subexpression_elimination", b"do_common_subexpression_elimination", "do_constant_folding", b"do_constant_folding", "do_function_inlining", b"do_function_inlining", "global_jit_level", b"global_jit_level", "max_folded_constant_in_bytes", b"max_folded_constant_in_bytes", "opt_level", b"opt_level"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "cpu_global_jit", + b"cpu_global_jit", + "do_common_subexpression_elimination", + b"do_common_subexpression_elimination", + "do_constant_folding", + b"do_constant_folding", + "do_function_inlining", + b"do_function_inlining", + "global_jit_level", + b"global_jit_level", + "max_folded_constant_in_bytes", + b"max_folded_constant_in_bytes", + "opt_level", + b"opt_level", + ], + ) -> None: ... global___OptimizerOptions = OptimizerOptions @@ -622,8 +723,32 @@ class GraphOptions(google.protobuf.message.Message): timeline_step: builtins.int | None = ..., rewrite_options: tensorflow.core.protobuf.rewriter_config_pb2.RewriterConfig | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["optimizer_options", b"optimizer_options", "rewrite_options", b"rewrite_options"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["build_cost_model", b"build_cost_model", "build_cost_model_after", b"build_cost_model_after", "enable_bfloat16_sendrecv", b"enable_bfloat16_sendrecv", "enable_recv_scheduling", b"enable_recv_scheduling", "infer_shapes", b"infer_shapes", "optimizer_options", b"optimizer_options", "place_pruned_graph", b"place_pruned_graph", "rewrite_options", b"rewrite_options", "timeline_step", b"timeline_step"]) -> None: ... + def HasField( + self, field_name: typing.Literal["optimizer_options", b"optimizer_options", "rewrite_options", b"rewrite_options"] + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "build_cost_model", + b"build_cost_model", + "build_cost_model_after", + b"build_cost_model_after", + "enable_bfloat16_sendrecv", + b"enable_bfloat16_sendrecv", + "enable_recv_scheduling", + b"enable_recv_scheduling", + "infer_shapes", + b"infer_shapes", + "optimizer_options", + b"optimizer_options", + "place_pruned_graph", + b"place_pruned_graph", + "rewrite_options", + b"rewrite_options", + "timeline_step", + b"timeline_step", + ], + ) -> None: ... global___GraphOptions = GraphOptions @@ -656,12 +781,7 @@ class ThreadPoolOptionProto(google.protobuf.message.Message): value as is specified on this call. - threadpools created this way are never garbage collected. """ - def __init__( - self, - *, - num_threads: builtins.int | None = ..., - global_name: builtins.str | None = ..., - ) -> None: ... + def __init__(self, *, num_threads: builtins.int | None = ..., global_name: builtins.str | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["global_name", b"global_name", "num_threads", b"num_threads"]) -> None: ... global___ThreadPoolOptionProto = ThreadPoolOptionProto @@ -685,12 +805,7 @@ class SessionMetadata(google.protobuf.message.Message): name: builtins.str version: builtins.int """The version is optional. If set, needs to be >= 0.""" - def __init__( - self, - *, - name: builtins.str | None = ..., - version: builtins.int | None = ..., - ) -> None: ... + def __init__(self, *, name: builtins.str | None = ..., version: builtins.int | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["name", b"name", "version", b"version"]) -> None: ... global___SessionMetadata = SessionMetadata @@ -711,12 +826,7 @@ class ConfigProto(google.protobuf.message.Message): VALUE_FIELD_NUMBER: builtins.int key: builtins.str value: builtins.int - def __init__( - self, - *, - key: builtins.str | None = ..., - value: builtins.int | None = ..., - ) -> None: ... + def __init__(self, *, key: builtins.str | None = ..., value: builtins.int | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["key", b"key", "value", b"value"]) -> None: ... @typing.final @@ -732,7 +842,10 @@ class ConfigProto(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _MlirBridgeRolloutEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[ConfigProto.Experimental._MlirBridgeRollout.ValueType], builtins.type): + class _MlirBridgeRolloutEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[ConfigProto.Experimental._MlirBridgeRollout.ValueType], + builtins.type, + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor MLIR_BRIDGE_ROLLOUT_UNSPECIFIED: ConfigProto.Experimental._MlirBridgeRollout.ValueType # 0 """If this field is left unspecified, the MLIR bridge may be selectively @@ -978,8 +1091,71 @@ class ConfigProto(google.protobuf.message.Message): disable_optimize_for_static_graph: builtins.bool | None = ..., disable_eager_executor_streaming_enqueue: builtins.bool | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["coordination_config", b"coordination_config", "session_metadata", b"session_metadata"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["backend_server_port", b"backend_server_port", "collective_deterministic_sequential_execution", b"collective_deterministic_sequential_execution", "collective_group_leader", b"collective_group_leader", "collective_nccl", b"collective_nccl", "coordination_config", b"coordination_config", "disable_eager_executor_streaming_enqueue", b"disable_eager_executor_streaming_enqueue", "disable_functional_ops_lowering", b"disable_functional_ops_lowering", "disable_optimize_for_static_graph", b"disable_optimize_for_static_graph", "disable_output_partition_graphs", b"disable_output_partition_graphs", "disable_thread_spinning", b"disable_thread_spinning", "enable_mlir_bridge", b"enable_mlir_bridge", "enable_mlir_graph_optimization", b"enable_mlir_graph_optimization", "enable_multi_host", b"enable_multi_host", "executor_type", b"executor_type", "mlir_bridge_rollout", b"mlir_bridge_rollout", "optimize_for_static_graph", b"optimize_for_static_graph", "recv_buf_max_chunk", b"recv_buf_max_chunk", "session_metadata", b"session_metadata", "share_cluster_devices_in_session", b"share_cluster_devices_in_session", "share_session_state_in_clusterspec_propagation", b"share_session_state_in_clusterspec_propagation", "stream_merge_threshold", b"stream_merge_threshold", "target_gpu", b"target_gpu", "target_tpu", b"target_tpu", "tfrt_use_ifrt", b"tfrt_use_ifrt", "use_numa_affinity", b"use_numa_affinity", "use_tfrt", b"use_tfrt", "xla_fusion_autotuner_thresh", b"xla_fusion_autotuner_thresh", "xla_prefer_single_graph_cluster", b"xla_prefer_single_graph_cluster"]) -> None: ... + def HasField( + self, + field_name: typing.Literal["coordination_config", b"coordination_config", "session_metadata", b"session_metadata"], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "backend_server_port", + b"backend_server_port", + "collective_deterministic_sequential_execution", + b"collective_deterministic_sequential_execution", + "collective_group_leader", + b"collective_group_leader", + "collective_nccl", + b"collective_nccl", + "coordination_config", + b"coordination_config", + "disable_eager_executor_streaming_enqueue", + b"disable_eager_executor_streaming_enqueue", + "disable_functional_ops_lowering", + b"disable_functional_ops_lowering", + "disable_optimize_for_static_graph", + b"disable_optimize_for_static_graph", + "disable_output_partition_graphs", + b"disable_output_partition_graphs", + "disable_thread_spinning", + b"disable_thread_spinning", + "enable_mlir_bridge", + b"enable_mlir_bridge", + "enable_mlir_graph_optimization", + b"enable_mlir_graph_optimization", + "enable_multi_host", + b"enable_multi_host", + "executor_type", + b"executor_type", + "mlir_bridge_rollout", + b"mlir_bridge_rollout", + "optimize_for_static_graph", + b"optimize_for_static_graph", + "recv_buf_max_chunk", + b"recv_buf_max_chunk", + "session_metadata", + b"session_metadata", + "share_cluster_devices_in_session", + b"share_cluster_devices_in_session", + "share_session_state_in_clusterspec_propagation", + b"share_session_state_in_clusterspec_propagation", + "stream_merge_threshold", + b"stream_merge_threshold", + "target_gpu", + b"target_gpu", + "target_tpu", + b"target_tpu", + "tfrt_use_ifrt", + b"tfrt_use_ifrt", + "use_numa_affinity", + b"use_numa_affinity", + "use_tfrt", + b"use_tfrt", + "xla_fusion_autotuner_thresh", + b"xla_fusion_autotuner_thresh", + "xla_prefer_single_graph_cluster", + b"xla_prefer_single_graph_cluster", + ], + ) -> None: ... DEVICE_COUNT_FIELD_NUMBER: builtins.int INTRA_OP_PARALLELISM_THREADS_FIELD_NUMBER: builtins.int @@ -1078,7 +1254,9 @@ class ConfigProto(google.protobuf.message.Message): """ @property - def session_inter_op_thread_pool(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___ThreadPoolOptionProto]: + def session_inter_op_thread_pool( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___ThreadPoolOptionProto]: """This option is experimental - it may be replaced with a different mechanism in the future. @@ -1151,8 +1329,64 @@ class ConfigProto(google.protobuf.message.Message): share_cluster_devices_in_session: builtins.bool | None = ..., experimental: global___ConfigProto.Experimental | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["cluster_def", b"cluster_def", "experimental", b"experimental", "gpu_options", b"gpu_options", "graph_options", b"graph_options", "pluggable_device_options", b"pluggable_device_options", "rpc_options", b"rpc_options"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["allow_soft_placement", b"allow_soft_placement", "cluster_def", b"cluster_def", "device_count", b"device_count", "device_filters", b"device_filters", "experimental", b"experimental", "gpu_options", b"gpu_options", "graph_options", b"graph_options", "inter_op_parallelism_threads", b"inter_op_parallelism_threads", "intra_op_parallelism_threads", b"intra_op_parallelism_threads", "isolate_session_state", b"isolate_session_state", "log_device_placement", b"log_device_placement", "operation_timeout_in_ms", b"operation_timeout_in_ms", "placement_period", b"placement_period", "pluggable_device_options", b"pluggable_device_options", "rpc_options", b"rpc_options", "session_inter_op_thread_pool", b"session_inter_op_thread_pool", "share_cluster_devices_in_session", b"share_cluster_devices_in_session", "use_per_session_threads", b"use_per_session_threads"]) -> None: ... + def HasField( + self, + field_name: typing.Literal[ + "cluster_def", + b"cluster_def", + "experimental", + b"experimental", + "gpu_options", + b"gpu_options", + "graph_options", + b"graph_options", + "pluggable_device_options", + b"pluggable_device_options", + "rpc_options", + b"rpc_options", + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "allow_soft_placement", + b"allow_soft_placement", + "cluster_def", + b"cluster_def", + "device_count", + b"device_count", + "device_filters", + b"device_filters", + "experimental", + b"experimental", + "gpu_options", + b"gpu_options", + "graph_options", + b"graph_options", + "inter_op_parallelism_threads", + b"inter_op_parallelism_threads", + "intra_op_parallelism_threads", + b"intra_op_parallelism_threads", + "isolate_session_state", + b"isolate_session_state", + "log_device_placement", + b"log_device_placement", + "operation_timeout_in_ms", + b"operation_timeout_in_ms", + "placement_period", + b"placement_period", + "pluggable_device_options", + b"pluggable_device_options", + "rpc_options", + b"rpc_options", + "session_inter_op_thread_pool", + b"session_inter_op_thread_pool", + "share_cluster_devices_in_session", + b"share_cluster_devices_in_session", + "use_per_session_threads", + b"use_per_session_threads", + ], + ) -> None: ... global___ConfigProto = ConfigProto @@ -1166,7 +1400,9 @@ class RunOptions(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _TraceLevelEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[RunOptions._TraceLevel.ValueType], builtins.type): + class _TraceLevelEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[RunOptions._TraceLevel.ValueType], builtins.type + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor NO_TRACE: RunOptions._TraceLevel.ValueType # 0 SOFTWARE_TRACE: RunOptions._TraceLevel.ValueType # 1 @@ -1203,11 +1439,7 @@ class RunOptions(google.protobuf.message.Message): """Priority of the request. The run handler thread pool will schedule ops based on the priority number. The larger number means higher priority. """ - def __init__( - self, - *, - priority: builtins.int | None = ..., - ) -> None: ... + def __init__(self, *, priority: builtins.int | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["priority", b"priority"]) -> None: ... COLLECTIVE_GRAPH_KEY_FIELD_NUMBER: builtins.int @@ -1234,8 +1466,20 @@ class RunOptions(google.protobuf.message.Message): use_run_handler_pool: builtins.bool | None = ..., run_handler_pool_options: global___RunOptions.Experimental.RunHandlerPoolOptions | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["run_handler_pool_options", b"run_handler_pool_options"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["collective_graph_key", b"collective_graph_key", "run_handler_pool_options", b"run_handler_pool_options", "use_run_handler_pool", b"use_run_handler_pool"]) -> None: ... + def HasField( + self, field_name: typing.Literal["run_handler_pool_options", b"run_handler_pool_options"] + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "collective_graph_key", + b"collective_graph_key", + "run_handler_pool_options", + b"run_handler_pool_options", + "use_run_handler_pool", + b"use_run_handler_pool", + ], + ) -> None: ... TRACE_LEVEL_FIELD_NUMBER: builtins.int TIMEOUT_IN_MS_FIELD_NUMBER: builtins.int @@ -1283,8 +1527,28 @@ class RunOptions(google.protobuf.message.Message): report_tensor_allocations_upon_oom: builtins.bool | None = ..., experimental: global___RunOptions.Experimental | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["debug_options", b"debug_options", "experimental", b"experimental"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["debug_options", b"debug_options", "experimental", b"experimental", "inter_op_thread_pool", b"inter_op_thread_pool", "output_partition_graphs", b"output_partition_graphs", "report_tensor_allocations_upon_oom", b"report_tensor_allocations_upon_oom", "timeout_in_ms", b"timeout_in_ms", "trace_level", b"trace_level"]) -> None: ... + def HasField( + self, field_name: typing.Literal["debug_options", b"debug_options", "experimental", b"experimental"] + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "debug_options", + b"debug_options", + "experimental", + b"experimental", + "inter_op_thread_pool", + b"inter_op_thread_pool", + "output_partition_graphs", + b"output_partition_graphs", + "report_tensor_allocations_upon_oom", + b"report_tensor_allocations_upon_oom", + "timeout_in_ms", + b"timeout_in_ms", + "trace_level", + b"trace_level", + ], + ) -> None: ... global___RunOptions = RunOptions @@ -1302,7 +1566,9 @@ class RunMetadata(google.protobuf.message.Message): PRE_OPTIMIZATION_GRAPH_FIELD_NUMBER: builtins.int POST_OPTIMIZATION_GRAPH_FIELD_NUMBER: builtins.int @property - def partition_graphs(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[tensorflow.core.framework.graph_pb2.GraphDef]: + def partition_graphs( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[tensorflow.core.framework.graph_pb2.GraphDef]: """TODO(nareshmodi): Include some sort of function/cache-key identifier?""" @property @@ -1316,8 +1582,23 @@ class RunMetadata(google.protobuf.message.Message): pre_optimization_graph: tensorflow.core.framework.graph_pb2.GraphDef | None = ..., post_optimization_graph: tensorflow.core.framework.graph_pb2.GraphDef | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["post_optimization_graph", b"post_optimization_graph", "pre_optimization_graph", b"pre_optimization_graph"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["partition_graphs", b"partition_graphs", "post_optimization_graph", b"post_optimization_graph", "pre_optimization_graph", b"pre_optimization_graph"]) -> None: ... + def HasField( + self, + field_name: typing.Literal[ + "post_optimization_graph", b"post_optimization_graph", "pre_optimization_graph", b"pre_optimization_graph" + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "partition_graphs", + b"partition_graphs", + "post_optimization_graph", + b"post_optimization_graph", + "pre_optimization_graph", + b"pre_optimization_graph", + ], + ) -> None: ... STEP_STATS_FIELD_NUMBER: builtins.int COST_GRAPH_FIELD_NUMBER: builtins.int @@ -1336,11 +1617,15 @@ class RunMetadata(google.protobuf.message.Message): """The cost graph for the computation defined by the run call.""" @property - def partition_graphs(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[tensorflow.core.framework.graph_pb2.GraphDef]: + def partition_graphs( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[tensorflow.core.framework.graph_pb2.GraphDef]: """Graphs of the partitions executed by executors.""" @property - def function_graphs(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___RunMetadata.FunctionGraphs]: + def function_graphs( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___RunMetadata.FunctionGraphs]: """This is only populated for graphs that are run as functions in TensorFlow V2. There will be an entry below for each function that is traced. The main use cases of the post_optimization_graph and the partition_graphs @@ -1366,8 +1651,27 @@ class RunMetadata(google.protobuf.message.Message): function_graphs: collections.abc.Iterable[global___RunMetadata.FunctionGraphs] | None = ..., session_metadata: global___SessionMetadata | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["cost_graph", b"cost_graph", "session_metadata", b"session_metadata", "step_stats", b"step_stats"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["cost_graph", b"cost_graph", "function_graphs", b"function_graphs", "partition_graphs", b"partition_graphs", "session_metadata", b"session_metadata", "step_stats", b"step_stats"]) -> None: ... + def HasField( + self, + field_name: typing.Literal[ + "cost_graph", b"cost_graph", "session_metadata", b"session_metadata", "step_stats", b"step_stats" + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "cost_graph", + b"cost_graph", + "function_graphs", + b"function_graphs", + "partition_graphs", + b"partition_graphs", + "session_metadata", + b"session_metadata", + "step_stats", + b"step_stats", + ], + ) -> None: ... global___RunMetadata = RunMetadata @@ -1387,12 +1691,7 @@ class TensorConnection(google.protobuf.message.Message): """A tensor name. The value of this tensor will be bound to the value of the tensor named in `from_tensor`. """ - def __init__( - self, - *, - from_tensor: builtins.str | None = ..., - to_tensor: builtins.str | None = ..., - ) -> None: ... + def __init__(self, *, from_tensor: builtins.str | None = ..., to_tensor: builtins.str | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["from_tensor", b"from_tensor", "to_tensor", b"to_tensor"]) -> None: ... global___TensorConnection = TensorConnection @@ -1415,12 +1714,7 @@ class CallableOptions(google.protobuf.message.Message): VALUE_FIELD_NUMBER: builtins.int key: builtins.str value: builtins.str - def __init__( - self, - *, - key: builtins.str | None = ..., - value: builtins.str | None = ..., - ) -> None: ... + def __init__(self, *, key: builtins.str | None = ..., value: builtins.str | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["key", b"key", "value", b"value"]) -> None: ... @typing.final @@ -1431,12 +1725,7 @@ class CallableOptions(google.protobuf.message.Message): VALUE_FIELD_NUMBER: builtins.int key: builtins.str value: builtins.str - def __init__( - self, - *, - key: builtins.str | None = ..., - value: builtins.str | None = ..., - ) -> None: ... + def __init__(self, *, key: builtins.str | None = ..., value: builtins.str | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["key", b"key", "value", b"value"]) -> None: ... FEED_FIELD_NUMBER: builtins.int @@ -1553,6 +1842,26 @@ class CallableOptions(google.protobuf.message.Message): fetch_skip_sync: builtins.bool | None = ..., ) -> None: ... def HasField(self, field_name: typing.Literal["run_options", b"run_options"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["feed", b"feed", "feed_devices", b"feed_devices", "fetch", b"fetch", "fetch_devices", b"fetch_devices", "fetch_skip_sync", b"fetch_skip_sync", "run_options", b"run_options", "target", b"target", "tensor_connection", b"tensor_connection"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "feed", + b"feed", + "feed_devices", + b"feed_devices", + "fetch", + b"fetch", + "fetch_devices", + b"fetch_devices", + "fetch_skip_sync", + b"fetch_skip_sync", + "run_options", + b"run_options", + "target", + b"target", + "tensor_connection", + b"tensor_connection", + ], + ) -> None: ... global___CallableOptions = CallableOptions diff --git a/stubs/tensorflow/tensorflow/core/protobuf/control_flow_pb2.pyi b/stubs/tensorflow/tensorflow/core/protobuf/control_flow_pb2.pyi index 713544819ed5..2c044a8c2220 100644 --- a/stubs/tensorflow/tensorflow/core/protobuf/control_flow_pb2.pyi +++ b/stubs/tensorflow/tensorflow/core/protobuf/control_flow_pb2.pyi @@ -30,12 +30,7 @@ class ValuesDef(google.protobuf.message.Message): VALUE_FIELD_NUMBER: builtins.int key: builtins.str value: builtins.str - def __init__( - self, - *, - key: builtins.str | None = ..., - value: builtins.str | None = ..., - ) -> None: ... + def __init__(self, *, key: builtins.str | None = ..., value: builtins.str | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["key", b"key", "value", b"value"]) -> None: ... VALUES_FIELD_NUMBER: builtins.int @@ -73,13 +68,14 @@ class ControlFlowContextDef(google.protobuf.message.Message): @property def while_ctxt(self) -> global___WhileContextDef: ... def __init__( - self, - *, - cond_ctxt: global___CondContextDef | None = ..., - while_ctxt: global___WhileContextDef | None = ..., + self, *, cond_ctxt: global___CondContextDef | None = ..., while_ctxt: global___WhileContextDef | None = ... + ) -> None: ... + def HasField( + self, field_name: typing.Literal["cond_ctxt", b"cond_ctxt", "ctxt", b"ctxt", "while_ctxt", b"while_ctxt"] + ) -> builtins.bool: ... + def ClearField( + self, field_name: typing.Literal["cond_ctxt", b"cond_ctxt", "ctxt", b"ctxt", "while_ctxt", b"while_ctxt"] ) -> None: ... - def HasField(self, field_name: typing.Literal["cond_ctxt", b"cond_ctxt", "ctxt", b"ctxt", "while_ctxt", b"while_ctxt"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["cond_ctxt", b"cond_ctxt", "ctxt", b"ctxt", "while_ctxt", b"while_ctxt"]) -> None: ... def WhichOneof(self, oneof_group: typing.Literal["ctxt", b"ctxt"]) -> typing.Literal["cond_ctxt", "while_ctxt"] | None: ... global___ControlFlowContextDef = ControlFlowContextDef @@ -109,7 +105,9 @@ class CondContextDef(google.protobuf.message.Message): """Values and external values in control flow context.""" @property - def nested_contexts(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___ControlFlowContextDef]: + def nested_contexts( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___ControlFlowContextDef]: """Contexts contained inside this context (e.g. nested conds).""" def __init__( @@ -123,7 +121,23 @@ class CondContextDef(google.protobuf.message.Message): nested_contexts: collections.abc.Iterable[global___ControlFlowContextDef] | None = ..., ) -> None: ... def HasField(self, field_name: typing.Literal["values_def", b"values_def"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["branch", b"branch", "context_name", b"context_name", "nested_contexts", b"nested_contexts", "pivot_name", b"pivot_name", "pred_name", b"pred_name", "values_def", b"values_def"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "branch", + b"branch", + "context_name", + b"context_name", + "nested_contexts", + b"nested_contexts", + "pivot_name", + b"pivot_name", + "pred_name", + b"pred_name", + "values_def", + b"values_def", + ], + ) -> None: ... global___CondContextDef = CondContextDef @@ -174,7 +188,9 @@ class WhileContextDef(google.protobuf.message.Message): """Values and external values in control flow context.""" @property - def nested_contexts(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___ControlFlowContextDef]: + def nested_contexts( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___ControlFlowContextDef]: """Contexts contained inside this context (e.g. nested whiles).""" def __init__( @@ -194,6 +210,34 @@ class WhileContextDef(google.protobuf.message.Message): nested_contexts: collections.abc.Iterable[global___ControlFlowContextDef] | None = ..., ) -> None: ... def HasField(self, field_name: typing.Literal["values_def", b"values_def"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["back_prop", b"back_prop", "context_name", b"context_name", "loop_enter_names", b"loop_enter_names", "loop_exit_names", b"loop_exit_names", "maximum_iterations_name", b"maximum_iterations_name", "nested_contexts", b"nested_contexts", "parallel_iterations", b"parallel_iterations", "pivot_for_body_name", b"pivot_for_body_name", "pivot_for_pred_name", b"pivot_for_pred_name", "pivot_name", b"pivot_name", "swap_memory", b"swap_memory", "values_def", b"values_def"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "back_prop", + b"back_prop", + "context_name", + b"context_name", + "loop_enter_names", + b"loop_enter_names", + "loop_exit_names", + b"loop_exit_names", + "maximum_iterations_name", + b"maximum_iterations_name", + "nested_contexts", + b"nested_contexts", + "parallel_iterations", + b"parallel_iterations", + "pivot_for_body_name", + b"pivot_for_body_name", + "pivot_for_pred_name", + b"pivot_for_pred_name", + "pivot_name", + b"pivot_name", + "swap_memory", + b"swap_memory", + "values_def", + b"values_def", + ], + ) -> None: ... global___WhileContextDef = WhileContextDef diff --git a/stubs/tensorflow/tensorflow/core/protobuf/core_platform_payloads_pb2.pyi b/stubs/tensorflow/tensorflow/core/protobuf/core_platform_payloads_pb2.pyi index 44eea2b74e65..39763f66f4e6 100644 --- a/stubs/tensorflow/tensorflow/core/protobuf/core_platform_payloads_pb2.pyi +++ b/stubs/tensorflow/tensorflow/core/protobuf/core_platform_payloads_pb2.pyi @@ -31,7 +31,9 @@ class ErrorSourceProto(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _ErrorSourceEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[ErrorSourceProto._ErrorSource.ValueType], builtins.type): + class _ErrorSourceEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[ErrorSourceProto._ErrorSource.ValueType], builtins.type + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor UNKNOWN: ErrorSourceProto._ErrorSource.ValueType # 0 TPU_COMPILE_OP: ErrorSourceProto._ErrorSource.ValueType # 1 @@ -58,11 +60,7 @@ class ErrorSourceProto(google.protobuf.message.Message): ERROR_SOURCE_FIELD_NUMBER: builtins.int error_source: global___ErrorSourceProto.ErrorSource.ValueType - def __init__( - self, - *, - error_source: global___ErrorSourceProto.ErrorSource.ValueType | None = ..., - ) -> None: ... + def __init__(self, *, error_source: global___ErrorSourceProto.ErrorSource.ValueType | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["error_source", b"error_source"]) -> None: ... global___ErrorSourceProto = ErrorSourceProto diff --git a/stubs/tensorflow/tensorflow/core/protobuf/data_service_pb2.pyi b/stubs/tensorflow/tensorflow/core/protobuf/data_service_pb2.pyi index 3c149e5ee5ac..cdafebecf7eb 100644 --- a/stubs/tensorflow/tensorflow/core/protobuf/data_service_pb2.pyi +++ b/stubs/tensorflow/tensorflow/core/protobuf/data_service_pb2.pyi @@ -22,7 +22,9 @@ class _DeploymentMode: ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType -class _DeploymentModeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_DeploymentMode.ValueType], builtins.type): +class _DeploymentModeEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_DeploymentMode.ValueType], builtins.type +): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor DEPLOYMENT_MODE_UNSPECIFIED: _DeploymentMode.ValueType # 0 DEPLOYMENT_MODE_COLOCATED: _DeploymentMode.ValueType # 1 @@ -58,7 +60,9 @@ class ProcessingModeDef(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _ShardingPolicyEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[ProcessingModeDef._ShardingPolicy.ValueType], builtins.type): + class _ShardingPolicyEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[ProcessingModeDef._ShardingPolicy.ValueType], builtins.type + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor OFF: ProcessingModeDef._ShardingPolicy.ValueType # 0 """No sharding will be performed. Each worker produces the entire dataset @@ -141,11 +145,7 @@ class ProcessingModeDef(google.protobuf.message.Message): SHARDING_POLICY_FIELD_NUMBER: builtins.int sharding_policy: global___ProcessingModeDef.ShardingPolicy.ValueType - def __init__( - self, - *, - sharding_policy: global___ProcessingModeDef.ShardingPolicy.ValueType | None = ..., - ) -> None: ... + def __init__(self, *, sharding_policy: global___ProcessingModeDef.ShardingPolicy.ValueType | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["sharding_policy", b"sharding_policy"]) -> None: ... global___ProcessingModeDef = ProcessingModeDef @@ -162,7 +162,9 @@ class DataServiceMetadata(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _CompressionEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[DataServiceMetadata._Compression.ValueType], builtins.type): + class _CompressionEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[DataServiceMetadata._Compression.ValueType], builtins.type + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor COMPRESSION_UNSPECIFIED: DataServiceMetadata._Compression.ValueType # 0 COMPRESSION_OFF: DataServiceMetadata._Compression.ValueType # 1 @@ -192,9 +194,25 @@ class DataServiceMetadata(google.protobuf.message.Message): compression: global___DataServiceMetadata.Compression.ValueType | None = ..., cardinality: builtins.int | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["element_spec", b"element_spec", "optional_element_spec", b"optional_element_spec"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["cardinality", b"cardinality", "compression", b"compression", "element_spec", b"element_spec", "optional_element_spec", b"optional_element_spec"]) -> None: ... - def WhichOneof(self, oneof_group: typing.Literal["optional_element_spec", b"optional_element_spec"]) -> typing.Literal["element_spec"] | None: ... + def HasField( + self, field_name: typing.Literal["element_spec", b"element_spec", "optional_element_spec", b"optional_element_spec"] + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "cardinality", + b"cardinality", + "compression", + b"compression", + "element_spec", + b"element_spec", + "optional_element_spec", + b"optional_element_spec", + ], + ) -> None: ... + def WhichOneof( + self, oneof_group: typing.Literal["optional_element_spec", b"optional_element_spec"] + ) -> typing.Literal["element_spec"] | None: ... global___DataServiceMetadata = DataServiceMetadata @@ -204,11 +222,7 @@ class CrossTrainerCacheOptions(google.protobuf.message.Message): TRAINER_ID_FIELD_NUMBER: builtins.int trainer_id: builtins.str - def __init__( - self, - *, - trainer_id: builtins.str | None = ..., - ) -> None: ... + def __init__(self, *, trainer_id: builtins.str | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["trainer_id", b"trainer_id"]) -> None: ... global___CrossTrainerCacheOptions = CrossTrainerCacheOptions @@ -223,11 +237,7 @@ class DataServiceConfig(google.protobuf.message.Message): DEPLOYMENT_MODE_FIELD_NUMBER: builtins.int deployment_mode: global___DeploymentMode.ValueType - def __init__( - self, - *, - deployment_mode: global___DeploymentMode.ValueType | None = ..., - ) -> None: ... + def __init__(self, *, deployment_mode: global___DeploymentMode.ValueType | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["deployment_mode", b"deployment_mode"]) -> None: ... global___DataServiceConfig = DataServiceConfig diff --git a/stubs/tensorflow/tensorflow/core/protobuf/debug_event_pb2.pyi b/stubs/tensorflow/tensorflow/core/protobuf/debug_event_pb2.pyi index 8ee6e76d65da..2906b660d309 100644 --- a/stubs/tensorflow/tensorflow/core/protobuf/debug_event_pb2.pyi +++ b/stubs/tensorflow/tensorflow/core/protobuf/debug_event_pb2.pyi @@ -26,7 +26,9 @@ class _TensorDebugMode: ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType -class _TensorDebugModeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_TensorDebugMode.ValueType], builtins.type): +class _TensorDebugModeEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_TensorDebugMode.ValueType], builtins.type +): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor UNSPECIFIED: _TensorDebugMode.ValueType # 0 NO_TENSOR: _TensorDebugMode.ValueType # 1 @@ -209,9 +211,76 @@ class DebugEvent(google.protobuf.message.Message): graph_id: builtins.str | None = ..., debugged_device: global___DebuggedDevice | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["debug_metadata", b"debug_metadata", "debugged_device", b"debugged_device", "debugged_graph", b"debugged_graph", "execution", b"execution", "graph_execution_trace", b"graph_execution_trace", "graph_id", b"graph_id", "graph_op_creation", b"graph_op_creation", "source_file", b"source_file", "stack_frame_with_id", b"stack_frame_with_id", "what", b"what"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["debug_metadata", b"debug_metadata", "debugged_device", b"debugged_device", "debugged_graph", b"debugged_graph", "execution", b"execution", "graph_execution_trace", b"graph_execution_trace", "graph_id", b"graph_id", "graph_op_creation", b"graph_op_creation", "source_file", b"source_file", "stack_frame_with_id", b"stack_frame_with_id", "step", b"step", "wall_time", b"wall_time", "what", b"what"]) -> None: ... - def WhichOneof(self, oneof_group: typing.Literal["what", b"what"]) -> typing.Literal["debug_metadata", "source_file", "stack_frame_with_id", "graph_op_creation", "debugged_graph", "execution", "graph_execution_trace", "graph_id", "debugged_device"] | None: ... + def HasField( + self, + field_name: typing.Literal[ + "debug_metadata", + b"debug_metadata", + "debugged_device", + b"debugged_device", + "debugged_graph", + b"debugged_graph", + "execution", + b"execution", + "graph_execution_trace", + b"graph_execution_trace", + "graph_id", + b"graph_id", + "graph_op_creation", + b"graph_op_creation", + "source_file", + b"source_file", + "stack_frame_with_id", + b"stack_frame_with_id", + "what", + b"what", + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "debug_metadata", + b"debug_metadata", + "debugged_device", + b"debugged_device", + "debugged_graph", + b"debugged_graph", + "execution", + b"execution", + "graph_execution_trace", + b"graph_execution_trace", + "graph_id", + b"graph_id", + "graph_op_creation", + b"graph_op_creation", + "source_file", + b"source_file", + "stack_frame_with_id", + b"stack_frame_with_id", + "step", + b"step", + "wall_time", + b"wall_time", + "what", + b"what", + ], + ) -> None: ... + def WhichOneof( + self, oneof_group: typing.Literal["what", b"what"] + ) -> ( + typing.Literal[ + "debug_metadata", + "source_file", + "stack_frame_with_id", + "graph_op_creation", + "debugged_graph", + "execution", + "graph_execution_trace", + "graph_id", + "debugged_device", + ] + | None + ): ... global___DebugEvent = DebugEvent @@ -243,7 +312,12 @@ class DebugMetadata(google.protobuf.message.Message): file_version: builtins.str | None = ..., tfdbg_run_id: builtins.str | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["file_version", b"file_version", "tensorflow_version", b"tensorflow_version", "tfdbg_run_id", b"tfdbg_run_id"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "file_version", b"file_version", "tensorflow_version", b"tensorflow_version", "tfdbg_run_id", b"tfdbg_run_id" + ], + ) -> None: ... global___DebugMetadata = DebugMetadata @@ -273,7 +347,9 @@ class SourceFile(google.protobuf.message.Message): host_name: builtins.str | None = ..., lines: collections.abc.Iterable[builtins.str] | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["file_path", b"file_path", "host_name", b"host_name", "lines", b"lines"]) -> None: ... + def ClearField( + self, field_name: typing.Literal["file_path", b"file_path", "host_name", b"host_name", "lines", b"lines"] + ) -> None: ... global___SourceFile = SourceFile @@ -326,12 +402,11 @@ class CodeLocation(google.protobuf.message.Message): """ def __init__( - self, - *, - host_name: builtins.str | None = ..., - stack_frame_ids: collections.abc.Iterable[builtins.str] | None = ..., + self, *, host_name: builtins.str | None = ..., stack_frame_ids: collections.abc.Iterable[builtins.str] | None = ... + ) -> None: ... + def ClearField( + self, field_name: typing.Literal["host_name", b"host_name", "stack_frame_ids", b"stack_frame_ids"] ) -> None: ... - def ClearField(self, field_name: typing.Literal["host_name", b"host_name", "stack_frame_ids", b"stack_frame_ids"]) -> None: ... global___CodeLocation = CodeLocation @@ -390,7 +465,29 @@ class GraphOpCreation(google.protobuf.message.Message): output_tensor_ids: collections.abc.Iterable[builtins.int] | None = ..., ) -> None: ... def HasField(self, field_name: typing.Literal["code_location", b"code_location"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["code_location", b"code_location", "device_name", b"device_name", "graph_id", b"graph_id", "graph_name", b"graph_name", "input_names", b"input_names", "num_outputs", b"num_outputs", "op_name", b"op_name", "op_type", b"op_type", "output_tensor_ids", b"output_tensor_ids"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "code_location", + b"code_location", + "device_name", + b"device_name", + "graph_id", + b"graph_id", + "graph_name", + b"graph_name", + "input_names", + b"input_names", + "num_outputs", + b"num_outputs", + "op_name", + b"op_name", + "op_type", + b"op_type", + "output_tensor_ids", + b"output_tensor_ids", + ], + ) -> None: ... global___GraphOpCreation = GraphOpCreation @@ -436,7 +533,23 @@ class DebuggedGraph(google.protobuf.message.Message): instrumented_graph_def: builtins.bytes | None = ..., outer_context_id: builtins.str | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["graph_id", b"graph_id", "graph_name", b"graph_name", "instrumented_graph_def", b"instrumented_graph_def", "instrumented_ops", b"instrumented_ops", "original_graph_def", b"original_graph_def", "outer_context_id", b"outer_context_id"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "graph_id", + b"graph_id", + "graph_name", + b"graph_name", + "instrumented_graph_def", + b"instrumented_graph_def", + "instrumented_ops", + b"instrumented_ops", + "original_graph_def", + b"original_graph_def", + "outer_context_id", + b"outer_context_id", + ], + ) -> None: ... global___DebuggedGraph = DebuggedGraph @@ -456,12 +569,7 @@ class DebuggedDevice(google.protobuf.message.Message): multi-host settings. TODO(cais): Test the uniqueness guarantee in multi-host settings. """ - def __init__( - self, - *, - device_name: builtins.str | None = ..., - device_id: builtins.int | None = ..., - ) -> None: ... + def __init__(self, *, device_name: builtins.str | None = ..., device_id: builtins.int | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["device_id", b"device_id", "device_name", b"device_name"]) -> None: ... global___DebuggedDevice = DebuggedDevice @@ -507,7 +615,9 @@ class Execution(google.protobuf.message.Message): """ @property - def tensor_protos(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[tensorflow.core.framework.tensor_pb2.TensorProto]: + def tensor_protos( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[tensorflow.core.framework.tensor_pb2.TensorProto]: """Output Tensor values in the type described by `tensor_value_type`. The length of this should match `num_outputs`. """ @@ -537,7 +647,29 @@ class Execution(google.protobuf.message.Message): output_tensor_device_ids: collections.abc.Iterable[builtins.int] | None = ..., ) -> None: ... def HasField(self, field_name: typing.Literal["code_location", b"code_location"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["code_location", b"code_location", "graph_id", b"graph_id", "input_tensor_ids", b"input_tensor_ids", "num_outputs", b"num_outputs", "op_type", b"op_type", "output_tensor_device_ids", b"output_tensor_device_ids", "output_tensor_ids", b"output_tensor_ids", "tensor_debug_mode", b"tensor_debug_mode", "tensor_protos", b"tensor_protos"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "code_location", + b"code_location", + "graph_id", + b"graph_id", + "input_tensor_ids", + b"input_tensor_ids", + "num_outputs", + b"num_outputs", + "op_type", + b"op_type", + "output_tensor_device_ids", + b"output_tensor_device_ids", + "output_tensor_ids", + b"output_tensor_ids", + "tensor_debug_mode", + b"tensor_debug_mode", + "tensor_protos", + b"tensor_protos", + ], + ) -> None: ... global___Execution = Execution @@ -593,6 +725,22 @@ class GraphExecutionTrace(google.protobuf.message.Message): device_name: builtins.str | None = ..., ) -> None: ... def HasField(self, field_name: typing.Literal["tensor_proto", b"tensor_proto"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["device_name", b"device_name", "op_name", b"op_name", "output_slot", b"output_slot", "tensor_debug_mode", b"tensor_debug_mode", "tensor_proto", b"tensor_proto", "tfdbg_context_id", b"tfdbg_context_id"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "device_name", + b"device_name", + "op_name", + b"op_name", + "output_slot", + b"output_slot", + "tensor_debug_mode", + b"tensor_debug_mode", + "tensor_proto", + b"tensor_proto", + "tfdbg_context_id", + b"tfdbg_context_id", + ], + ) -> None: ... global___GraphExecutionTrace = GraphExecutionTrace diff --git a/stubs/tensorflow/tensorflow/core/protobuf/debug_pb2.pyi b/stubs/tensorflow/tensorflow/core/protobuf/debug_pb2.pyi index 5ee6b532d850..28a2c99bc415 100644 --- a/stubs/tensorflow/tensorflow/core/protobuf/debug_pb2.pyi +++ b/stubs/tensorflow/tensorflow/core/protobuf/debug_pb2.pyi @@ -80,7 +80,21 @@ class DebugTensorWatch(google.protobuf.message.Message): debug_urls: collections.abc.Iterable[builtins.str] | None = ..., tolerate_debug_op_creation_failures: builtins.bool | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["debug_ops", b"debug_ops", "debug_urls", b"debug_urls", "node_name", b"node_name", "output_slot", b"output_slot", "tolerate_debug_op_creation_failures", b"tolerate_debug_op_creation_failures"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "debug_ops", + b"debug_ops", + "debug_urls", + b"debug_urls", + "node_name", + b"node_name", + "output_slot", + b"output_slot", + "tolerate_debug_op_creation_failures", + b"tolerate_debug_op_creation_failures", + ], + ) -> None: ... global___DebugTensorWatch = DebugTensorWatch @@ -105,7 +119,9 @@ class DebugOptions(google.protobuf.message.Message): are cleaned up from the disk after each Session.run. """ @property - def debug_tensor_watch_opts(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___DebugTensorWatch]: + def debug_tensor_watch_opts( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___DebugTensorWatch]: """Debugging options""" def __init__( @@ -115,7 +131,17 @@ class DebugOptions(google.protobuf.message.Message): global_step: builtins.int | None = ..., reset_disk_byte_usage: builtins.bool | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["debug_tensor_watch_opts", b"debug_tensor_watch_opts", "global_step", b"global_step", "reset_disk_byte_usage", b"reset_disk_byte_usage"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "debug_tensor_watch_opts", + b"debug_tensor_watch_opts", + "global_step", + b"global_step", + "reset_disk_byte_usage", + b"reset_disk_byte_usage", + ], + ) -> None: ... global___DebugOptions = DebugOptions @@ -149,7 +175,12 @@ class DebuggedSourceFile(google.protobuf.message.Message): bytes: builtins.int | None = ..., lines: collections.abc.Iterable[builtins.str] | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["bytes", b"bytes", "file_path", b"file_path", "host", b"host", "last_modified", b"last_modified", "lines", b"lines"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "bytes", b"bytes", "file_path", b"file_path", "host", b"host", "last_modified", b"last_modified", "lines", b"lines" + ], + ) -> None: ... global___DebuggedSourceFile = DebuggedSourceFile @@ -162,11 +193,7 @@ class DebuggedSourceFiles(google.protobuf.message.Message): def source_files(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___DebuggedSourceFile]: """A collection of source code files.""" - def __init__( - self, - *, - source_files: collections.abc.Iterable[global___DebuggedSourceFile] | None = ..., - ) -> None: ... + def __init__(self, *, source_files: collections.abc.Iterable[global___DebuggedSourceFile] | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["source_files", b"source_files"]) -> None: ... global___DebuggedSourceFiles = DebuggedSourceFiles diff --git a/stubs/tensorflow/tensorflow/core/protobuf/device_filters_pb2.pyi b/stubs/tensorflow/tensorflow/core/protobuf/device_filters_pb2.pyi index f4e0c8efe920..a9fd3a142db7 100644 --- a/stubs/tensorflow/tensorflow/core/protobuf/device_filters_pb2.pyi +++ b/stubs/tensorflow/tensorflow/core/protobuf/device_filters_pb2.pyi @@ -67,11 +67,7 @@ class TaskDeviceFilters(google.protobuf.message.Message): DEVICE_FILTERS_FIELD_NUMBER: builtins.int @property def device_filters(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]: ... - def __init__( - self, - *, - device_filters: collections.abc.Iterable[builtins.str] | None = ..., - ) -> None: ... + def __init__(self, *, device_filters: collections.abc.Iterable[builtins.str] | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["device_filters", b"device_filters"]) -> None: ... global___TaskDeviceFilters = TaskDeviceFilters @@ -91,12 +87,7 @@ class JobDeviceFilters(google.protobuf.message.Message): key: builtins.int @property def value(self) -> global___TaskDeviceFilters: ... - def __init__( - self, - *, - key: builtins.int | None = ..., - value: global___TaskDeviceFilters | None = ..., - ) -> None: ... + def __init__(self, *, key: builtins.int | None = ..., value: global___TaskDeviceFilters | None = ...) -> None: ... def HasField(self, field_name: typing.Literal["value", b"value"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["key", b"key", "value", b"value"]) -> None: ... @@ -127,11 +118,7 @@ class ClusterDeviceFilters(google.protobuf.message.Message): JOBS_FIELD_NUMBER: builtins.int @property def jobs(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___JobDeviceFilters]: ... - def __init__( - self, - *, - jobs: collections.abc.Iterable[global___JobDeviceFilters] | None = ..., - ) -> None: ... + def __init__(self, *, jobs: collections.abc.Iterable[global___JobDeviceFilters] | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["jobs", b"jobs"]) -> None: ... global___ClusterDeviceFilters = ClusterDeviceFilters diff --git a/stubs/tensorflow/tensorflow/core/protobuf/device_properties_pb2.pyi b/stubs/tensorflow/tensorflow/core/protobuf/device_properties_pb2.pyi index af4bdd7a1f88..3aadb10563ec 100644 --- a/stubs/tensorflow/tensorflow/core/protobuf/device_properties_pb2.pyi +++ b/stubs/tensorflow/tensorflow/core/protobuf/device_properties_pb2.pyi @@ -39,12 +39,7 @@ class DeviceProperties(google.protobuf.message.Message): VALUE_FIELD_NUMBER: builtins.int key: builtins.str value: builtins.str - def __init__( - self, - *, - key: builtins.str | None = ..., - value: builtins.str | None = ..., - ) -> None: ... + def __init__(self, *, key: builtins.str | None = ..., value: builtins.str | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["key", b"key", "value", b"value"]) -> None: ... TYPE_FIELD_NUMBER: builtins.int @@ -109,7 +104,37 @@ class DeviceProperties(google.protobuf.message.Message): memory_size: builtins.int | None = ..., bandwidth: builtins.int | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["bandwidth", b"bandwidth", "environment", b"environment", "frequency", b"frequency", "l1_cache_size", b"l1_cache_size", "l2_cache_size", b"l2_cache_size", "l3_cache_size", b"l3_cache_size", "memory_size", b"memory_size", "model", b"model", "num_cores", b"num_cores", "num_registers", b"num_registers", "shared_memory_size_per_multiprocessor", b"shared_memory_size_per_multiprocessor", "type", b"type", "vendor", b"vendor"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "bandwidth", + b"bandwidth", + "environment", + b"environment", + "frequency", + b"frequency", + "l1_cache_size", + b"l1_cache_size", + "l2_cache_size", + b"l2_cache_size", + "l3_cache_size", + b"l3_cache_size", + "memory_size", + b"memory_size", + "model", + b"model", + "num_cores", + b"num_cores", + "num_registers", + b"num_registers", + "shared_memory_size_per_multiprocessor", + b"shared_memory_size_per_multiprocessor", + "type", + b"type", + "vendor", + b"vendor", + ], + ) -> None: ... global___DeviceProperties = DeviceProperties @@ -122,12 +147,7 @@ class NamedDevice(google.protobuf.message.Message): name: builtins.str @property def properties(self) -> global___DeviceProperties: ... - def __init__( - self, - *, - name: builtins.str | None = ..., - properties: global___DeviceProperties | None = ..., - ) -> None: ... + def __init__(self, *, name: builtins.str | None = ..., properties: global___DeviceProperties | None = ...) -> None: ... def HasField(self, field_name: typing.Literal["properties", b"properties"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["name", b"name", "properties", b"properties"]) -> None: ... diff --git a/stubs/tensorflow/tensorflow/core/protobuf/fingerprint_pb2.pyi b/stubs/tensorflow/tensorflow/core/protobuf/fingerprint_pb2.pyi index 83956b31c972..e5c00ec8506d 100644 --- a/stubs/tensorflow/tensorflow/core/protobuf/fingerprint_pb2.pyi +++ b/stubs/tensorflow/tensorflow/core/protobuf/fingerprint_pb2.pyi @@ -55,6 +55,22 @@ class FingerprintDef(google.protobuf.message.Message): version: tensorflow.core.framework.versions_pb2.VersionDef | None = ..., ) -> None: ... def HasField(self, field_name: typing.Literal["version", b"version"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["checkpoint_hash", b"checkpoint_hash", "graph_def_program_hash", b"graph_def_program_hash", "saved_model_checksum", b"saved_model_checksum", "saved_object_graph_hash", b"saved_object_graph_hash", "signature_def_hash", b"signature_def_hash", "version", b"version"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "checkpoint_hash", + b"checkpoint_hash", + "graph_def_program_hash", + b"graph_def_program_hash", + "saved_model_checksum", + b"saved_model_checksum", + "saved_object_graph_hash", + b"saved_object_graph_hash", + "signature_def_hash", + b"signature_def_hash", + "version", + b"version", + ], + ) -> None: ... global___FingerprintDef = FingerprintDef diff --git a/stubs/tensorflow/tensorflow/core/protobuf/meta_graph_pb2.pyi b/stubs/tensorflow/tensorflow/core/protobuf/meta_graph_pb2.pyi index 78ea938af3f4..e5357894c7a2 100644 --- a/stubs/tensorflow/tensorflow/core/protobuf/meta_graph_pb2.pyi +++ b/stubs/tensorflow/tensorflow/core/protobuf/meta_graph_pb2.pyi @@ -55,12 +55,7 @@ class MetaGraphDef(google.protobuf.message.Message): VALUE_FIELD_NUMBER: builtins.int key: builtins.str value: builtins.str - def __init__( - self, - *, - key: builtins.str | None = ..., - value: builtins.str | None = ..., - ) -> None: ... + def __init__(self, *, key: builtins.str | None = ..., value: builtins.str | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["key", b"key", "value", b"value"]) -> None: ... META_GRAPH_VERSION_FIELD_NUMBER: builtins.int @@ -127,8 +122,30 @@ class MetaGraphDef(google.protobuf.message.Message): stripped_default_attrs: builtins.bool | None = ..., function_aliases: collections.abc.Mapping[builtins.str, builtins.str] | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["any_info", b"any_info", "stripped_op_list", b"stripped_op_list"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["any_info", b"any_info", "function_aliases", b"function_aliases", "meta_graph_version", b"meta_graph_version", "stripped_default_attrs", b"stripped_default_attrs", "stripped_op_list", b"stripped_op_list", "tags", b"tags", "tensorflow_git_version", b"tensorflow_git_version", "tensorflow_version", b"tensorflow_version"]) -> None: ... + def HasField( + self, field_name: typing.Literal["any_info", b"any_info", "stripped_op_list", b"stripped_op_list"] + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "any_info", + b"any_info", + "function_aliases", + b"function_aliases", + "meta_graph_version", + b"meta_graph_version", + "stripped_default_attrs", + b"stripped_default_attrs", + "stripped_op_list", + b"stripped_op_list", + "tags", + b"tags", + "tensorflow_git_version", + b"tensorflow_git_version", + "tensorflow_version", + b"tensorflow_version", + ], + ) -> None: ... @typing.final class CollectionDefEntry(google.protobuf.message.Message): @@ -139,12 +156,7 @@ class MetaGraphDef(google.protobuf.message.Message): key: builtins.str @property def value(self) -> global___CollectionDef: ... - def __init__( - self, - *, - key: builtins.str | None = ..., - value: global___CollectionDef | None = ..., - ) -> None: ... + def __init__(self, *, key: builtins.str | None = ..., value: global___CollectionDef | None = ...) -> None: ... def HasField(self, field_name: typing.Literal["value", b"value"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["key", b"key", "value", b"value"]) -> None: ... @@ -157,12 +169,7 @@ class MetaGraphDef(google.protobuf.message.Message): key: builtins.str @property def value(self) -> global___SignatureDef: ... - def __init__( - self, - *, - key: builtins.str | None = ..., - value: global___SignatureDef | None = ..., - ) -> None: ... + def __init__(self, *, key: builtins.str | None = ..., value: global___SignatureDef | None = ...) -> None: ... def HasField(self, field_name: typing.Literal["value", b"value"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["key", b"key", "value", b"value"]) -> None: ... @@ -214,8 +221,38 @@ class MetaGraphDef(google.protobuf.message.Message): asset_file_def: collections.abc.Iterable[global___AssetFileDef] | None = ..., object_graph_def: tensorflow.core.protobuf.saved_object_graph_pb2.SavedObjectGraph | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["graph_def", b"graph_def", "meta_info_def", b"meta_info_def", "object_graph_def", b"object_graph_def", "saver_def", b"saver_def"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["asset_file_def", b"asset_file_def", "collection_def", b"collection_def", "graph_def", b"graph_def", "meta_info_def", b"meta_info_def", "object_graph_def", b"object_graph_def", "saver_def", b"saver_def", "signature_def", b"signature_def"]) -> None: ... + def HasField( + self, + field_name: typing.Literal[ + "graph_def", + b"graph_def", + "meta_info_def", + b"meta_info_def", + "object_graph_def", + b"object_graph_def", + "saver_def", + b"saver_def", + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "asset_file_def", + b"asset_file_def", + "collection_def", + b"collection_def", + "graph_def", + b"graph_def", + "meta_info_def", + b"meta_info_def", + "object_graph_def", + b"object_graph_def", + "saver_def", + b"saver_def", + "signature_def", + b"signature_def", + ], + ) -> None: ... global___MetaGraphDef = MetaGraphDef @@ -306,11 +343,7 @@ class CollectionDef(google.protobuf.message.Message): VALUE_FIELD_NUMBER: builtins.int @property def value(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]: ... - def __init__( - self, - *, - value: collections.abc.Iterable[builtins.str] | None = ..., - ) -> None: ... + def __init__(self, *, value: collections.abc.Iterable[builtins.str] | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["value", b"value"]) -> None: ... @typing.final @@ -335,11 +368,7 @@ class CollectionDef(google.protobuf.message.Message): VALUE_FIELD_NUMBER: builtins.int @property def value(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.bytes]: ... - def __init__( - self, - *, - value: collections.abc.Iterable[builtins.bytes] | None = ..., - ) -> None: ... + def __init__(self, *, value: collections.abc.Iterable[builtins.bytes] | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["value", b"value"]) -> None: ... @typing.final @@ -351,11 +380,7 @@ class CollectionDef(google.protobuf.message.Message): VALUE_FIELD_NUMBER: builtins.int @property def value(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.int]: ... - def __init__( - self, - *, - value: collections.abc.Iterable[builtins.int] | None = ..., - ) -> None: ... + def __init__(self, *, value: collections.abc.Iterable[builtins.int] | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["value", b"value"]) -> None: ... @typing.final @@ -367,11 +392,7 @@ class CollectionDef(google.protobuf.message.Message): VALUE_FIELD_NUMBER: builtins.int @property def value(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.float]: ... - def __init__( - self, - *, - value: collections.abc.Iterable[builtins.float] | None = ..., - ) -> None: ... + def __init__(self, *, value: collections.abc.Iterable[builtins.float] | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["value", b"value"]) -> None: ... @typing.final @@ -383,11 +404,7 @@ class CollectionDef(google.protobuf.message.Message): VALUE_FIELD_NUMBER: builtins.int @property def value(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[google.protobuf.any_pb2.Any]: ... - def __init__( - self, - *, - value: collections.abc.Iterable[google.protobuf.any_pb2.Any] | None = ..., - ) -> None: ... + def __init__(self, *, value: collections.abc.Iterable[google.protobuf.any_pb2.Any] | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["value", b"value"]) -> None: ... NODE_LIST_FIELD_NUMBER: builtins.int @@ -414,9 +431,43 @@ class CollectionDef(google.protobuf.message.Message): float_list: global___CollectionDef.FloatList | None = ..., any_list: global___CollectionDef.AnyList | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["any_list", b"any_list", "bytes_list", b"bytes_list", "float_list", b"float_list", "int64_list", b"int64_list", "kind", b"kind", "node_list", b"node_list"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["any_list", b"any_list", "bytes_list", b"bytes_list", "float_list", b"float_list", "int64_list", b"int64_list", "kind", b"kind", "node_list", b"node_list"]) -> None: ... - def WhichOneof(self, oneof_group: typing.Literal["kind", b"kind"]) -> typing.Literal["node_list", "bytes_list", "int64_list", "float_list", "any_list"] | None: ... + def HasField( + self, + field_name: typing.Literal[ + "any_list", + b"any_list", + "bytes_list", + b"bytes_list", + "float_list", + b"float_list", + "int64_list", + b"int64_list", + "kind", + b"kind", + "node_list", + b"node_list", + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "any_list", + b"any_list", + "bytes_list", + b"bytes_list", + "float_list", + b"float_list", + "int64_list", + b"int64_list", + "kind", + b"kind", + "node_list", + b"node_list", + ], + ) -> None: ... + def WhichOneof( + self, oneof_group: typing.Literal["kind", b"kind"] + ) -> typing.Literal["node_list", "bytes_list", "int64_list", "float_list", "any_list"] | None: ... global___CollectionDef = CollectionDef @@ -454,7 +505,17 @@ class TensorInfo(google.protobuf.message.Message): indices_tensor_name: builtins.str | None = ..., dense_shape_tensor_name: builtins.str | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["dense_shape_tensor_name", b"dense_shape_tensor_name", "indices_tensor_name", b"indices_tensor_name", "values_tensor_name", b"values_tensor_name"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "dense_shape_tensor_name", + b"dense_shape_tensor_name", + "indices_tensor_name", + b"indices_tensor_name", + "values_tensor_name", + b"values_tensor_name", + ], + ) -> None: ... @typing.final class CompositeTensor(google.protobuf.message.Message): @@ -517,9 +578,41 @@ class TensorInfo(google.protobuf.message.Message): dtype: tensorflow.core.framework.types_pb2.DataType.ValueType | None = ..., tensor_shape: tensorflow.core.framework.tensor_shape_pb2.TensorShapeProto | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["composite_tensor", b"composite_tensor", "coo_sparse", b"coo_sparse", "encoding", b"encoding", "name", b"name", "tensor_shape", b"tensor_shape"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["composite_tensor", b"composite_tensor", "coo_sparse", b"coo_sparse", "dtype", b"dtype", "encoding", b"encoding", "name", b"name", "tensor_shape", b"tensor_shape"]) -> None: ... - def WhichOneof(self, oneof_group: typing.Literal["encoding", b"encoding"]) -> typing.Literal["name", "coo_sparse", "composite_tensor"] | None: ... + def HasField( + self, + field_name: typing.Literal[ + "composite_tensor", + b"composite_tensor", + "coo_sparse", + b"coo_sparse", + "encoding", + b"encoding", + "name", + b"name", + "tensor_shape", + b"tensor_shape", + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "composite_tensor", + b"composite_tensor", + "coo_sparse", + b"coo_sparse", + "dtype", + b"dtype", + "encoding", + b"encoding", + "name", + b"name", + "tensor_shape", + b"tensor_shape", + ], + ) -> None: ... + def WhichOneof( + self, oneof_group: typing.Literal["encoding", b"encoding"] + ) -> typing.Literal["name", "coo_sparse", "composite_tensor"] | None: ... global___TensorInfo = TensorInfo @@ -540,12 +633,7 @@ class SignatureDef(google.protobuf.message.Message): key: builtins.str @property def value(self) -> global___TensorInfo: ... - def __init__( - self, - *, - key: builtins.str | None = ..., - value: global___TensorInfo | None = ..., - ) -> None: ... + def __init__(self, *, key: builtins.str | None = ..., value: global___TensorInfo | None = ...) -> None: ... def HasField(self, field_name: typing.Literal["value", b"value"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["key", b"key", "value", b"value"]) -> None: ... @@ -558,12 +646,7 @@ class SignatureDef(google.protobuf.message.Message): key: builtins.str @property def value(self) -> global___TensorInfo: ... - def __init__( - self, - *, - key: builtins.str | None = ..., - value: global___TensorInfo | None = ..., - ) -> None: ... + def __init__(self, *, key: builtins.str | None = ..., value: global___TensorInfo | None = ...) -> None: ... def HasField(self, field_name: typing.Literal["value", b"value"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["key", b"key", "value", b"value"]) -> None: ... @@ -577,10 +660,7 @@ class SignatureDef(google.protobuf.message.Message): @property def value(self) -> tensorflow.core.framework.tensor_pb2.TensorProto: ... def __init__( - self, - *, - key: builtins.str | None = ..., - value: tensorflow.core.framework.tensor_pb2.TensorProto | None = ..., + self, *, key: builtins.str | None = ..., value: tensorflow.core.framework.tensor_pb2.TensorProto | None = ... ) -> None: ... def HasField(self, field_name: typing.Literal["value", b"value"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["key", b"key", "value", b"value"]) -> None: ... @@ -607,7 +687,9 @@ class SignatureDef(google.protobuf.message.Message): """Named output parameters.""" @property - def defaults(self) -> google.protobuf.internal.containers.MessageMap[builtins.str, tensorflow.core.framework.tensor_pb2.TensorProto]: + def defaults( + self, + ) -> google.protobuf.internal.containers.MessageMap[builtins.str, tensorflow.core.framework.tensor_pb2.TensorProto]: """Named input to corresponding default values if any.""" def __init__( @@ -618,7 +700,12 @@ class SignatureDef(google.protobuf.message.Message): method_name: builtins.str | None = ..., defaults: collections.abc.Mapping[builtins.str, tensorflow.core.framework.tensor_pb2.TensorProto] | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["defaults", b"defaults", "inputs", b"inputs", "method_name", b"method_name", "outputs", b"outputs"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "defaults", b"defaults", "inputs", b"inputs", "method_name", b"method_name", "outputs", b"outputs" + ], + ) -> None: ... global___SignatureDef = SignatureDef @@ -641,12 +728,7 @@ class AssetFileDef(google.protobuf.message.Message): def tensor_info(self) -> global___TensorInfo: """The tensor to bind the asset filename to.""" - def __init__( - self, - *, - tensor_info: global___TensorInfo | None = ..., - filename: builtins.str | None = ..., - ) -> None: ... + def __init__(self, *, tensor_info: global___TensorInfo | None = ..., filename: builtins.str | None = ...) -> None: ... def HasField(self, field_name: typing.Literal["tensor_info", b"tensor_info"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["filename", b"filename", "tensor_info", b"tensor_info"]) -> None: ... diff --git a/stubs/tensorflow/tensorflow/core/protobuf/named_tensor_pb2.pyi b/stubs/tensorflow/tensorflow/core/protobuf/named_tensor_pb2.pyi index c1c17417d31c..5128864c6868 100644 --- a/stubs/tensorflow/tensorflow/core/protobuf/named_tensor_pb2.pyi +++ b/stubs/tensorflow/tensorflow/core/protobuf/named_tensor_pb2.pyi @@ -33,10 +33,7 @@ class NamedTensorProto(google.protobuf.message.Message): """ def __init__( - self, - *, - name: builtins.str | None = ..., - tensor: tensorflow.core.framework.tensor_pb2.TensorProto | None = ..., + self, *, name: builtins.str | None = ..., tensor: tensorflow.core.framework.tensor_pb2.TensorProto | None = ... ) -> None: ... def HasField(self, field_name: typing.Literal["tensor", b"tensor"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["name", b"name", "tensor", b"tensor"]) -> None: ... diff --git a/stubs/tensorflow/tensorflow/core/protobuf/queue_runner_pb2.pyi b/stubs/tensorflow/tensorflow/core/protobuf/queue_runner_pb2.pyi index f93fe61c3752..68d0cda5c153 100644 --- a/stubs/tensorflow/tensorflow/core/protobuf/queue_runner_pb2.pyi +++ b/stubs/tensorflow/tensorflow/core/protobuf/queue_runner_pb2.pyi @@ -36,7 +36,9 @@ class QueueRunnerDef(google.protobuf.message.Message): """A list of enqueue operations.""" @property - def queue_closed_exception_types(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[tensorflow.tsl.protobuf.error_codes_pb2.Code.ValueType]: + def queue_closed_exception_types( + self, + ) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[tensorflow.tsl.protobuf.error_codes_pb2.Code.ValueType]: """A list of exception types considered to signal a safely closed queue if raised during enqueue operations. """ @@ -48,8 +50,24 @@ class QueueRunnerDef(google.protobuf.message.Message): enqueue_op_name: collections.abc.Iterable[builtins.str] | None = ..., close_op_name: builtins.str | None = ..., cancel_op_name: builtins.str | None = ..., - queue_closed_exception_types: collections.abc.Iterable[tensorflow.tsl.protobuf.error_codes_pb2.Code.ValueType] | None = ..., + queue_closed_exception_types: ( + collections.abc.Iterable[tensorflow.tsl.protobuf.error_codes_pb2.Code.ValueType] | None + ) = ..., + ) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "cancel_op_name", + b"cancel_op_name", + "close_op_name", + b"close_op_name", + "enqueue_op_name", + b"enqueue_op_name", + "queue_closed_exception_types", + b"queue_closed_exception_types", + "queue_name", + b"queue_name", + ], ) -> None: ... - def ClearField(self, field_name: typing.Literal["cancel_op_name", b"cancel_op_name", "close_op_name", b"close_op_name", "enqueue_op_name", b"enqueue_op_name", "queue_closed_exception_types", b"queue_closed_exception_types", "queue_name", b"queue_name"]) -> None: ... global___QueueRunnerDef = QueueRunnerDef diff --git a/stubs/tensorflow/tensorflow/core/protobuf/remote_tensor_handle_pb2.pyi b/stubs/tensorflow/tensorflow/core/protobuf/remote_tensor_handle_pb2.pyi index 6558122b328b..4450b2e472ff 100644 --- a/stubs/tensorflow/tensorflow/core/protobuf/remote_tensor_handle_pb2.pyi +++ b/stubs/tensorflow/tensorflow/core/protobuf/remote_tensor_handle_pb2.pyi @@ -60,7 +60,9 @@ class RemoteTensorHandle(google.protobuf.message.Message): dtype: tensorflow.core.framework.types_pb2.DataType.ValueType """Tensor type.""" @property - def resource_dtypes_and_shapes(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___ResourceDtypeAndShape]: + def resource_dtypes_and_shapes( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___ResourceDtypeAndShape]: """Optional data types and shapes of a remote resource variable.""" def __init__( @@ -73,6 +75,22 @@ class RemoteTensorHandle(google.protobuf.message.Message): dtype: tensorflow.core.framework.types_pb2.DataType.ValueType | None = ..., resource_dtypes_and_shapes: collections.abc.Iterable[global___ResourceDtypeAndShape] | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["device", b"device", "dtype", b"dtype", "op_device", b"op_device", "op_id", b"op_id", "output_num", b"output_num", "resource_dtypes_and_shapes", b"resource_dtypes_and_shapes"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "device", + b"device", + "dtype", + b"dtype", + "op_device", + b"op_device", + "op_id", + b"op_id", + "output_num", + b"output_num", + "resource_dtypes_and_shapes", + b"resource_dtypes_and_shapes", + ], + ) -> None: ... global___RemoteTensorHandle = RemoteTensorHandle diff --git a/stubs/tensorflow/tensorflow/core/protobuf/rewriter_config_pb2.pyi b/stubs/tensorflow/tensorflow/core/protobuf/rewriter_config_pb2.pyi index ebe2295bc872..0779991c7dd3 100644 --- a/stubs/tensorflow/tensorflow/core/protobuf/rewriter_config_pb2.pyi +++ b/stubs/tensorflow/tensorflow/core/protobuf/rewriter_config_pb2.pyi @@ -30,12 +30,7 @@ class AutoParallelOptions(google.protobuf.message.Message): NUM_REPLICAS_FIELD_NUMBER: builtins.int enable: builtins.bool num_replicas: builtins.int - def __init__( - self, - *, - enable: builtins.bool | None = ..., - num_replicas: builtins.int | None = ..., - ) -> None: ... + def __init__(self, *, enable: builtins.bool | None = ..., num_replicas: builtins.int | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["enable", b"enable", "num_replicas", b"num_replicas"]) -> None: ... global___AutoParallelOptions = AutoParallelOptions @@ -49,11 +44,7 @@ class ScopedAllocatorOptions(google.protobuf.message.Message): def enable_op(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]: """If present, only perform optimization for these ops.""" - def __init__( - self, - *, - enable_op: collections.abc.Iterable[builtins.str] | None = ..., - ) -> None: ... + def __init__(self, *, enable_op: collections.abc.Iterable[builtins.str] | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["enable_op", b"enable_op"]) -> None: ... global___ScopedAllocatorOptions = ScopedAllocatorOptions @@ -70,7 +61,9 @@ class RewriterConfig(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _ToggleEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[RewriterConfig._Toggle.ValueType], builtins.type): + class _ToggleEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[RewriterConfig._Toggle.ValueType], builtins.type + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor DEFAULT: RewriterConfig._Toggle.ValueType # 0 ON: RewriterConfig._Toggle.ValueType # 1 @@ -118,7 +111,9 @@ class RewriterConfig(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _CpuLayoutEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[RewriterConfig._CpuLayout.ValueType], builtins.type): + class _CpuLayoutEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[RewriterConfig._CpuLayout.ValueType], builtins.type + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor NO_CONVERSION_ON_CPU: RewriterConfig._CpuLayout.ValueType # 0 NCHW_TO_NHWC: RewriterConfig._CpuLayout.ValueType # 1 @@ -135,7 +130,9 @@ class RewriterConfig(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _NumIterationsTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[RewriterConfig._NumIterationsType.ValueType], builtins.type): + class _NumIterationsTypeEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[RewriterConfig._NumIterationsType.ValueType], builtins.type + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor DEFAULT_NUM_ITERS: RewriterConfig._NumIterationsType.ValueType # 0 ONE: RewriterConfig._NumIterationsType.ValueType # 1 @@ -154,7 +151,9 @@ class RewriterConfig(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _MemOptTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[RewriterConfig._MemOptType.ValueType], builtins.type): + class _MemOptTypeEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[RewriterConfig._MemOptType.ValueType], builtins.type + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor DEFAULT_MEM_OPT: RewriterConfig._MemOptType.ValueType # 0 """The default setting (SCHEDULING and SWAPPING HEURISTICS only)""" @@ -225,10 +224,7 @@ class RewriterConfig(google.protobuf.message.Message): @property def value(self) -> tensorflow.core.framework.attr_value_pb2.AttrValue: ... def __init__( - self, - *, - key: builtins.str | None = ..., - value: tensorflow.core.framework.attr_value_pb2.AttrValue | None = ..., + self, *, key: builtins.str | None = ..., value: tensorflow.core.framework.attr_value_pb2.AttrValue | None = ... ) -> None: ... def HasField(self, field_name: typing.Literal["value", b"value"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["key", b"key", "value", b"value"]) -> None: ... @@ -237,7 +233,9 @@ class RewriterConfig(google.protobuf.message.Message): PARAMETER_MAP_FIELD_NUMBER: builtins.int name: builtins.str @property - def parameter_map(self) -> google.protobuf.internal.containers.MessageMap[builtins.str, tensorflow.core.framework.attr_value_pb2.AttrValue]: ... + def parameter_map( + self, + ) -> google.protobuf.internal.containers.MessageMap[builtins.str, tensorflow.core.framework.attr_value_pb2.AttrValue]: ... def __init__( self, *, @@ -438,7 +436,9 @@ class RewriterConfig(google.protobuf.message.Message): """ @property - def custom_optimizers(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___RewriterConfig.CustomGraphOptimizer]: + def custom_optimizers( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___RewriterConfig.CustomGraphOptimizer]: """list of CustomGraphOptimizers to apply.""" @property @@ -492,7 +492,97 @@ class RewriterConfig(google.protobuf.message.Message): inter_optimizer_verifier_config: tensorflow.core.protobuf.verifier_config_pb2.VerifierConfig | None = ..., post_optimization_verifier_config: tensorflow.core.protobuf.verifier_config_pb2.VerifierConfig | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["auto_parallel", b"auto_parallel", "inter_optimizer_verifier_config", b"inter_optimizer_verifier_config", "post_optimization_verifier_config", b"post_optimization_verifier_config", "scoped_allocator_opts", b"scoped_allocator_opts"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["arithmetic_optimization", b"arithmetic_optimization", "auto_mixed_precision", b"auto_mixed_precision", "auto_mixed_precision_cpu", b"auto_mixed_precision_cpu", "auto_mixed_precision_mkl", b"auto_mixed_precision_mkl", "auto_mixed_precision_onednn_bfloat16", b"auto_mixed_precision_onednn_bfloat16", "auto_parallel", b"auto_parallel", "common_subgraph_elimination", b"common_subgraph_elimination", "constant_folding", b"constant_folding", "cpu_layout_conversion", b"cpu_layout_conversion", "custom_optimizers", b"custom_optimizers", "debug_stripper", b"debug_stripper", "dependency_optimization", b"dependency_optimization", "disable_meta_optimizer", b"disable_meta_optimizer", "disable_model_pruning", b"disable_model_pruning", "disable_tfg_optimizer", b"disable_tfg_optimizer", "experimental_conditional_code_motion", b"experimental_conditional_code_motion", "experimental_disable_compressed_tensor_optimization", b"experimental_disable_compressed_tensor_optimization", "experimental_disable_folding_quantization_emulation", b"experimental_disable_folding_quantization_emulation", "fail_on_optimizer_errors", b"fail_on_optimizer_errors", "function_optimization", b"function_optimization", "implementation_selector", b"implementation_selector", "inter_optimizer_verifier_config", b"inter_optimizer_verifier_config", "layout_optimizer", b"layout_optimizer", "loop_optimization", b"loop_optimization", "memory_optimization", b"memory_optimization", "memory_optimizer_target_node_name_scope", b"memory_optimizer_target_node_name_scope", "meta_optimizer_iterations", b"meta_optimizer_iterations", "meta_optimizer_timeout_ms", b"meta_optimizer_timeout_ms", "min_graph_nodes", b"min_graph_nodes", "optimizers", b"optimizers", "pin_to_host_optimization", b"pin_to_host_optimization", "post_optimization_verifier_config", b"post_optimization_verifier_config", "remapping", b"remapping", "scoped_allocator_optimization", b"scoped_allocator_optimization", "scoped_allocator_opts", b"scoped_allocator_opts", "shape_optimization", b"shape_optimization", "use_plugin_optimizers", b"use_plugin_optimizers"]) -> None: ... + def HasField( + self, + field_name: typing.Literal[ + "auto_parallel", + b"auto_parallel", + "inter_optimizer_verifier_config", + b"inter_optimizer_verifier_config", + "post_optimization_verifier_config", + b"post_optimization_verifier_config", + "scoped_allocator_opts", + b"scoped_allocator_opts", + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "arithmetic_optimization", + b"arithmetic_optimization", + "auto_mixed_precision", + b"auto_mixed_precision", + "auto_mixed_precision_cpu", + b"auto_mixed_precision_cpu", + "auto_mixed_precision_mkl", + b"auto_mixed_precision_mkl", + "auto_mixed_precision_onednn_bfloat16", + b"auto_mixed_precision_onednn_bfloat16", + "auto_parallel", + b"auto_parallel", + "common_subgraph_elimination", + b"common_subgraph_elimination", + "constant_folding", + b"constant_folding", + "cpu_layout_conversion", + b"cpu_layout_conversion", + "custom_optimizers", + b"custom_optimizers", + "debug_stripper", + b"debug_stripper", + "dependency_optimization", + b"dependency_optimization", + "disable_meta_optimizer", + b"disable_meta_optimizer", + "disable_model_pruning", + b"disable_model_pruning", + "disable_tfg_optimizer", + b"disable_tfg_optimizer", + "experimental_conditional_code_motion", + b"experimental_conditional_code_motion", + "experimental_disable_compressed_tensor_optimization", + b"experimental_disable_compressed_tensor_optimization", + "experimental_disable_folding_quantization_emulation", + b"experimental_disable_folding_quantization_emulation", + "fail_on_optimizer_errors", + b"fail_on_optimizer_errors", + "function_optimization", + b"function_optimization", + "implementation_selector", + b"implementation_selector", + "inter_optimizer_verifier_config", + b"inter_optimizer_verifier_config", + "layout_optimizer", + b"layout_optimizer", + "loop_optimization", + b"loop_optimization", + "memory_optimization", + b"memory_optimization", + "memory_optimizer_target_node_name_scope", + b"memory_optimizer_target_node_name_scope", + "meta_optimizer_iterations", + b"meta_optimizer_iterations", + "meta_optimizer_timeout_ms", + b"meta_optimizer_timeout_ms", + "min_graph_nodes", + b"min_graph_nodes", + "optimizers", + b"optimizers", + "pin_to_host_optimization", + b"pin_to_host_optimization", + "post_optimization_verifier_config", + b"post_optimization_verifier_config", + "remapping", + b"remapping", + "scoped_allocator_optimization", + b"scoped_allocator_optimization", + "scoped_allocator_opts", + b"scoped_allocator_opts", + "shape_optimization", + b"shape_optimization", + "use_plugin_optimizers", + b"use_plugin_optimizers", + ], + ) -> None: ... global___RewriterConfig = RewriterConfig diff --git a/stubs/tensorflow/tensorflow/core/protobuf/saved_model_pb2.pyi b/stubs/tensorflow/tensorflow/core/protobuf/saved_model_pb2.pyi index cd2bdb7a2a13..e387ee87563d 100644 --- a/stubs/tensorflow/tensorflow/core/protobuf/saved_model_pb2.pyi +++ b/stubs/tensorflow/tensorflow/core/protobuf/saved_model_pb2.pyi @@ -30,7 +30,11 @@ class SavedModel(google.protobuf.message.Message): at release will be 1. """ @property - def meta_graphs(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[tensorflow.core.protobuf.meta_graph_pb2.MetaGraphDef]: + def meta_graphs( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[ + tensorflow.core.protobuf.meta_graph_pb2.MetaGraphDef + ]: """One or more MetaGraphs.""" def __init__( @@ -39,6 +43,9 @@ class SavedModel(google.protobuf.message.Message): saved_model_schema_version: builtins.int | None = ..., meta_graphs: collections.abc.Iterable[tensorflow.core.protobuf.meta_graph_pb2.MetaGraphDef] | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["meta_graphs", b"meta_graphs", "saved_model_schema_version", b"saved_model_schema_version"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal["meta_graphs", b"meta_graphs", "saved_model_schema_version", b"saved_model_schema_version"], + ) -> None: ... global___SavedModel = SavedModel diff --git a/stubs/tensorflow/tensorflow/core/protobuf/saved_object_graph_pb2.pyi b/stubs/tensorflow/tensorflow/core/protobuf/saved_object_graph_pb2.pyi index 4a6f0c1d51e0..ddc1f9d6a6ca 100644 --- a/stubs/tensorflow/tensorflow/core/protobuf/saved_object_graph_pb2.pyi +++ b/stubs/tensorflow/tensorflow/core/protobuf/saved_object_graph_pb2.pyi @@ -46,12 +46,7 @@ class SavedObjectGraph(google.protobuf.message.Message): key: builtins.str @property def value(self) -> global___SavedConcreteFunction: ... - def __init__( - self, - *, - key: builtins.str | None = ..., - value: global___SavedConcreteFunction | None = ..., - ) -> None: ... + def __init__(self, *, key: builtins.str | None = ..., value: global___SavedConcreteFunction | None = ...) -> None: ... def HasField(self, field_name: typing.Literal["value", b"value"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["key", b"key", "value", b"value"]) -> None: ... @@ -94,12 +89,7 @@ class SavedObject(google.protobuf.message.Message): key: builtins.str @property def value(self) -> global___SaveableObject: ... - def __init__( - self, - *, - key: builtins.str | None = ..., - value: global___SaveableObject | None = ..., - ) -> None: ... + def __init__(self, *, key: builtins.str | None = ..., value: global___SaveableObject | None = ...) -> None: ... def HasField(self, field_name: typing.Literal["value", b"value"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["key", b"key", "value", b"value"]) -> None: ... @@ -141,7 +131,11 @@ class SavedObject(google.protobuf.message.Message): `registered_saver` is defined for each SavedObject. """ @property - def children(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[tensorflow.core.protobuf.trackable_object_graph_pb2.TrackableObjectGraph.TrackableObject.ObjectReference]: + def children( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[ + tensorflow.core.protobuf.trackable_object_graph_pb2.TrackableObjectGraph.TrackableObject.ObjectReference + ]: """Objects which this object depends on: named edges in the dependency graph. @@ -150,14 +144,22 @@ class SavedObject(google.protobuf.message.Message): """ @property - def dependencies(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[tensorflow.core.protobuf.trackable_object_graph_pb2.TrackableObjectGraph.TrackableObject.ObjectReference]: + def dependencies( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[ + tensorflow.core.protobuf.trackable_object_graph_pb2.TrackableObjectGraph.TrackableObject.ObjectReference + ]: """Ordered list of dependencies that must be loaded before this object. SavedModel loads with the bottom-up approach, by first creating all objects (in the order defined by the dependencies), then connecting the edges. """ @property - def slot_variables(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[tensorflow.core.protobuf.trackable_object_graph_pb2.TrackableObjectGraph.TrackableObject.SlotVariableReference]: + def slot_variables( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[ + tensorflow.core.protobuf.trackable_object_graph_pb2.TrackableObjectGraph.TrackableObject.SlotVariableReference + ]: """Slot variables owned by this object. This describes the three-way (optimizer, variable, slot variable) relationship; none of the three depend on the others directly. @@ -199,9 +201,24 @@ class SavedObject(google.protobuf.message.Message): def __init__( self, *, - children: collections.abc.Iterable[tensorflow.core.protobuf.trackable_object_graph_pb2.TrackableObjectGraph.TrackableObject.ObjectReference] | None = ..., - dependencies: collections.abc.Iterable[tensorflow.core.protobuf.trackable_object_graph_pb2.TrackableObjectGraph.TrackableObject.ObjectReference] | None = ..., - slot_variables: collections.abc.Iterable[tensorflow.core.protobuf.trackable_object_graph_pb2.TrackableObjectGraph.TrackableObject.SlotVariableReference] | None = ..., + children: ( + collections.abc.Iterable[ + tensorflow.core.protobuf.trackable_object_graph_pb2.TrackableObjectGraph.TrackableObject.ObjectReference + ] + | None + ) = ..., + dependencies: ( + collections.abc.Iterable[ + tensorflow.core.protobuf.trackable_object_graph_pb2.TrackableObjectGraph.TrackableObject.ObjectReference + ] + | None + ) = ..., + slot_variables: ( + collections.abc.Iterable[ + tensorflow.core.protobuf.trackable_object_graph_pb2.TrackableObjectGraph.TrackableObject.SlotVariableReference + ] + | None + ) = ..., user_object: global___SavedUserObject | None = ..., asset: global___SavedAsset | None = ..., function: global___SavedFunction | None = ..., @@ -215,9 +232,76 @@ class SavedObject(google.protobuf.message.Message): serialized_user_proto: google.protobuf.any_pb2.Any | None = ..., registered_saver: builtins.str | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["asset", b"asset", "bare_concrete_function", b"bare_concrete_function", "captured_tensor", b"captured_tensor", "constant", b"constant", "function", b"function", "kind", b"kind", "resource", b"resource", "serialized_user_proto", b"serialized_user_proto", "user_object", b"user_object", "variable", b"variable"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["asset", b"asset", "bare_concrete_function", b"bare_concrete_function", "captured_tensor", b"captured_tensor", "children", b"children", "constant", b"constant", "dependencies", b"dependencies", "function", b"function", "kind", b"kind", "registered_name", b"registered_name", "registered_saver", b"registered_saver", "resource", b"resource", "saveable_objects", b"saveable_objects", "serialized_user_proto", b"serialized_user_proto", "slot_variables", b"slot_variables", "user_object", b"user_object", "variable", b"variable"]) -> None: ... - def WhichOneof(self, oneof_group: typing.Literal["kind", b"kind"]) -> typing.Literal["user_object", "asset", "function", "variable", "bare_concrete_function", "constant", "resource", "captured_tensor"] | None: ... + def HasField( + self, + field_name: typing.Literal[ + "asset", + b"asset", + "bare_concrete_function", + b"bare_concrete_function", + "captured_tensor", + b"captured_tensor", + "constant", + b"constant", + "function", + b"function", + "kind", + b"kind", + "resource", + b"resource", + "serialized_user_proto", + b"serialized_user_proto", + "user_object", + b"user_object", + "variable", + b"variable", + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "asset", + b"asset", + "bare_concrete_function", + b"bare_concrete_function", + "captured_tensor", + b"captured_tensor", + "children", + b"children", + "constant", + b"constant", + "dependencies", + b"dependencies", + "function", + b"function", + "kind", + b"kind", + "registered_name", + b"registered_name", + "registered_saver", + b"registered_saver", + "resource", + b"resource", + "saveable_objects", + b"saveable_objects", + "serialized_user_proto", + b"serialized_user_proto", + "slot_variables", + b"slot_variables", + "user_object", + b"user_object", + "variable", + b"variable", + ], + ) -> None: ... + def WhichOneof( + self, oneof_group: typing.Literal["kind", b"kind"] + ) -> ( + typing.Literal[ + "user_object", "asset", "function", "variable", "bare_concrete_function", "constant", "resource", "captured_tensor" + ] + | None + ): ... global___SavedObject = SavedObject @@ -257,7 +341,9 @@ class SavedUserObject(google.protobuf.message.Message): metadata: builtins.str | None = ..., ) -> None: ... def HasField(self, field_name: typing.Literal["version", b"version"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["identifier", b"identifier", "metadata", b"metadata", "version", b"version"]) -> None: ... + def ClearField( + self, field_name: typing.Literal["identifier", b"identifier", "metadata", b"metadata", "version", b"version"] + ) -> None: ... global___SavedUserObject = SavedUserObject @@ -279,11 +365,7 @@ class SavedAsset(google.protobuf.message.Message): Only the field `AssetFileDef.filename` is used. Other fields, such as `AssetFileDef.tensor_info`, MUST be ignored. """ - def __init__( - self, - *, - asset_file_def_index: builtins.int | None = ..., - ) -> None: ... + def __init__(self, *, asset_file_def_index: builtins.int | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["asset_file_def_index", b"asset_file_def_index"]) -> None: ... global___SavedAsset = SavedAsset @@ -307,7 +389,9 @@ class SavedFunction(google.protobuf.message.Message): function_spec: global___FunctionSpec | None = ..., ) -> None: ... def HasField(self, field_name: typing.Literal["function_spec", b"function_spec"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["concrete_functions", b"concrete_functions", "function_spec", b"function_spec"]) -> None: ... + def ClearField( + self, field_name: typing.Literal["concrete_functions", b"concrete_functions", "function_spec", b"function_spec"] + ) -> None: ... global___SavedFunction = SavedFunction @@ -321,12 +405,7 @@ class CapturedTensor(google.protobuf.message.Message): """Name of captured tensor""" concrete_function: builtins.str """Name of concrete function which contains the computed graph tensor.""" - def __init__( - self, - *, - name: builtins.str | None = ..., - concrete_function: builtins.str | None = ..., - ) -> None: ... + def __init__(self, *, name: builtins.str | None = ..., concrete_function: builtins.str | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["concrete_function", b"concrete_function", "name", b"name"]) -> None: ... global___CapturedTensor = CapturedTensor @@ -364,8 +443,23 @@ class SavedConcreteFunction(google.protobuf.message.Message): canonicalized_input_signature: tensorflow.core.protobuf.struct_pb2.StructuredValue | None = ..., output_signature: tensorflow.core.protobuf.struct_pb2.StructuredValue | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["canonicalized_input_signature", b"canonicalized_input_signature", "output_signature", b"output_signature"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["bound_inputs", b"bound_inputs", "canonicalized_input_signature", b"canonicalized_input_signature", "output_signature", b"output_signature"]) -> None: ... + def HasField( + self, + field_name: typing.Literal[ + "canonicalized_input_signature", b"canonicalized_input_signature", "output_signature", b"output_signature" + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "bound_inputs", + b"bound_inputs", + "canonicalized_input_signature", + b"canonicalized_input_signature", + "output_signature", + b"output_signature", + ], + ) -> None: ... global___SavedConcreteFunction = SavedConcreteFunction @@ -404,7 +498,19 @@ class SavedBareConcreteFunction(google.protobuf.message.Message): function_spec: global___FunctionSpec | None = ..., ) -> None: ... def HasField(self, field_name: typing.Literal["function_spec", b"function_spec"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["allowed_positional_arguments", b"allowed_positional_arguments", "argument_keywords", b"argument_keywords", "concrete_function_name", b"concrete_function_name", "function_spec", b"function_spec"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "allowed_positional_arguments", + b"allowed_positional_arguments", + "argument_keywords", + b"argument_keywords", + "concrete_function_name", + b"concrete_function_name", + "function_spec", + b"function_spec", + ], + ) -> None: ... global___SavedBareConcreteFunction = SavedBareConcreteFunction @@ -415,11 +521,7 @@ class SavedConstant(google.protobuf.message.Message): OPERATION_FIELD_NUMBER: builtins.int operation: builtins.str """An Operation name for a ConstantOp in this SavedObjectGraph's MetaGraph.""" - def __init__( - self, - *, - operation: builtins.str | None = ..., - ) -> None: ... + def __init__(self, *, operation: builtins.str | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["operation", b"operation"]) -> None: ... global___SavedConstant = SavedConstant @@ -449,7 +551,9 @@ class SavedVariable(google.protobuf.message.Message): @property def shape(self) -> tensorflow.core.framework.tensor_shape_pb2.TensorShapeProto: ... @property - def experimental_distributed_variable_components(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___SavedVariable]: + def experimental_distributed_variable_components( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___SavedVariable]: """List of component variables for a distributed variable. When this field is non-empty, the SavedVariable will be assumed @@ -471,7 +575,27 @@ class SavedVariable(google.protobuf.message.Message): experimental_distributed_variable_components: collections.abc.Iterable[global___SavedVariable] | None = ..., ) -> None: ... def HasField(self, field_name: typing.Literal["shape", b"shape"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["aggregation", b"aggregation", "device", b"device", "dtype", b"dtype", "experimental_distributed_variable_components", b"experimental_distributed_variable_components", "name", b"name", "shape", b"shape", "synchronization", b"synchronization", "trainable", b"trainable"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "aggregation", + b"aggregation", + "device", + b"device", + "dtype", + b"dtype", + "experimental_distributed_variable_components", + b"experimental_distributed_variable_components", + "name", + b"name", + "shape", + b"shape", + "synchronization", + b"synchronization", + "trainable", + b"trainable", + ], + ) -> None: ... global___SavedVariable = SavedVariable @@ -487,7 +611,9 @@ class FunctionSpec(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _JitCompileEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[FunctionSpec._JitCompile.ValueType], builtins.type): + class _JitCompileEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[FunctionSpec._JitCompile.ValueType], builtins.type + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor DEFAULT: FunctionSpec._JitCompile.ValueType # 0 ON: FunctionSpec._JitCompile.ValueType # 1 @@ -531,8 +657,22 @@ class FunctionSpec(google.protobuf.message.Message): input_signature: tensorflow.core.protobuf.struct_pb2.StructuredValue | None = ..., jit_compile: global___FunctionSpec.JitCompile.ValueType | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["fullargspec", b"fullargspec", "input_signature", b"input_signature"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["fullargspec", b"fullargspec", "input_signature", b"input_signature", "is_method", b"is_method", "jit_compile", b"jit_compile"]) -> None: ... + def HasField( + self, field_name: typing.Literal["fullargspec", b"fullargspec", "input_signature", b"input_signature"] + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "fullargspec", + b"fullargspec", + "input_signature", + b"input_signature", + "is_method", + b"is_method", + "jit_compile", + b"jit_compile", + ], + ) -> None: ... global___FunctionSpec = FunctionSpec @@ -551,11 +691,7 @@ class SavedResource(google.protobuf.message.Message): creation function, e.g. "CPU". An empty string allows the user to select a device. """ - def __init__( - self, - *, - device: builtins.str | None = ..., - ) -> None: ... + def __init__(self, *, device: builtins.str | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["device", b"device"]) -> None: ... global___SavedResource = SavedResource @@ -571,12 +707,9 @@ class SaveableObject(google.protobuf.message.Message): These functions save and restore directly from tensors. """ restore_function: builtins.int - def __init__( - self, - *, - save_function: builtins.int | None = ..., - restore_function: builtins.int | None = ..., + def __init__(self, *, save_function: builtins.int | None = ..., restore_function: builtins.int | None = ...) -> None: ... + def ClearField( + self, field_name: typing.Literal["restore_function", b"restore_function", "save_function", b"save_function"] ) -> None: ... - def ClearField(self, field_name: typing.Literal["restore_function", b"restore_function", "save_function", b"save_function"]) -> None: ... global___SaveableObject = SaveableObject diff --git a/stubs/tensorflow/tensorflow/core/protobuf/saver_pb2.pyi b/stubs/tensorflow/tensorflow/core/protobuf/saver_pb2.pyi index 1f89eebf8f98..42159a4dbb60 100644 --- a/stubs/tensorflow/tensorflow/core/protobuf/saver_pb2.pyi +++ b/stubs/tensorflow/tensorflow/core/protobuf/saver_pb2.pyi @@ -28,7 +28,9 @@ class SaverDef(google.protobuf.message.Message): 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): + 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.""" @@ -88,6 +90,24 @@ class SaverDef(google.protobuf.message.Message): 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: ... + 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 diff --git a/stubs/tensorflow/tensorflow/core/protobuf/service_config_pb2.pyi b/stubs/tensorflow/tensorflow/core/protobuf/service_config_pb2.pyi index d5e74f6d52f9..af7c8f598a43 100644 --- a/stubs/tensorflow/tensorflow/core/protobuf/service_config_pb2.pyi +++ b/stubs/tensorflow/tensorflow/core/protobuf/service_config_pb2.pyi @@ -110,7 +110,35 @@ class DispatcherConfig(google.protobuf.message.Message): 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: ... + 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 @@ -212,6 +240,36 @@ class WorkerConfig(google.protobuf.message.Message): 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: ... + 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 diff --git a/stubs/tensorflow/tensorflow/core/protobuf/snapshot_pb2.pyi b/stubs/tensorflow/tensorflow/core/protobuf/snapshot_pb2.pyi index daeb2da4ad65..10ff6f28e41f 100644 --- a/stubs/tensorflow/tensorflow/core/protobuf/snapshot_pb2.pyi +++ b/stubs/tensorflow/tensorflow/core/protobuf/snapshot_pb2.pyi @@ -27,11 +27,13 @@ class SnapshotRecord(google.protobuf.message.Message): TENSOR_FIELD_NUMBER: builtins.int @property - def tensor(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[tensorflow.core.framework.tensor_pb2.TensorProto]: ... - def __init__( + def tensor( self, - *, - tensor: collections.abc.Iterable[tensorflow.core.framework.tensor_pb2.TensorProto] | None = ..., + ) -> 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: ... @@ -64,7 +66,9 @@ class SnapshotMetadataRecord(google.protobuf.message.Message): """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]: + 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__( @@ -78,7 +82,25 @@ class SnapshotMetadataRecord(google.protobuf.message.Message): 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: ... + 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 @@ -101,7 +123,9 @@ class TensorMetadata(google.protobuf.message.Message): 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: ... + def ClearField( + self, field_name: typing.Literal["tensor_shape", b"tensor_shape", "tensor_size_bytes", b"tensor_size_bytes"] + ) -> None: ... global___TensorMetadata = TensorMetadata @@ -114,11 +138,7 @@ class SnapshotTensorMetadata(google.protobuf.message.Message): 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 __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 @@ -138,12 +158,7 @@ class DistributedSnapshotMetadata(google.protobuf.message.Message): `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 __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 diff --git a/stubs/tensorflow/tensorflow/core/protobuf/struct_pb2.pyi b/stubs/tensorflow/tensorflow/core/protobuf/struct_pb2.pyi index 0081e3011610..3d3bdfcaf808 100644 --- a/stubs/tensorflow/tensorflow/core/protobuf/struct_pb2.pyi +++ b/stubs/tensorflow/tensorflow/core/protobuf/struct_pb2.pyi @@ -151,9 +151,107 @@ class StructuredValue(google.protobuf.message.Message): tensor_value: tensorflow.core.framework.tensor_pb2.TensorProto | None = ..., numpy_value: tensorflow.core.framework.tensor_pb2.TensorProto | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["bool_value", b"bool_value", "bounded_tensor_spec_value", b"bounded_tensor_spec_value", "dict_value", b"dict_value", "float64_value", b"float64_value", "int64_value", b"int64_value", "kind", b"kind", "list_value", b"list_value", "named_tuple_value", b"named_tuple_value", "none_value", b"none_value", "numpy_value", b"numpy_value", "string_value", b"string_value", "tensor_dtype_value", b"tensor_dtype_value", "tensor_shape_value", b"tensor_shape_value", "tensor_spec_value", b"tensor_spec_value", "tensor_value", b"tensor_value", "tuple_value", b"tuple_value", "type_spec_value", b"type_spec_value"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["bool_value", b"bool_value", "bounded_tensor_spec_value", b"bounded_tensor_spec_value", "dict_value", b"dict_value", "float64_value", b"float64_value", "int64_value", b"int64_value", "kind", b"kind", "list_value", b"list_value", "named_tuple_value", b"named_tuple_value", "none_value", b"none_value", "numpy_value", b"numpy_value", "string_value", b"string_value", "tensor_dtype_value", b"tensor_dtype_value", "tensor_shape_value", b"tensor_shape_value", "tensor_spec_value", b"tensor_spec_value", "tensor_value", b"tensor_value", "tuple_value", b"tuple_value", "type_spec_value", b"type_spec_value"]) -> None: ... - def WhichOneof(self, oneof_group: typing.Literal["kind", b"kind"]) -> typing.Literal["none_value", "float64_value", "int64_value", "string_value", "bool_value", "tensor_shape_value", "tensor_dtype_value", "tensor_spec_value", "type_spec_value", "bounded_tensor_spec_value", "list_value", "tuple_value", "dict_value", "named_tuple_value", "tensor_value", "numpy_value"] | None: ... + def HasField( + self, + field_name: typing.Literal[ + "bool_value", + b"bool_value", + "bounded_tensor_spec_value", + b"bounded_tensor_spec_value", + "dict_value", + b"dict_value", + "float64_value", + b"float64_value", + "int64_value", + b"int64_value", + "kind", + b"kind", + "list_value", + b"list_value", + "named_tuple_value", + b"named_tuple_value", + "none_value", + b"none_value", + "numpy_value", + b"numpy_value", + "string_value", + b"string_value", + "tensor_dtype_value", + b"tensor_dtype_value", + "tensor_shape_value", + b"tensor_shape_value", + "tensor_spec_value", + b"tensor_spec_value", + "tensor_value", + b"tensor_value", + "tuple_value", + b"tuple_value", + "type_spec_value", + b"type_spec_value", + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "bool_value", + b"bool_value", + "bounded_tensor_spec_value", + b"bounded_tensor_spec_value", + "dict_value", + b"dict_value", + "float64_value", + b"float64_value", + "int64_value", + b"int64_value", + "kind", + b"kind", + "list_value", + b"list_value", + "named_tuple_value", + b"named_tuple_value", + "none_value", + b"none_value", + "numpy_value", + b"numpy_value", + "string_value", + b"string_value", + "tensor_dtype_value", + b"tensor_dtype_value", + "tensor_shape_value", + b"tensor_shape_value", + "tensor_spec_value", + b"tensor_spec_value", + "tensor_value", + b"tensor_value", + "tuple_value", + b"tuple_value", + "type_spec_value", + b"type_spec_value", + ], + ) -> None: ... + def WhichOneof( + self, oneof_group: typing.Literal["kind", b"kind"] + ) -> ( + typing.Literal[ + "none_value", + "float64_value", + "int64_value", + "string_value", + "bool_value", + "tensor_shape_value", + "tensor_dtype_value", + "tensor_spec_value", + "type_spec_value", + "bounded_tensor_spec_value", + "list_value", + "tuple_value", + "dict_value", + "named_tuple_value", + "tensor_value", + "numpy_value", + ] + | None + ): ... global___StructuredValue = StructuredValue @@ -163,9 +261,7 @@ class NoneValue(google.protobuf.message.Message): DESCRIPTOR: google.protobuf.descriptor.Descriptor - def __init__( - self, - ) -> None: ... + def __init__(self) -> None: ... global___NoneValue = NoneValue @@ -178,11 +274,7 @@ class ListValue(google.protobuf.message.Message): VALUES_FIELD_NUMBER: builtins.int @property def values(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___StructuredValue]: ... - def __init__( - self, - *, - values: collections.abc.Iterable[global___StructuredValue] | None = ..., - ) -> None: ... + def __init__(self, *, values: collections.abc.Iterable[global___StructuredValue] | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["values", b"values"]) -> None: ... global___ListValue = ListValue @@ -196,11 +288,7 @@ class TupleValue(google.protobuf.message.Message): VALUES_FIELD_NUMBER: builtins.int @property def values(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___StructuredValue]: ... - def __init__( - self, - *, - values: collections.abc.Iterable[global___StructuredValue] | None = ..., - ) -> None: ... + def __init__(self, *, values: collections.abc.Iterable[global___StructuredValue] | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["values", b"values"]) -> None: ... global___TupleValue = TupleValue @@ -222,23 +310,14 @@ class DictValue(google.protobuf.message.Message): key: builtins.str @property def value(self) -> global___StructuredValue: ... - def __init__( - self, - *, - key: builtins.str | None = ..., - value: global___StructuredValue | None = ..., - ) -> None: ... + def __init__(self, *, key: builtins.str | None = ..., value: global___StructuredValue | None = ...) -> None: ... def HasField(self, field_name: typing.Literal["value", b"value"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["key", b"key", "value", b"value"]) -> None: ... FIELDS_FIELD_NUMBER: builtins.int @property def fields(self) -> google.protobuf.internal.containers.MessageMap[builtins.str, global___StructuredValue]: ... - def __init__( - self, - *, - fields: collections.abc.Mapping[builtins.str, global___StructuredValue] | None = ..., - ) -> None: ... + def __init__(self, *, fields: collections.abc.Mapping[builtins.str, global___StructuredValue] | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["fields", b"fields"]) -> None: ... global___DictValue = DictValue @@ -254,12 +333,7 @@ class PairValue(google.protobuf.message.Message): key: builtins.str @property def value(self) -> global___StructuredValue: ... - def __init__( - self, - *, - key: builtins.str | None = ..., - value: global___StructuredValue | None = ..., - ) -> None: ... + def __init__(self, *, key: builtins.str | None = ..., value: global___StructuredValue | None = ...) -> None: ... def HasField(self, field_name: typing.Literal["value", b"value"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["key", b"key", "value", b"value"]) -> None: ... @@ -277,10 +351,7 @@ class NamedTupleValue(google.protobuf.message.Message): @property def values(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___PairValue]: ... def __init__( - self, - *, - name: builtins.str | None = ..., - values: collections.abc.Iterable[global___PairValue] | None = ..., + self, *, name: builtins.str | None = ..., values: collections.abc.Iterable[global___PairValue] | None = ... ) -> None: ... def ClearField(self, field_name: typing.Literal["name", b"name", "values", b"values"]) -> None: ... @@ -339,8 +410,15 @@ class BoundedTensorSpecProto(google.protobuf.message.Message): minimum: tensorflow.core.framework.tensor_pb2.TensorProto | None = ..., maximum: tensorflow.core.framework.tensor_pb2.TensorProto | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["maximum", b"maximum", "minimum", b"minimum", "shape", b"shape"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["dtype", b"dtype", "maximum", b"maximum", "minimum", b"minimum", "name", b"name", "shape", b"shape"]) -> None: ... + def HasField( + self, field_name: typing.Literal["maximum", b"maximum", "minimum", b"minimum", "shape", b"shape"] + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "dtype", b"dtype", "maximum", b"maximum", "minimum", b"minimum", "name", b"name", "shape", b"shape" + ], + ) -> None: ... global___BoundedTensorSpecProto = BoundedTensorSpecProto @@ -354,7 +432,9 @@ class TypeSpecProto(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _TypeSpecClassEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[TypeSpecProto._TypeSpecClass.ValueType], builtins.type): + class _TypeSpecClassEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[TypeSpecProto._TypeSpecClass.ValueType], builtins.type + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor UNKNOWN: TypeSpecProto._TypeSpecClass.ValueType # 0 SPARSE_TENSOR_SPEC: TypeSpecProto._TypeSpecClass.ValueType # 1 @@ -439,6 +519,18 @@ class TypeSpecProto(google.protobuf.message.Message): num_flat_components: builtins.int | None = ..., ) -> None: ... def HasField(self, field_name: typing.Literal["type_state", b"type_state"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["num_flat_components", b"num_flat_components", "type_spec_class", b"type_spec_class", "type_spec_class_name", b"type_spec_class_name", "type_state", b"type_state"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "num_flat_components", + b"num_flat_components", + "type_spec_class", + b"type_spec_class", + "type_spec_class_name", + b"type_spec_class_name", + "type_state", + b"type_state", + ], + ) -> None: ... global___TypeSpecProto = TypeSpecProto diff --git a/stubs/tensorflow/tensorflow/core/protobuf/tensor_bundle_pb2.pyi b/stubs/tensorflow/tensorflow/core/protobuf/tensor_bundle_pb2.pyi index b1b95ae3fab4..58f69470b734 100644 --- a/stubs/tensorflow/tensorflow/core/protobuf/tensor_bundle_pb2.pyi +++ b/stubs/tensorflow/tensorflow/core/protobuf/tensor_bundle_pb2.pyi @@ -43,7 +43,9 @@ class BundleHeaderProto(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _EndiannessEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[BundleHeaderProto._Endianness.ValueType], builtins.type): + class _EndiannessEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[BundleHeaderProto._Endianness.ValueType], builtins.type + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor LITTLE: BundleHeaderProto._Endianness.ValueType # 0 BIG: BundleHeaderProto._Endianness.ValueType # 1 @@ -77,7 +79,9 @@ class BundleHeaderProto(google.protobuf.message.Message): version: tensorflow.core.framework.versions_pb2.VersionDef | None = ..., ) -> None: ... def HasField(self, field_name: typing.Literal["version", b"version"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["endianness", b"endianness", "num_shards", b"num_shards", "version", b"version"]) -> None: ... + def ClearField( + self, field_name: typing.Literal["endianness", b"endianness", "num_shards", b"num_shards", "version", b"version"] + ) -> None: ... global___BundleHeaderProto = BundleHeaderProto @@ -107,7 +111,11 @@ class BundleEntryProto(google.protobuf.message.Message): @property def shape(self) -> tensorflow.core.framework.tensor_shape_pb2.TensorShapeProto: ... @property - def slices(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[tensorflow.core.framework.tensor_slice_pb2.TensorSliceProto]: + def slices( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[ + tensorflow.core.framework.tensor_slice_pb2.TensorSliceProto + ]: """Iff present, this entry represents a partitioned tensor. The previous fields are interpreted as follows: @@ -129,6 +137,24 @@ class BundleEntryProto(google.protobuf.message.Message): slices: collections.abc.Iterable[tensorflow.core.framework.tensor_slice_pb2.TensorSliceProto] | None = ..., ) -> None: ... def HasField(self, field_name: typing.Literal["shape", b"shape"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["crc32c", b"crc32c", "dtype", b"dtype", "offset", b"offset", "shape", b"shape", "shard_id", b"shard_id", "size", b"size", "slices", b"slices"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "crc32c", + b"crc32c", + "dtype", + b"dtype", + "offset", + b"offset", + "shape", + b"shape", + "shard_id", + b"shard_id", + "size", + b"size", + "slices", + b"slices", + ], + ) -> None: ... global___BundleEntryProto = BundleEntryProto diff --git a/stubs/tensorflow/tensorflow/core/protobuf/tensorflow_server_pb2.pyi b/stubs/tensorflow/tensorflow/core/protobuf/tensorflow_server_pb2.pyi index e5120748aeef..470738c92cb2 100644 --- a/stubs/tensorflow/tensorflow/core/protobuf/tensorflow_server_pb2.pyi +++ b/stubs/tensorflow/tensorflow/core/protobuf/tensorflow_server_pb2.pyi @@ -89,7 +89,37 @@ class ServerDef(google.protobuf.message.Message): port: builtins.int | None = ..., cluster_device_filters: tensorflow.core.protobuf.device_filters_pb2.ClusterDeviceFilters | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["cluster", b"cluster", "cluster_device_filters", b"cluster_device_filters", "default_session_config", b"default_session_config"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["cluster", b"cluster", "cluster_device_filters", b"cluster_device_filters", "default_session_config", b"default_session_config", "job_name", b"job_name", "port", b"port", "protocol", b"protocol", "replica", b"replica", "task_index", b"task_index"]) -> None: ... + def HasField( + self, + field_name: typing.Literal[ + "cluster", + b"cluster", + "cluster_device_filters", + b"cluster_device_filters", + "default_session_config", + b"default_session_config", + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "cluster", + b"cluster", + "cluster_device_filters", + b"cluster_device_filters", + "default_session_config", + b"default_session_config", + "job_name", + b"job_name", + "port", + b"port", + "protocol", + b"protocol", + "replica", + b"replica", + "task_index", + b"task_index", + ], + ) -> None: ... global___ServerDef = ServerDef diff --git a/stubs/tensorflow/tensorflow/core/protobuf/tpu/compilation_result_pb2.pyi b/stubs/tensorflow/tensorflow/core/protobuf/tpu/compilation_result_pb2.pyi index 6d04ebf01637..1220c78dc5f5 100644 --- a/stubs/tensorflow/tensorflow/core/protobuf/tpu/compilation_result_pb2.pyi +++ b/stubs/tensorflow/tensorflow/core/protobuf/tpu/compilation_result_pb2.pyi @@ -35,7 +35,9 @@ class CompilationResultProto(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _ErrorCodeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[CompilationResultProto._ErrorCode.ValueType], builtins.type): + class _ErrorCodeEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[CompilationResultProto._ErrorCode.ValueType], builtins.type + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor UNKNOWN: CompilationResultProto._ErrorCode.ValueType # 0 OUT_OF_MEMORY: CompilationResultProto._ErrorCode.ValueType # 1 @@ -53,7 +55,9 @@ class CompilationResultProto(google.protobuf.message.Message): status_error_message: builtins.str error_code: global___CompilationResultProto.ErrorCode.ValueType @property - def hlo_protos(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[tensorflow.compiler.xla.service.hlo_pb2.HloProto]: + def hlo_protos( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[tensorflow.compiler.xla.service.hlo_pb2.HloProto]: """HLO proto.""" def __init__( @@ -64,6 +68,18 @@ class CompilationResultProto(google.protobuf.message.Message): hlo_protos: collections.abc.Iterable[tensorflow.compiler.xla.service.hlo_pb2.HloProto] | None = ..., error_code: global___CompilationResultProto.ErrorCode.ValueType | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["error_code", b"error_code", "hlo_protos", b"hlo_protos", "status_code", b"status_code", "status_error_message", b"status_error_message"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "error_code", + b"error_code", + "hlo_protos", + b"hlo_protos", + "status_code", + b"status_code", + "status_error_message", + b"status_error_message", + ], + ) -> None: ... global___CompilationResultProto = CompilationResultProto diff --git a/stubs/tensorflow/tensorflow/core/protobuf/tpu/dynamic_padding_pb2.pyi b/stubs/tensorflow/tensorflow/core/protobuf/tpu/dynamic_padding_pb2.pyi index e8e639bc0b2d..d898f55864e3 100644 --- a/stubs/tensorflow/tensorflow/core/protobuf/tpu/dynamic_padding_pb2.pyi +++ b/stubs/tensorflow/tensorflow/core/protobuf/tpu/dynamic_padding_pb2.pyi @@ -37,6 +37,11 @@ class PaddingMap(google.protobuf.message.Message): shape_index: builtins.int | None = ..., padding_arg_index: builtins.int | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["arg_index", b"arg_index", "padding_arg_index", b"padding_arg_index", "shape_index", b"shape_index"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "arg_index", b"arg_index", "padding_arg_index", b"padding_arg_index", "shape_index", b"shape_index" + ], + ) -> None: ... global___PaddingMap = PaddingMap diff --git a/stubs/tensorflow/tensorflow/core/protobuf/tpu/optimization_parameters_pb2.pyi b/stubs/tensorflow/tensorflow/core/protobuf/tpu/optimization_parameters_pb2.pyi index 5e08f7ce0e06..9b914134520d 100644 --- a/stubs/tensorflow/tensorflow/core/protobuf/tpu/optimization_parameters_pb2.pyi +++ b/stubs/tensorflow/tensorflow/core/protobuf/tpu/optimization_parameters_pb2.pyi @@ -83,7 +83,10 @@ class SimulatedQuantization(google.protobuf.message.Message): num_buckets: builtins.int | None = ..., ) -> None: ... def HasField(self, field_name: typing.Literal["clipping_limits", b"clipping_limits"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["clipping_limits", b"clipping_limits", "enabled", b"enabled", "num_buckets", b"num_buckets"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal["clipping_limits", b"clipping_limits", "enabled", b"enabled", "num_buckets", b"num_buckets"], + ) -> None: ... global___SimulatedQuantization = SimulatedQuantization @@ -126,11 +129,7 @@ class DynamicLearningRate(google.protobuf.message.Message): particular tag is specified by populating its corresponding index in the list of learning_rate scalars. """ - def __init__( - self, - *, - tag: builtins.int | None = ..., - ) -> None: ... + def __init__(self, *, tag: builtins.int | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["tag", b"tag"]) -> None: ... global___DynamicLearningRate = DynamicLearningRate @@ -146,15 +145,16 @@ class LearningRate(google.protobuf.message.Message): constant: builtins.float @property def dynamic(self) -> global___DynamicLearningRate: ... - def __init__( - self, - *, - constant: builtins.float | None = ..., - dynamic: global___DynamicLearningRate | None = ..., + def __init__(self, *, constant: builtins.float | None = ..., dynamic: global___DynamicLearningRate | None = ...) -> None: ... + def HasField( + self, field_name: typing.Literal["constant", b"constant", "dynamic", b"dynamic", "learning_rate", b"learning_rate"] + ) -> builtins.bool: ... + def ClearField( + self, field_name: typing.Literal["constant", b"constant", "dynamic", b"dynamic", "learning_rate", b"learning_rate"] ) -> None: ... - def HasField(self, field_name: typing.Literal["constant", b"constant", "dynamic", b"dynamic", "learning_rate", b"learning_rate"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["constant", b"constant", "dynamic", b"dynamic", "learning_rate", b"learning_rate"]) -> None: ... - def WhichOneof(self, oneof_group: typing.Literal["learning_rate", b"learning_rate"]) -> typing.Literal["constant", "dynamic"] | None: ... + def WhichOneof( + self, oneof_group: typing.Literal["learning_rate", b"learning_rate"] + ) -> typing.Literal["constant", "dynamic"] | None: ... global___LearningRate = LearningRate @@ -169,9 +169,7 @@ class AdagradParameters(google.protobuf.message.Message): DESCRIPTOR: google.protobuf.descriptor.Descriptor - def __init__( - self, - ) -> None: ... + def __init__(self) -> None: ... global___AdagradParameters = AdagradParameters @@ -216,7 +214,21 @@ class AdagradMomentumParameters(google.protobuf.message.Message): beta2: builtins.float | None = ..., epsilon: builtins.float | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["beta2", b"beta2", "epsilon", b"epsilon", "exponent", b"exponent", "momentum", b"momentum", "use_nesterov", b"use_nesterov"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "beta2", + b"beta2", + "epsilon", + b"epsilon", + "exponent", + b"exponent", + "momentum", + b"momentum", + "use_nesterov", + b"use_nesterov", + ], + ) -> None: ... global___AdagradMomentumParameters = AdagradMomentumParameters @@ -249,7 +261,17 @@ class BoundedAdagradParameters(google.protobuf.message.Message): max_var_update: builtins.float | None = ..., max_accumulator: builtins.float | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["max_accumulator", b"max_accumulator", "max_var_update", b"max_var_update", "update_accumulator_first", b"update_accumulator_first"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "max_accumulator", + b"max_accumulator", + "max_var_update", + b"max_var_update", + "update_accumulator_first", + b"update_accumulator_first", + ], + ) -> None: ... global___BoundedAdagradParameters = BoundedAdagradParameters @@ -261,9 +283,7 @@ class StochasticGradientDescentParameters(google.protobuf.message.Message): DESCRIPTOR: google.protobuf.descriptor.Descriptor - def __init__( - self, - ) -> None: ... + def __init__(self) -> None: ... global___StochasticGradientDescentParameters = StochasticGradientDescentParameters @@ -317,7 +337,23 @@ class FtrlParameters(google.protobuf.message.Message): multiply_linear_by_lr: builtins.bool | None = ..., allow_zero_accumulator: builtins.bool | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["allow_zero_accumulator", b"allow_zero_accumulator", "beta", b"beta", "l1", b"l1", "l2", b"l2", "lr_power", b"lr_power", "multiply_linear_by_lr", b"multiply_linear_by_lr"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "allow_zero_accumulator", + b"allow_zero_accumulator", + "beta", + b"beta", + "l1", + b"l1", + "l2", + b"l2", + "lr_power", + b"lr_power", + "multiply_linear_by_lr", + b"multiply_linear_by_lr", + ], + ) -> None: ... global___FtrlParameters = FtrlParameters @@ -366,7 +402,21 @@ class AdamParameters(google.protobuf.message.Message): use_non_lazy_adam: builtins.bool | None = ..., use_sum_inside_sqrt: builtins.bool | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["beta1", b"beta1", "beta2", b"beta2", "epsilon", b"epsilon", "use_non_lazy_adam", b"use_non_lazy_adam", "use_sum_inside_sqrt", b"use_sum_inside_sqrt"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "beta1", + b"beta1", + "beta2", + b"beta2", + "epsilon", + b"epsilon", + "use_non_lazy_adam", + b"use_non_lazy_adam", + "use_sum_inside_sqrt", + b"use_sum_inside_sqrt", + ], + ) -> None: ... global___AdamParameters = AdamParameters @@ -382,12 +432,7 @@ class MomentumParameters(google.protobuf.message.Message): USE_NESTEROV_FIELD_NUMBER: builtins.int momentum: builtins.float use_nesterov: builtins.bool - def __init__( - self, - *, - momentum: builtins.float | None = ..., - use_nesterov: builtins.bool | None = ..., - ) -> None: ... + def __init__(self, *, momentum: builtins.float | None = ..., use_nesterov: builtins.bool | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["momentum", b"momentum", "use_nesterov", b"use_nesterov"]) -> None: ... global___MomentumParameters = MomentumParameters @@ -416,7 +461,9 @@ class LionParameters(google.protobuf.message.Message): beta2: builtins.float | None = ..., use_non_lazy_lion: builtins.bool | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["beta1", b"beta1", "beta2", b"beta2", "use_non_lazy_lion", b"use_non_lazy_lion"]) -> None: ... + def ClearField( + self, field_name: typing.Literal["beta1", b"beta1", "beta2", b"beta2", "use_non_lazy_lion", b"use_non_lazy_lion"] + ) -> None: ... global___LionParameters = LionParameters @@ -435,11 +482,7 @@ class RmsPropParameters(google.protobuf.message.Message): momentum: builtins.float epsilon: builtins.float def __init__( - self, - *, - rho: builtins.float | None = ..., - momentum: builtins.float | None = ..., - epsilon: builtins.float | None = ..., + self, *, rho: builtins.float | None = ..., momentum: builtins.float | None = ..., epsilon: builtins.float | None = ... ) -> None: ... def ClearField(self, field_name: typing.Literal["epsilon", b"epsilon", "momentum", b"momentum", "rho", b"rho"]) -> None: ... @@ -460,11 +503,7 @@ class CenteredRmsPropParameters(google.protobuf.message.Message): momentum: builtins.float epsilon: builtins.float def __init__( - self, - *, - rho: builtins.float | None = ..., - momentum: builtins.float | None = ..., - epsilon: builtins.float | None = ..., + self, *, rho: builtins.float | None = ..., momentum: builtins.float | None = ..., epsilon: builtins.float | None = ... ) -> None: ... def ClearField(self, field_name: typing.Literal["epsilon", b"epsilon", "momentum", b"momentum", "rho", b"rho"]) -> None: ... @@ -516,7 +555,35 @@ class MdlAdagradLightParameters(google.protobuf.message.Message): hard_limit_min_benefit: builtins.bool | None = ..., mdl_regularize: builtins.bool | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["benefit_revisit_scale", b"benefit_revisit_scale", "hard_limit_min_benefit", b"hard_limit_min_benefit", "l2", b"l2", "lr_power", b"lr_power", "max_event_benefit", b"max_event_benefit", "max_total_benefit", b"max_total_benefit", "mdl_benefit_rampup_coeff", b"mdl_benefit_rampup_coeff", "mdl_hard_limit", b"mdl_hard_limit", "mdl_min_weight", b"mdl_min_weight", "mdl_mix_in_margin", b"mdl_mix_in_margin", "mdl_regularize", b"mdl_regularize", "min_servable_mdl_benefit", b"min_servable_mdl_benefit"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "benefit_revisit_scale", + b"benefit_revisit_scale", + "hard_limit_min_benefit", + b"hard_limit_min_benefit", + "l2", + b"l2", + "lr_power", + b"lr_power", + "max_event_benefit", + b"max_event_benefit", + "max_total_benefit", + b"max_total_benefit", + "mdl_benefit_rampup_coeff", + b"mdl_benefit_rampup_coeff", + "mdl_hard_limit", + b"mdl_hard_limit", + "mdl_min_weight", + b"mdl_min_weight", + "mdl_mix_in_margin", + b"mdl_mix_in_margin", + "mdl_regularize", + b"mdl_regularize", + "min_servable_mdl_benefit", + b"min_servable_mdl_benefit", + ], + ) -> None: ... global___MdlAdagradLightParameters = MdlAdagradLightParameters @@ -532,12 +599,7 @@ class AdadeltaParameters(google.protobuf.message.Message): EPSILON_FIELD_NUMBER: builtins.int rho: builtins.float epsilon: builtins.float - def __init__( - self, - *, - rho: builtins.float | None = ..., - epsilon: builtins.float | None = ..., - ) -> None: ... + def __init__(self, *, rho: builtins.float | None = ..., epsilon: builtins.float | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["epsilon", b"epsilon", "rho", b"rho"]) -> None: ... global___AdadeltaParameters = AdadeltaParameters @@ -554,12 +616,7 @@ class ProximalAdagradParameters(google.protobuf.message.Message): L2_FIELD_NUMBER: builtins.int l1: builtins.float l2: builtins.float - def __init__( - self, - *, - l1: builtins.float | None = ..., - l2: builtins.float | None = ..., - ) -> None: ... + def __init__(self, *, l1: builtins.float | None = ..., l2: builtins.float | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["l1", b"l1", "l2", b"l2"]) -> None: ... global___ProximalAdagradParameters = ProximalAdagradParameters @@ -589,11 +646,7 @@ class OnlineYogiParameters(google.protobuf.message.Message): beta2: builtins.float """\\beta_2 from Algorithm 2 in the paper.""" def __init__( - self, - *, - l1: builtins.float | None = ..., - l2: builtins.float | None = ..., - beta2: builtins.float | None = ..., + self, *, l1: builtins.float | None = ..., l2: builtins.float | None = ..., beta2: builtins.float | None = ... ) -> None: ... def ClearField(self, field_name: typing.Literal["beta2", b"beta2", "l1", b"l1", "l2", b"l2"]) -> None: ... @@ -638,7 +691,9 @@ class ProximalYogiParameters(google.protobuf.message.Message): beta2: builtins.float | None = ..., epsilon: builtins.float | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["beta1", b"beta1", "beta2", b"beta2", "epsilon", b"epsilon", "l1", b"l1", "l2", b"l2"]) -> None: ... + def ClearField( + self, field_name: typing.Literal["beta1", b"beta1", "beta2", b"beta2", "epsilon", b"epsilon", "l1", b"l1", "l2", b"l2"] + ) -> None: ... global___ProximalYogiParameters = ProximalYogiParameters @@ -696,7 +751,19 @@ class FrequencyEstimatorParameters(google.protobuf.message.Message): outlier_threshold: builtins.float | None = ..., weight_exponent: builtins.float | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["max_delta", b"max_delta", "outlier_threshold", b"outlier_threshold", "tau", b"tau", "weight_exponent", b"weight_exponent"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "max_delta", + b"max_delta", + "outlier_threshold", + b"outlier_threshold", + "tau", + b"tau", + "weight_exponent", + b"weight_exponent", + ], + ) -> None: ... global___FrequencyEstimatorParameters = FrequencyEstimatorParameters @@ -728,11 +795,7 @@ class UserDefinedProgramParameters(google.protobuf.message.Message): PROGRAM_FIELD_NUMBER: builtins.int @property def program(self) -> tensorflow.compiler.xla.service.hlo_pb2.HloModuleProto: ... - def __init__( - self, - *, - program: tensorflow.compiler.xla.service.hlo_pb2.HloModuleProto | None = ..., - ) -> None: ... + def __init__(self, *, program: tensorflow.compiler.xla.service.hlo_pb2.HloModuleProto | None = ...) -> None: ... def HasField(self, field_name: typing.Literal["program", b"program"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["program", b"program"]) -> None: ... @@ -748,9 +811,7 @@ class AssignParameters(google.protobuf.message.Message): DESCRIPTOR: google.protobuf.descriptor.Descriptor - def __init__( - self, - ) -> None: ... + def __init__(self) -> None: ... global___AssignParameters = AssignParameters @@ -768,7 +829,9 @@ class GradientAccumulationStatus(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _StatusEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[GradientAccumulationStatus._Status.ValueType], builtins.type): + class _StatusEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[GradientAccumulationStatus._Status.ValueType], builtins.type + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor UNSPECIFIED: GradientAccumulationStatus._Status.ValueType # 0 ENABLED: GradientAccumulationStatus._Status.ValueType # 1 @@ -781,9 +844,7 @@ class GradientAccumulationStatus(google.protobuf.message.Message): ENABLED: GradientAccumulationStatus.Status.ValueType # 1 DISABLED: GradientAccumulationStatus.Status.ValueType # 2 - def __init__( - self, - ) -> None: ... + def __init__(self) -> None: ... global___GradientAccumulationStatus = GradientAccumulationStatus @@ -835,7 +896,9 @@ class LowDimensionalPackingStatus(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _StatusEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[LowDimensionalPackingStatus._Status.ValueType], builtins.type): + class _StatusEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[LowDimensionalPackingStatus._Status.ValueType], builtins.type + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor UNSPECIFIED: LowDimensionalPackingStatus._Status.ValueType # 0 ENABLED: LowDimensionalPackingStatus._Status.ValueType # 1 @@ -858,9 +921,7 @@ class LowDimensionalPackingStatus(google.protobuf.message.Message): ENABLED: LowDimensionalPackingStatus.Status.ValueType # 1 DISABLED: LowDimensionalPackingStatus.Status.ValueType # 2 - def __init__( - self, - ) -> None: ... + def __init__(self) -> None: ... global___LowDimensionalPackingStatus = LowDimensionalPackingStatus @@ -876,7 +937,10 @@ class HotIdReplicationConfiguration(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _StatusEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[HotIdReplicationConfiguration._Status.ValueType], builtins.type): + class _StatusEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[HotIdReplicationConfiguration._Status.ValueType], + builtins.type, + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor UNSPECIFIED: HotIdReplicationConfiguration._Status.ValueType # 0 ENABLED: HotIdReplicationConfiguration._Status.ValueType # 1 @@ -897,11 +961,7 @@ class HotIdReplicationConfiguration(google.protobuf.message.Message): STATUS_FIELD_NUMBER: builtins.int status: global___HotIdReplicationConfiguration.Status.ValueType - def __init__( - self, - *, - status: global___HotIdReplicationConfiguration.Status.ValueType | None = ..., - ) -> None: ... + def __init__(self, *, status: global___HotIdReplicationConfiguration.Status.ValueType | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["status", b"status"]) -> None: ... global___HotIdReplicationConfiguration = HotIdReplicationConfiguration @@ -1056,9 +1116,145 @@ class OptimizationParameters(google.protobuf.message.Message): user_defined_program: global___UserDefinedProgramParameters | None = ..., assign: global___AssignParameters | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["adadelta", b"adadelta", "adagrad", b"adagrad", "adagrad_momentum", b"adagrad_momentum", "adam", b"adam", "assign", b"assign", "bounded_adagrad", b"bounded_adagrad", "centered_rms_prop", b"centered_rms_prop", "clipping_limits", b"clipping_limits", "frequency_estimator", b"frequency_estimator", "ftrl", b"ftrl", "gradient_clipping_limits", b"gradient_clipping_limits", "hot_id_replication_configuration", b"hot_id_replication_configuration", "learning_rate", b"learning_rate", "lion", b"lion", "mdl_adagrad_light", b"mdl_adagrad_light", "momentum", b"momentum", "online_yogi", b"online_yogi", "parameters", b"parameters", "proximal_adagrad", b"proximal_adagrad", "proximal_yogi", b"proximal_yogi", "rms_prop", b"rms_prop", "simulated_quantization", b"simulated_quantization", "stochastic_gradient_descent", b"stochastic_gradient_descent", "user_defined_program", b"user_defined_program"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["adadelta", b"adadelta", "adagrad", b"adagrad", "adagrad_momentum", b"adagrad_momentum", "adam", b"adam", "assign", b"assign", "bounded_adagrad", b"bounded_adagrad", "centered_rms_prop", b"centered_rms_prop", "clipping_limits", b"clipping_limits", "frequency_estimator", b"frequency_estimator", "ftrl", b"ftrl", "gradient_accumulation_status", b"gradient_accumulation_status", "gradient_clipping_limits", b"gradient_clipping_limits", "hot_id_replication_configuration", b"hot_id_replication_configuration", "learning_rate", b"learning_rate", "lion", b"lion", "low_dimensional_packing_status", b"low_dimensional_packing_status", "mdl_adagrad_light", b"mdl_adagrad_light", "momentum", b"momentum", "multiply_weight_decay_factor_by_learning_rate", b"multiply_weight_decay_factor_by_learning_rate", "online_yogi", b"online_yogi", "parameters", b"parameters", "proximal_adagrad", b"proximal_adagrad", "proximal_yogi", b"proximal_yogi", "rms_prop", b"rms_prop", "simulated_quantization", b"simulated_quantization", "stochastic_gradient_descent", b"stochastic_gradient_descent", "user_defined_program", b"user_defined_program", "weight_decay_factor", b"weight_decay_factor"]) -> None: ... - def WhichOneof(self, oneof_group: typing.Literal["parameters", b"parameters"]) -> typing.Literal["adagrad", "adagrad_momentum", "bounded_adagrad", "stochastic_gradient_descent", "ftrl", "adam", "momentum", "lion", "rms_prop", "centered_rms_prop", "mdl_adagrad_light", "adadelta", "proximal_adagrad", "online_yogi", "proximal_yogi", "frequency_estimator", "user_defined_program", "assign"] | None: ... + def HasField( + self, + field_name: typing.Literal[ + "adadelta", + b"adadelta", + "adagrad", + b"adagrad", + "adagrad_momentum", + b"adagrad_momentum", + "adam", + b"adam", + "assign", + b"assign", + "bounded_adagrad", + b"bounded_adagrad", + "centered_rms_prop", + b"centered_rms_prop", + "clipping_limits", + b"clipping_limits", + "frequency_estimator", + b"frequency_estimator", + "ftrl", + b"ftrl", + "gradient_clipping_limits", + b"gradient_clipping_limits", + "hot_id_replication_configuration", + b"hot_id_replication_configuration", + "learning_rate", + b"learning_rate", + "lion", + b"lion", + "mdl_adagrad_light", + b"mdl_adagrad_light", + "momentum", + b"momentum", + "online_yogi", + b"online_yogi", + "parameters", + b"parameters", + "proximal_adagrad", + b"proximal_adagrad", + "proximal_yogi", + b"proximal_yogi", + "rms_prop", + b"rms_prop", + "simulated_quantization", + b"simulated_quantization", + "stochastic_gradient_descent", + b"stochastic_gradient_descent", + "user_defined_program", + b"user_defined_program", + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "adadelta", + b"adadelta", + "adagrad", + b"adagrad", + "adagrad_momentum", + b"adagrad_momentum", + "adam", + b"adam", + "assign", + b"assign", + "bounded_adagrad", + b"bounded_adagrad", + "centered_rms_prop", + b"centered_rms_prop", + "clipping_limits", + b"clipping_limits", + "frequency_estimator", + b"frequency_estimator", + "ftrl", + b"ftrl", + "gradient_accumulation_status", + b"gradient_accumulation_status", + "gradient_clipping_limits", + b"gradient_clipping_limits", + "hot_id_replication_configuration", + b"hot_id_replication_configuration", + "learning_rate", + b"learning_rate", + "lion", + b"lion", + "low_dimensional_packing_status", + b"low_dimensional_packing_status", + "mdl_adagrad_light", + b"mdl_adagrad_light", + "momentum", + b"momentum", + "multiply_weight_decay_factor_by_learning_rate", + b"multiply_weight_decay_factor_by_learning_rate", + "online_yogi", + b"online_yogi", + "parameters", + b"parameters", + "proximal_adagrad", + b"proximal_adagrad", + "proximal_yogi", + b"proximal_yogi", + "rms_prop", + b"rms_prop", + "simulated_quantization", + b"simulated_quantization", + "stochastic_gradient_descent", + b"stochastic_gradient_descent", + "user_defined_program", + b"user_defined_program", + "weight_decay_factor", + b"weight_decay_factor", + ], + ) -> None: ... + def WhichOneof( + self, oneof_group: typing.Literal["parameters", b"parameters"] + ) -> ( + typing.Literal[ + "adagrad", + "adagrad_momentum", + "bounded_adagrad", + "stochastic_gradient_descent", + "ftrl", + "adam", + "momentum", + "lion", + "rms_prop", + "centered_rms_prop", + "mdl_adagrad_light", + "adadelta", + "proximal_adagrad", + "online_yogi", + "proximal_yogi", + "frequency_estimator", + "user_defined_program", + "assign", + ] + | None + ): ... global___OptimizationParameters = OptimizationParameters @@ -1079,9 +1275,7 @@ class StateVariableSpecification(google.protobuf.message.Message): DESCRIPTOR: google.protobuf.descriptor.Descriptor - def __init__( - self, - ) -> None: ... + def __init__(self) -> None: ... @typing.final class FillWithConstant(google.protobuf.message.Message): @@ -1094,11 +1288,7 @@ class StateVariableSpecification(google.protobuf.message.Message): INITIAL_VALUE_FIELD_NUMBER: builtins.int initial_value: builtins.float - def __init__( - self, - *, - initial_value: builtins.float | None = ..., - ) -> None: ... + def __init__(self, *, initial_value: builtins.float | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["initial_value", b"initial_value"]) -> None: ... NAME_FIELD_NUMBER: builtins.int @@ -1117,8 +1307,20 @@ class StateVariableSpecification(google.protobuf.message.Message): user_defined: global___StateVariableSpecification.UserDefined | None = ..., fill_with_constant: global___StateVariableSpecification.FillWithConstant | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["fill_with_constant", b"fill_with_constant", "usage", b"usage", "user_defined", b"user_defined"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["fill_with_constant", b"fill_with_constant", "name", b"name", "usage", b"usage", "user_defined", b"user_defined"]) -> None: ... - def WhichOneof(self, oneof_group: typing.Literal["usage", b"usage"]) -> typing.Literal["user_defined", "fill_with_constant"] | None: ... + def HasField( + self, + field_name: typing.Literal[ + "fill_with_constant", b"fill_with_constant", "usage", b"usage", "user_defined", b"user_defined" + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "fill_with_constant", b"fill_with_constant", "name", b"name", "usage", b"usage", "user_defined", b"user_defined" + ], + ) -> None: ... + def WhichOneof( + self, oneof_group: typing.Literal["usage", b"usage"] + ) -> typing.Literal["user_defined", "fill_with_constant"] | None: ... global___StateVariableSpecification = StateVariableSpecification diff --git a/stubs/tensorflow/tensorflow/core/protobuf/tpu/topology_pb2.pyi b/stubs/tensorflow/tensorflow/core/protobuf/tpu/topology_pb2.pyi index d9d75f0b13da..574cd7a67e8e 100644 --- a/stubs/tensorflow/tensorflow/core/protobuf/tpu/topology_pb2.pyi +++ b/stubs/tensorflow/tensorflow/core/protobuf/tpu/topology_pb2.pyi @@ -30,7 +30,9 @@ class TPUHardwareFeature(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _EmbeddingFeatureEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[TPUHardwareFeature._EmbeddingFeature.ValueType], builtins.type): + class _EmbeddingFeatureEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[TPUHardwareFeature._EmbeddingFeature.ValueType], builtins.type + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor UNSUPPORTED: TPUHardwareFeature._EmbeddingFeature.ValueType # 0 """No embedding lookup accelerator available on the tpu.""" @@ -72,7 +74,12 @@ class TPUHardwareFeature(google.protobuf.message.Message): embedding_feature: global___TPUHardwareFeature.EmbeddingFeature.ValueType | None = ..., num_embedding_devices_per_chip: builtins.int | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["embedding_feature", b"embedding_feature", "num_embedding_devices_per_chip", b"num_embedding_devices_per_chip"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "embedding_feature", b"embedding_feature", "num_embedding_devices_per_chip", b"num_embedding_devices_per_chip" + ], + ) -> None: ... global___TPUHardwareFeature = TPUHardwareFeature @@ -123,6 +130,20 @@ class TopologyProto(google.protobuf.message.Message): tpu_hardware_feature: global___TPUHardwareFeature | None = ..., ) -> None: ... def HasField(self, field_name: typing.Literal["tpu_hardware_feature", b"tpu_hardware_feature"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["device_coordinates", b"device_coordinates", "mesh_shape", b"mesh_shape", "num_tasks", b"num_tasks", "num_tpu_devices_per_task", b"num_tpu_devices_per_task", "tpu_hardware_feature", b"tpu_hardware_feature"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "device_coordinates", + b"device_coordinates", + "mesh_shape", + b"mesh_shape", + "num_tasks", + b"num_tasks", + "num_tpu_devices_per_task", + b"num_tpu_devices_per_task", + "tpu_hardware_feature", + b"tpu_hardware_feature", + ], + ) -> None: ... global___TopologyProto = TopologyProto diff --git a/stubs/tensorflow/tensorflow/core/protobuf/tpu/tpu_embedding_configuration_pb2.pyi b/stubs/tensorflow/tensorflow/core/protobuf/tpu/tpu_embedding_configuration_pb2.pyi index c33925975268..32d67050f571 100644 --- a/stubs/tensorflow/tensorflow/core/protobuf/tpu/tpu_embedding_configuration_pb2.pyi +++ b/stubs/tensorflow/tensorflow/core/protobuf/tpu/tpu_embedding_configuration_pb2.pyi @@ -29,7 +29,9 @@ class TPUEmbeddingConfiguration(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _ModeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[TPUEmbeddingConfiguration._Mode.ValueType], builtins.type): + class _ModeEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[TPUEmbeddingConfiguration._Mode.ValueType], builtins.type + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor UNSPECIFIED: TPUEmbeddingConfiguration._Mode.ValueType # 0 INFERENCE: TPUEmbeddingConfiguration._Mode.ValueType # 1 @@ -50,7 +52,10 @@ class TPUEmbeddingConfiguration(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _ShardingStrategyEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[TPUEmbeddingConfiguration._ShardingStrategy.ValueType], builtins.type): + class _ShardingStrategyEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[TPUEmbeddingConfiguration._ShardingStrategy.ValueType], + builtins.type, + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor DIV_DEFAULT: TPUEmbeddingConfiguration._ShardingStrategy.ValueType # 0 MOD: TPUEmbeddingConfiguration._ShardingStrategy.ValueType # 1 @@ -107,8 +112,24 @@ class TPUEmbeddingConfiguration(google.protobuf.message.Message): num_features: builtins.int | None = ..., optimization_parameters: tensorflow.core.protobuf.tpu.optimization_parameters_pb2.OptimizationParameters | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["optimization_parameters", b"optimization_parameters"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["dimension", b"dimension", "name", b"name", "num_features", b"num_features", "optimization_parameters", b"optimization_parameters", "vocabulary_size", b"vocabulary_size"]) -> None: ... + def HasField( + self, field_name: typing.Literal["optimization_parameters", b"optimization_parameters"] + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "dimension", + b"dimension", + "name", + b"name", + "num_features", + b"num_features", + "optimization_parameters", + b"optimization_parameters", + "vocabulary_size", + b"vocabulary_size", + ], + ) -> None: ... @typing.final class FeatureDescriptor(google.protobuf.message.Message): @@ -141,7 +162,9 @@ class TPUEmbeddingConfiguration(google.protobuf.message.Message): table_id: builtins.int | None = ..., input_shape: collections.abc.Iterable[builtins.int] | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["input_shape", b"input_shape", "name", b"name", "table_id", b"table_id"]) -> None: ... + def ClearField( + self, field_name: typing.Literal["input_shape", b"input_shape", "name", b"name", "table_id", b"table_id"] + ) -> None: ... @typing.final class SpmdSharding(google.protobuf.message.Message): @@ -159,13 +182,10 @@ class TPUEmbeddingConfiguration(google.protobuf.message.Message): """Whether SPMD sharding is enabled.""" num_cores_per_replica: builtins.int """Number of cores per replica.""" - def __init__( - self, - *, - enabled: builtins.bool | None = ..., - num_cores_per_replica: builtins.int | None = ..., + def __init__(self, *, enabled: builtins.bool | None = ..., num_cores_per_replica: builtins.int | None = ...) -> None: ... + def ClearField( + self, field_name: typing.Literal["enabled", b"enabled", "num_cores_per_replica", b"num_cores_per_replica"] ) -> None: ... - def ClearField(self, field_name: typing.Literal["enabled", b"enabled", "num_cores_per_replica", b"num_cores_per_replica"]) -> None: ... TABLE_DESCRIPTOR_FIELD_NUMBER: builtins.int MODE_FIELD_NUMBER: builtins.int @@ -232,9 +252,17 @@ class TPUEmbeddingConfiguration(google.protobuf.message.Message): models to reuse embedding lookup statistics. """ @property - def table_descriptor(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___TPUEmbeddingConfiguration.TableDescriptor]: ... + def table_descriptor( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[ + global___TPUEmbeddingConfiguration.TableDescriptor + ]: ... @property - def feature_descriptor(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___TPUEmbeddingConfiguration.FeatureDescriptor]: + def feature_descriptor( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[ + global___TPUEmbeddingConfiguration.FeatureDescriptor + ]: """If the feature_descriptor field is populated, the model should NOT populate TableDescriptor.num_features and batch_size_per_tensor_core. These two fields will be auto-populated by the TPUEmbedding rewrite passes. @@ -257,7 +285,31 @@ class TPUEmbeddingConfiguration(google.protobuf.message.Message): spmd_sharding: global___TPUEmbeddingConfiguration.SpmdSharding | None = ..., ) -> None: ... def HasField(self, field_name: typing.Literal["spmd_sharding", b"spmd_sharding"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["batch_size_per_tensor_core", b"batch_size_per_tensor_core", "feature_descriptor", b"feature_descriptor", "mode", b"mode", "num_hosts", b"num_hosts", "num_tensor_cores", b"num_tensor_cores", "pipeline_execution_with_tensor_core", b"pipeline_execution_with_tensor_core", "profile_data_directory", b"profile_data_directory", "sharding_strategy", b"sharding_strategy", "spmd_sharding", b"spmd_sharding", "table_descriptor", b"table_descriptor"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "batch_size_per_tensor_core", + b"batch_size_per_tensor_core", + "feature_descriptor", + b"feature_descriptor", + "mode", + b"mode", + "num_hosts", + b"num_hosts", + "num_tensor_cores", + b"num_tensor_cores", + "pipeline_execution_with_tensor_core", + b"pipeline_execution_with_tensor_core", + "profile_data_directory", + b"profile_data_directory", + "sharding_strategy", + b"sharding_strategy", + "spmd_sharding", + b"spmd_sharding", + "table_descriptor", + b"table_descriptor", + ], + ) -> None: ... global___TPUEmbeddingConfiguration = TPUEmbeddingConfiguration @@ -269,8 +321,6 @@ class TPUEmbeddingError(google.protobuf.message.Message): DESCRIPTOR: google.protobuf.descriptor.Descriptor - def __init__( - self, - ) -> None: ... + def __init__(self) -> None: ... global___TPUEmbeddingError = TPUEmbeddingError diff --git a/stubs/tensorflow/tensorflow/core/protobuf/trackable_object_graph_pb2.pyi b/stubs/tensorflow/tensorflow/core/protobuf/trackable_object_graph_pb2.pyi index 63730b6360c6..f25a3bfda24f 100644 --- a/stubs/tensorflow/tensorflow/core/protobuf/trackable_object_graph_pb2.pyi +++ b/stubs/tensorflow/tensorflow/core/protobuf/trackable_object_graph_pb2.pyi @@ -39,12 +39,7 @@ class TrackableObjectGraph(google.protobuf.message.Message): """ local_name: builtins.str """A user-provided name for the edge.""" - def __init__( - self, - *, - node_id: builtins.int | None = ..., - local_name: builtins.str | None = ..., - ) -> None: ... + def __init__(self, *, node_id: builtins.int | None = ..., local_name: builtins.str | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["local_name", b"local_name", "node_id", b"node_id"]) -> None: ... @typing.final @@ -74,7 +69,9 @@ class TrackableObjectGraph(google.protobuf.message.Message): full_name: builtins.str | None = ..., checkpoint_key: builtins.str | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["checkpoint_key", b"checkpoint_key", "full_name", b"full_name", "name", b"name"]) -> None: ... + def ClearField( + self, field_name: typing.Literal["checkpoint_key", b"checkpoint_key", "full_name", b"full_name", "name", b"name"] + ) -> None: ... @typing.final class SlotVariableReference(google.protobuf.message.Message): @@ -100,7 +97,17 @@ class TrackableObjectGraph(google.protobuf.message.Message): slot_name: builtins.str | None = ..., slot_variable_node_id: builtins.int | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["original_variable_node_id", b"original_variable_node_id", "slot_name", b"slot_name", "slot_variable_node_id", b"slot_variable_node_id"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "original_variable_node_id", + b"original_variable_node_id", + "slot_name", + b"slot_name", + "slot_variable_node_id", + b"slot_variable_node_id", + ], + ) -> None: ... CHILDREN_FIELD_NUMBER: builtins.int ATTRIBUTES_FIELD_NUMBER: builtins.int @@ -108,15 +115,27 @@ class TrackableObjectGraph(google.protobuf.message.Message): REGISTERED_SAVER_FIELD_NUMBER: builtins.int HAS_CHECKPOINT_VALUES_FIELD_NUMBER: builtins.int @property - def children(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___TrackableObjectGraph.TrackableObject.ObjectReference]: + def children( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[ + global___TrackableObjectGraph.TrackableObject.ObjectReference + ]: """Objects which this object depends on.""" @property - def attributes(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___TrackableObjectGraph.TrackableObject.SerializedTensor]: + def attributes( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[ + global___TrackableObjectGraph.TrackableObject.SerializedTensor + ]: """Serialized data specific to this object.""" @property - def slot_variables(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___TrackableObjectGraph.TrackableObject.SlotVariableReference]: + def slot_variables( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[ + global___TrackableObjectGraph.TrackableObject.SlotVariableReference + ]: """Slot variables owned by this object.""" @property @@ -138,20 +157,41 @@ class TrackableObjectGraph(google.protobuf.message.Message): *, children: collections.abc.Iterable[global___TrackableObjectGraph.TrackableObject.ObjectReference] | None = ..., attributes: collections.abc.Iterable[global___TrackableObjectGraph.TrackableObject.SerializedTensor] | None = ..., - slot_variables: collections.abc.Iterable[global___TrackableObjectGraph.TrackableObject.SlotVariableReference] | None = ..., + slot_variables: ( + collections.abc.Iterable[global___TrackableObjectGraph.TrackableObject.SlotVariableReference] | None + ) = ..., registered_saver: global___RegisteredSaver | None = ..., has_checkpoint_values: google.protobuf.wrappers_pb2.BoolValue | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["has_checkpoint_values", b"has_checkpoint_values", "registered_saver", b"registered_saver"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["attributes", b"attributes", "children", b"children", "has_checkpoint_values", b"has_checkpoint_values", "registered_saver", b"registered_saver", "slot_variables", b"slot_variables"]) -> None: ... + def HasField( + self, + field_name: typing.Literal[ + "has_checkpoint_values", b"has_checkpoint_values", "registered_saver", b"registered_saver" + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "attributes", + b"attributes", + "children", + b"children", + "has_checkpoint_values", + b"has_checkpoint_values", + "registered_saver", + b"registered_saver", + "slot_variables", + b"slot_variables", + ], + ) -> None: ... NODES_FIELD_NUMBER: builtins.int @property - def nodes(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___TrackableObjectGraph.TrackableObject]: ... - def __init__( + def nodes( self, - *, - nodes: collections.abc.Iterable[global___TrackableObjectGraph.TrackableObject] | None = ..., + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___TrackableObjectGraph.TrackableObject]: ... + def __init__( + self, *, nodes: collections.abc.Iterable[global___TrackableObjectGraph.TrackableObject] | None = ... ) -> None: ... def ClearField(self, field_name: typing.Literal["nodes", b"nodes"]) -> None: ... @@ -167,12 +207,7 @@ class RegisteredSaver(google.protobuf.message.Message): """The name of the registered saver/restore function.""" object_name: builtins.str """Unique auto-generated name of the object.""" - def __init__( - self, - *, - name: builtins.str | None = ..., - object_name: builtins.str | None = ..., - ) -> None: ... + def __init__(self, *, name: builtins.str | None = ..., object_name: builtins.str | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["name", b"name", "object_name", b"object_name"]) -> None: ... global___RegisteredSaver = RegisteredSaver diff --git a/stubs/tensorflow/tensorflow/core/protobuf/transport_options_pb2.pyi b/stubs/tensorflow/tensorflow/core/protobuf/transport_options_pb2.pyi index 26d7b1406dae..38596e5e8783 100644 --- a/stubs/tensorflow/tensorflow/core/protobuf/transport_options_pb2.pyi +++ b/stubs/tensorflow/tensorflow/core/protobuf/transport_options_pb2.pyi @@ -22,11 +22,7 @@ class RecvBufRespExtra(google.protobuf.message.Message): 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 __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 diff --git a/stubs/tensorflow/tensorflow/core/protobuf/verifier_config_pb2.pyi b/stubs/tensorflow/tensorflow/core/protobuf/verifier_config_pb2.pyi index 405ab99a2dd5..c65f7c9cdb24 100644 --- a/stubs/tensorflow/tensorflow/core/protobuf/verifier_config_pb2.pyi +++ b/stubs/tensorflow/tensorflow/core/protobuf/verifier_config_pb2.pyi @@ -28,7 +28,9 @@ class VerifierConfig(google.protobuf.message.Message): 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): + 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 @@ -53,6 +55,11 @@ class VerifierConfig(google.protobuf.message.Message): 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: ... + 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 diff --git a/stubs/tensorflow/tensorflow/core/util/event_pb2.pyi b/stubs/tensorflow/tensorflow/core/util/event_pb2.pyi index 9d6a986624bc..7a8ac4afd856 100644 --- a/stubs/tensorflow/tensorflow/core/util/event_pb2.pyi +++ b/stubs/tensorflow/tensorflow/core/util/event_pb2.pyi @@ -25,7 +25,9 @@ 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): +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.""" @@ -53,7 +55,9 @@ 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): +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 @@ -142,9 +146,64 @@ class Event(google.protobuf.message.Message): 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: ... + 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 @@ -159,11 +218,7 @@ class SourceMetadata(google.protobuf.message.Message): """Low level name of the summary writer, such as `tensorflow.core.util.events_writer`. """ - def __init__( - self, - *, - writer: builtins.str | None = ..., - ) -> None: ... + def __init__(self, *, writer: builtins.str | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["writer", b"writer"]) -> None: ... global___SourceMetadata = SourceMetadata @@ -182,7 +237,9 @@ class LogMessage(google.protobuf.message.Message): 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): + 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 @@ -214,10 +271,7 @@ class LogMessage(google.protobuf.message.Message): level: global___LogMessage.Level.ValueType message: builtins.str def __init__( - self, - *, - level: global___LogMessage.Level.ValueType | None = ..., - message: builtins.str | None = ..., + 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: ... @@ -233,7 +287,9 @@ class SessionLog(google.protobuf.message.Message): 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): + 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 @@ -260,7 +316,9 @@ class SessionLog(google.protobuf.message.Message): 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: ... + def ClearField( + self, field_name: typing.Literal["checkpoint_path", b"checkpoint_path", "msg", b"msg", "status", b"status"] + ) -> None: ... global___SessionLog = SessionLog @@ -278,12 +336,7 @@ class TaggedRunMetadata(google.protobuf.message.Message): """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 __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 @@ -294,11 +347,7 @@ class WatchdogConfig(google.protobuf.message.Message): TIMEOUT_MS_FIELD_NUMBER: builtins.int timeout_ms: builtins.int - def __init__( - self, - *, - timeout_ms: builtins.int | None = ..., - ) -> None: ... + 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 @@ -309,11 +358,7 @@ class RequestedExitCode(google.protobuf.message.Message): EXIT_CODE_FIELD_NUMBER: builtins.int exit_code: builtins.int - def __init__( - self, - *, - exit_code: builtins.int | None = ..., - ) -> None: ... + 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 @@ -337,8 +382,15 @@ class WorkerHeartbeatRequest(google.protobuf.message.Message): 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: ... + 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 @@ -360,6 +412,8 @@ class WorkerHeartbeatResponse(google.protobuf.message.Message): 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: ... + def ClearField( + self, field_name: typing.Literal["health_status", b"health_status", "hostname", b"hostname", "worker_log", b"worker_log"] + ) -> None: ... global___WorkerHeartbeatResponse = WorkerHeartbeatResponse diff --git a/stubs/tensorflow/tensorflow/core/util/memmapped_file_system_pb2.pyi b/stubs/tensorflow/tensorflow/core/util/memmapped_file_system_pb2.pyi index 266b0ed6e5d3..c3b2b662d9d4 100644 --- a/stubs/tensorflow/tensorflow/core/util/memmapped_file_system_pb2.pyi +++ b/stubs/tensorflow/tensorflow/core/util/memmapped_file_system_pb2.pyi @@ -40,11 +40,7 @@ class MemmappedFileSystemDirectoryElement(google.protobuf.message.Message): name: builtins.str length: builtins.int def __init__( - self, - *, - offset: builtins.int | None = ..., - name: builtins.str | None = ..., - length: builtins.int | None = ..., + 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: ... @@ -58,11 +54,11 @@ class MemmappedFileSystemDirectory(google.protobuf.message.Message): ELEMENT_FIELD_NUMBER: builtins.int @property - def element(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___MemmappedFileSystemDirectoryElement]: ... - def __init__( + def element( self, - *, - element: collections.abc.Iterable[global___MemmappedFileSystemDirectoryElement] | None = ..., + ) -> 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: ... diff --git a/stubs/tensorflow/tensorflow/core/util/saved_tensor_slice_pb2.pyi b/stubs/tensorflow/tensorflow/core/util/saved_tensor_slice_pb2.pyi index 9b492465ac4f..03a7ef0e2e94 100644 --- a/stubs/tensorflow/tensorflow/core/util/saved_tensor_slice_pb2.pyi +++ b/stubs/tensorflow/tensorflow/core/util/saved_tensor_slice_pb2.pyi @@ -52,7 +52,11 @@ class SavedSliceMeta(google.protobuf.message.Message): """Shape of the tensor""" @property - def slice(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[tensorflow.core.framework.tensor_slice_pb2.TensorSliceProto]: + def slice( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[ + tensorflow.core.framework.tensor_slice_pb2.TensorSliceProto + ]: """Explicit list of slices saved in the checkpoint file.""" def __init__( @@ -64,7 +68,9 @@ class SavedSliceMeta(google.protobuf.message.Message): slice: collections.abc.Iterable[tensorflow.core.framework.tensor_slice_pb2.TensorSliceProto] | None = ..., ) -> None: ... def HasField(self, field_name: typing.Literal["shape", b"shape"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["name", b"name", "shape", b"shape", "slice", b"slice", "type", b"type"]) -> None: ... + def ClearField( + self, field_name: typing.Literal["name", b"name", "shape", b"shape", "slice", b"slice", "type", b"type"] + ) -> None: ... global___SavedSliceMeta = SavedSliceMeta @@ -158,12 +164,7 @@ class SavedTensorSlices(google.protobuf.message.Message): def data(self) -> global___SavedSlice: """This exists in all but the first item of each checkpoint file.""" - def __init__( - self, - *, - meta: global___SavedTensorSliceMeta | None = ..., - data: global___SavedSlice | None = ..., - ) -> None: ... + def __init__(self, *, meta: global___SavedTensorSliceMeta | None = ..., data: global___SavedSlice | None = ...) -> None: ... def HasField(self, field_name: typing.Literal["data", b"data", "meta", b"meta"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["data", b"data", "meta", b"meta"]) -> None: ... diff --git a/stubs/tensorflow/tensorflow/python/keras/protobuf/projector_config_pb2.pyi b/stubs/tensorflow/tensorflow/python/keras/protobuf/projector_config_pb2.pyi index 22be35d7d6bc..9141272b1c52 100644 --- a/stubs/tensorflow/tensorflow/python/keras/protobuf/projector_config_pb2.pyi +++ b/stubs/tensorflow/tensorflow/python/keras/protobuf/projector_config_pb2.pyi @@ -28,12 +28,11 @@ class SpriteMetadata(google.protobuf.message.Message): """[width, height] of a single image in the sprite.""" def __init__( - self, - *, - image_path: builtins.str | None = ..., - single_image_dim: collections.abc.Iterable[builtins.int] | None = ..., + self, *, image_path: builtins.str | None = ..., single_image_dim: collections.abc.Iterable[builtins.int] | None = ... + ) -> None: ... + def ClearField( + self, field_name: typing.Literal["image_path", b"image_path", "single_image_dim", b"single_image_dim"] ) -> None: ... - def ClearField(self, field_name: typing.Literal["image_path", b"image_path", "single_image_dim", b"single_image_dim"]) -> None: ... global___SpriteMetadata = SpriteMetadata @@ -73,7 +72,23 @@ class EmbeddingInfo(google.protobuf.message.Message): tensor_path: builtins.str | None = ..., ) -> None: ... def HasField(self, field_name: typing.Literal["sprite", b"sprite"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["bookmarks_path", b"bookmarks_path", "metadata_path", b"metadata_path", "sprite", b"sprite", "tensor_name", b"tensor_name", "tensor_path", b"tensor_path", "tensor_shape", b"tensor_shape"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "bookmarks_path", + b"bookmarks_path", + "metadata_path", + b"metadata_path", + "sprite", + b"sprite", + "tensor_name", + b"tensor_name", + "tensor_path", + b"tensor_path", + "tensor_shape", + b"tensor_shape", + ], + ) -> None: ... global___EmbeddingInfo = EmbeddingInfo @@ -99,6 +114,16 @@ class ProjectorConfig(google.protobuf.message.Message): embeddings: collections.abc.Iterable[global___EmbeddingInfo] | None = ..., model_checkpoint_dir: builtins.str | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["embeddings", b"embeddings", "model_checkpoint_dir", b"model_checkpoint_dir", "model_checkpoint_path", b"model_checkpoint_path"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "embeddings", + b"embeddings", + "model_checkpoint_dir", + b"model_checkpoint_dir", + "model_checkpoint_path", + b"model_checkpoint_path", + ], + ) -> None: ... global___ProjectorConfig = ProjectorConfig diff --git a/stubs/tensorflow/tensorflow/python/keras/protobuf/saved_metadata_pb2.pyi b/stubs/tensorflow/tensorflow/python/keras/protobuf/saved_metadata_pb2.pyi index 0133517d273f..e4a13c9db33a 100644 --- a/stubs/tensorflow/tensorflow/python/keras/protobuf/saved_metadata_pb2.pyi +++ b/stubs/tensorflow/tensorflow/python/keras/protobuf/saved_metadata_pb2.pyi @@ -25,11 +25,7 @@ class SavedMetadata(google.protobuf.message.Message): Keras object is stored. """ - def __init__( - self, - *, - nodes: collections.abc.Iterable[global___SavedObject] | None = ..., - ) -> None: ... + def __init__(self, *, nodes: collections.abc.Iterable[global___SavedObject] | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["nodes", b"nodes"]) -> None: ... global___SavedMetadata = SavedMetadata @@ -74,6 +70,20 @@ class SavedObject(google.protobuf.message.Message): version: tensorflow.python.keras.protobuf.versions_pb2.VersionDef | None = ..., ) -> None: ... def HasField(self, field_name: typing.Literal["version", b"version"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["identifier", b"identifier", "metadata", b"metadata", "node_id", b"node_id", "node_path", b"node_path", "version", b"version"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "identifier", + b"identifier", + "metadata", + b"metadata", + "node_id", + b"node_id", + "node_path", + b"node_path", + "version", + b"version", + ], + ) -> None: ... global___SavedObject = SavedObject diff --git a/stubs/tensorflow/tensorflow/python/keras/protobuf/versions_pb2.pyi b/stubs/tensorflow/tensorflow/python/keras/protobuf/versions_pb2.pyi index 55d2e725bc0d..10936e5157fd 100644 --- a/stubs/tensorflow/tensorflow/python/keras/protobuf/versions_pb2.pyi +++ b/stubs/tensorflow/tensorflow/python/keras/protobuf/versions_pb2.pyi @@ -55,6 +55,9 @@ class VersionDef(google.protobuf.message.Message): min_consumer: builtins.int | None = ..., bad_consumers: collections.abc.Iterable[builtins.int] | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["bad_consumers", b"bad_consumers", "min_consumer", b"min_consumer", "producer", b"producer"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal["bad_consumers", b"bad_consumers", "min_consumer", b"min_consumer", "producer", b"producer"], + ) -> None: ... global___VersionDef = VersionDef diff --git a/stubs/tensorflow/tensorflow/tsl/protobuf/coordination_config_pb2.pyi b/stubs/tensorflow/tensorflow/tsl/protobuf/coordination_config_pb2.pyi index a2eb29dda885..f784e1621046 100644 --- a/stubs/tensorflow/tensorflow/tsl/protobuf/coordination_config_pb2.pyi +++ b/stubs/tensorflow/tensorflow/tsl/protobuf/coordination_config_pb2.pyi @@ -25,12 +25,7 @@ class CoordinatedJob(google.protobuf.message.Message): NUM_TASKS_FIELD_NUMBER: builtins.int name: builtins.str num_tasks: builtins.int - def __init__( - self, - *, - name: builtins.str | None = ..., - num_tasks: builtins.int | None = ..., - ) -> None: ... + def __init__(self, *, name: builtins.str | None = ..., num_tasks: builtins.int | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["name", b"name", "num_tasks", b"num_tasks"]) -> None: ... global___CoordinatedJob = CoordinatedJob @@ -101,7 +96,9 @@ class CoordinationServiceConfig(google.protobuf.message.Message): propagation mechanism. """ @property - def coordinated_job_list(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___CoordinatedJob]: ... + def coordinated_job_list( + self, + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___CoordinatedJob]: ... @property def recoverable_jobs(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]: """The list of jobs which are recoverable. If a task in this list fails, @@ -126,6 +123,34 @@ class CoordinationServiceConfig(google.protobuf.message.Message): force_disable: builtins.bool | None = ..., poll_for_error_from_service_at_startup: builtins.bool | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["agent_destruction_without_shutdown", b"agent_destruction_without_shutdown", "allow_new_incarnation_to_reconnect", b"allow_new_incarnation_to_reconnect", "cluster_register_timeout_in_ms", b"cluster_register_timeout_in_ms", "coordinated_job_list", b"coordinated_job_list", "enable_health_check", b"enable_health_check", "force_disable", b"force_disable", "heartbeat_timeout_in_ms", b"heartbeat_timeout_in_ms", "poll_for_error_from_service_at_startup", b"poll_for_error_from_service_at_startup", "recoverable_jobs", b"recoverable_jobs", "service_leader", b"service_leader", "service_type", b"service_type", "shutdown_barrier_timeout_in_ms", b"shutdown_barrier_timeout_in_ms"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "agent_destruction_without_shutdown", + b"agent_destruction_without_shutdown", + "allow_new_incarnation_to_reconnect", + b"allow_new_incarnation_to_reconnect", + "cluster_register_timeout_in_ms", + b"cluster_register_timeout_in_ms", + "coordinated_job_list", + b"coordinated_job_list", + "enable_health_check", + b"enable_health_check", + "force_disable", + b"force_disable", + "heartbeat_timeout_in_ms", + b"heartbeat_timeout_in_ms", + "poll_for_error_from_service_at_startup", + b"poll_for_error_from_service_at_startup", + "recoverable_jobs", + b"recoverable_jobs", + "service_leader", + b"service_leader", + "service_type", + b"service_type", + "shutdown_barrier_timeout_in_ms", + b"shutdown_barrier_timeout_in_ms", + ], + ) -> None: ... global___CoordinationServiceConfig = CoordinationServiceConfig diff --git a/stubs/tensorflow/tensorflow/tsl/protobuf/coordination_service_pb2.pyi b/stubs/tensorflow/tensorflow/tsl/protobuf/coordination_service_pb2.pyi index 82468c997f24..27721cfbefe4 100644 --- a/stubs/tensorflow/tensorflow/tsl/protobuf/coordination_service_pb2.pyi +++ b/stubs/tensorflow/tensorflow/tsl/protobuf/coordination_service_pb2.pyi @@ -25,7 +25,9 @@ class _CoordinatedTaskState: ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType -class _CoordinatedTaskStateEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_CoordinatedTaskState.ValueType], builtins.type): +class _CoordinatedTaskStateEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_CoordinatedTaskState.ValueType], builtins.type +): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor TASKSTATE_UNSPECIFIED: _CoordinatedTaskState.ValueType # 0 """TASKSTATE_UNSPECIFIED is an invalid state such that indicates a bug.""" @@ -63,12 +65,7 @@ class CoordinatedTask(google.protobuf.message.Message): TASK_ID_FIELD_NUMBER: builtins.int job_name: builtins.str task_id: builtins.int - def __init__( - self, - *, - job_name: builtins.str | None = ..., - task_id: builtins.int | None = ..., - ) -> None: ... + def __init__(self, *, job_name: builtins.str | None = ..., task_id: builtins.int | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["job_name", b"job_name", "task_id", b"task_id"]) -> None: ... global___CoordinatedTask = CoordinatedTask @@ -95,13 +92,12 @@ class CoordinationServiceError(google.protobuf.message.Message): """ def __init__( - self, - *, - is_reported_error: builtins.bool | None = ..., - source_task: global___CoordinatedTask | None = ..., + self, *, is_reported_error: builtins.bool | None = ..., source_task: global___CoordinatedTask | None = ... ) -> None: ... def HasField(self, field_name: typing.Literal["source_task", b"source_task"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["is_reported_error", b"is_reported_error", "source_task", b"source_task"]) -> None: ... + def ClearField( + self, field_name: typing.Literal["is_reported_error", b"is_reported_error", "source_task", b"source_task"] + ) -> None: ... global___CoordinationServiceError = CoordinationServiceError @@ -131,7 +127,21 @@ class CoordinatedTaskStateInfo(google.protobuf.message.Message): error_payload: global___CoordinationServiceError | None = ..., ) -> None: ... def HasField(self, field_name: typing.Literal["error_payload", b"error_payload", "task", b"task"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["error_code", b"error_code", "error_message", b"error_message", "error_payload", b"error_payload", "state", b"state", "task", b"task"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "error_code", + b"error_code", + "error_message", + b"error_message", + "error_payload", + b"error_payload", + "state", + b"state", + "task", + b"task", + ], + ) -> None: ... global___CoordinatedTaskStateInfo = CoordinatedTaskStateInfo @@ -144,11 +154,7 @@ class DeviceInfo(google.protobuf.message.Message): DEVICE_FIELD_NUMBER: builtins.int @property def device(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[google.protobuf.any_pb2.Any]: ... - def __init__( - self, - *, - device: collections.abc.Iterable[google.protobuf.any_pb2.Any] | None = ..., - ) -> None: ... + def __init__(self, *, device: collections.abc.Iterable[google.protobuf.any_pb2.Any] | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["device", b"device"]) -> None: ... global___DeviceInfo = DeviceInfo @@ -168,12 +174,7 @@ class RegisterTaskRequest(google.protobuf.message.Message): incarnation: builtins.int @property def source_task(self) -> global___CoordinatedTask: ... - def __init__( - self, - *, - incarnation: builtins.int | None = ..., - source_task: global___CoordinatedTask | None = ..., - ) -> None: ... + def __init__(self, *, incarnation: builtins.int | None = ..., source_task: global___CoordinatedTask | None = ...) -> None: ... def HasField(self, field_name: typing.Literal["source_task", b"source_task"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["incarnation", b"incarnation", "source_task", b"source_task"]) -> None: ... @@ -185,11 +186,7 @@ class RegisterTaskResponse(google.protobuf.message.Message): LEADER_INCARNATION_FIELD_NUMBER: builtins.int leader_incarnation: builtins.int - def __init__( - self, - *, - leader_incarnation: builtins.int | None = ..., - ) -> None: ... + def __init__(self, *, leader_incarnation: builtins.int | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["leader_incarnation", b"leader_incarnation"]) -> None: ... global___RegisterTaskResponse = RegisterTaskResponse @@ -205,12 +202,7 @@ class HeartbeatRequest(google.protobuf.message.Message): incarnation: builtins.int @property def source_task(self) -> global___CoordinatedTask: ... - def __init__( - self, - *, - incarnation: builtins.int | None = ..., - source_task: global___CoordinatedTask | None = ..., - ) -> None: ... + def __init__(self, *, incarnation: builtins.int | None = ..., source_task: global___CoordinatedTask | None = ...) -> None: ... def HasField(self, field_name: typing.Literal["source_task", b"source_task"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["incarnation", b"incarnation", "source_task", b"source_task"]) -> None: ... @@ -225,11 +217,7 @@ class HeartbeatResponse(google.protobuf.message.Message): """If there are failures in cluster, use additional metadata in response to broadcast error code and message to other tasks. """ - def __init__( - self, - *, - leader_incarnation: builtins.int | None = ..., - ) -> None: ... + def __init__(self, *, leader_incarnation: builtins.int | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["leader_incarnation", b"leader_incarnation"]) -> None: ... global___HeartbeatResponse = HeartbeatResponse @@ -241,11 +229,7 @@ class PollForErrorRequest(google.protobuf.message.Message): SOURCE_TASK_FIELD_NUMBER: builtins.int @property def source_task(self) -> global___CoordinatedTask: ... - def __init__( - self, - *, - source_task: global___CoordinatedTask | None = ..., - ) -> None: ... + def __init__(self, *, source_task: global___CoordinatedTask | None = ...) -> None: ... def HasField(self, field_name: typing.Literal["source_task", b"source_task"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["source_task", b"source_task"]) -> None: ... @@ -255,9 +239,7 @@ global___PollForErrorRequest = PollForErrorRequest class PollForErrorResponse(google.protobuf.message.Message): DESCRIPTOR: google.protobuf.descriptor.Descriptor - def __init__( - self, - ) -> None: ... + def __init__(self) -> None: ... global___PollForErrorResponse = PollForErrorResponse @@ -276,12 +258,11 @@ class WaitForAllTasksRequest(google.protobuf.message.Message): """All local device attributes on the request sender;""" def __init__( - self, - *, - source_task: global___CoordinatedTask | None = ..., - device_info: global___DeviceInfo | None = ..., + self, *, source_task: global___CoordinatedTask | None = ..., device_info: global___DeviceInfo | None = ... ) -> None: ... - def HasField(self, field_name: typing.Literal["device_info", b"device_info", "source_task", b"source_task"]) -> builtins.bool: ... + def HasField( + self, field_name: typing.Literal["device_info", b"device_info", "source_task", b"source_task"] + ) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["device_info", b"device_info", "source_task", b"source_task"]) -> None: ... global___WaitForAllTasksRequest = WaitForAllTasksRequest @@ -298,13 +279,12 @@ class WaitForAllTasksResponse(google.protobuf.message.Message): """All devices in the cluster.""" def __init__( - self, - *, - leader_incarnation: builtins.int | None = ..., - device_info: global___DeviceInfo | None = ..., + self, *, leader_incarnation: builtins.int | None = ..., device_info: global___DeviceInfo | None = ... ) -> None: ... def HasField(self, field_name: typing.Literal["device_info", b"device_info"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["device_info", b"device_info", "leader_incarnation", b"leader_incarnation"]) -> None: ... + def ClearField( + self, field_name: typing.Literal["device_info", b"device_info", "leader_incarnation", b"leader_incarnation"] + ) -> None: ... global___WaitForAllTasksResponse = WaitForAllTasksResponse @@ -317,11 +297,7 @@ class ShutdownTaskRequest(google.protobuf.message.Message): SOURCE_TASK_FIELD_NUMBER: builtins.int @property def source_task(self) -> global___CoordinatedTask: ... - def __init__( - self, - *, - source_task: global___CoordinatedTask | None = ..., - ) -> None: ... + def __init__(self, *, source_task: global___CoordinatedTask | None = ...) -> None: ... def HasField(self, field_name: typing.Literal["source_task", b"source_task"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["source_task", b"source_task"]) -> None: ... @@ -331,9 +307,7 @@ global___ShutdownTaskRequest = ShutdownTaskRequest class ShutdownTaskResponse(google.protobuf.message.Message): DESCRIPTOR: google.protobuf.descriptor.Descriptor - def __init__( - self, - ) -> None: ... + def __init__(self) -> None: ... global___ShutdownTaskResponse = ShutdownTaskResponse @@ -346,11 +320,7 @@ class ResetTaskRequest(google.protobuf.message.Message): SOURCE_TASK_FIELD_NUMBER: builtins.int @property def source_task(self) -> global___CoordinatedTask: ... - def __init__( - self, - *, - source_task: global___CoordinatedTask | None = ..., - ) -> None: ... + def __init__(self, *, source_task: global___CoordinatedTask | None = ...) -> None: ... def HasField(self, field_name: typing.Literal["source_task", b"source_task"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["source_task", b"source_task"]) -> None: ... @@ -360,9 +330,7 @@ global___ResetTaskRequest = ResetTaskRequest class ResetTaskResponse(google.protobuf.message.Message): DESCRIPTOR: google.protobuf.descriptor.Descriptor - def __init__( - self, - ) -> None: ... + def __init__(self) -> None: ... global___ResetTaskResponse = ResetTaskResponse @@ -387,7 +355,12 @@ class ReportErrorToTaskRequest(google.protobuf.message.Message): error_payload: global___CoordinationServiceError | None = ..., ) -> None: ... def HasField(self, field_name: typing.Literal["error_payload", b"error_payload"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["error_code", b"error_code", "error_message", b"error_message", "error_payload", b"error_payload"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "error_code", b"error_code", "error_message", b"error_message", "error_payload", b"error_payload" + ], + ) -> None: ... global___ReportErrorToTaskRequest = ReportErrorToTaskRequest @@ -395,9 +368,7 @@ global___ReportErrorToTaskRequest = ReportErrorToTaskRequest class ReportErrorToTaskResponse(google.protobuf.message.Message): DESCRIPTOR: google.protobuf.descriptor.Descriptor - def __init__( - self, - ) -> None: ... + def __init__(self) -> None: ... global___ReportErrorToTaskResponse = ReportErrorToTaskResponse @@ -422,7 +393,12 @@ class ReportErrorToServiceRequest(google.protobuf.message.Message): error_origin: global___CoordinatedTask | None = ..., ) -> None: ... def HasField(self, field_name: typing.Literal["error_origin", b"error_origin"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["error_code", b"error_code", "error_message", b"error_message", "error_origin", b"error_origin"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "error_code", b"error_code", "error_message", b"error_message", "error_origin", b"error_origin" + ], + ) -> None: ... global___ReportErrorToServiceRequest = ReportErrorToServiceRequest @@ -430,9 +406,7 @@ global___ReportErrorToServiceRequest = ReportErrorToServiceRequest class ReportErrorToServiceResponse(google.protobuf.message.Message): DESCRIPTOR: google.protobuf.descriptor.Descriptor - def __init__( - self, - ) -> None: ... + def __init__(self) -> None: ... global___ReportErrorToServiceResponse = ReportErrorToServiceResponse @@ -445,11 +419,7 @@ class GetTaskStateRequest(google.protobuf.message.Message): SOURCE_TASK_FIELD_NUMBER: builtins.int @property def source_task(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___CoordinatedTask]: ... - def __init__( - self, - *, - source_task: collections.abc.Iterable[global___CoordinatedTask] | None = ..., - ) -> None: ... + def __init__(self, *, source_task: collections.abc.Iterable[global___CoordinatedTask] | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["source_task", b"source_task"]) -> None: ... global___GetTaskStateRequest = GetTaskStateRequest @@ -460,12 +430,10 @@ class GetTaskStateResponse(google.protobuf.message.Message): TASK_STATE_FIELD_NUMBER: builtins.int @property - def task_state(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___CoordinatedTaskStateInfo]: ... - def __init__( + def task_state( self, - *, - task_state: collections.abc.Iterable[global___CoordinatedTaskStateInfo] | None = ..., - ) -> None: ... + ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___CoordinatedTaskStateInfo]: ... + def __init__(self, *, task_state: collections.abc.Iterable[global___CoordinatedTaskStateInfo] | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["task_state", b"task_state"]) -> None: ... global___GetTaskStateResponse = GetTaskStateResponse @@ -483,12 +451,7 @@ class KeyValueEntry(google.protobuf.message.Message): VALUE_FIELD_NUMBER: builtins.int key: builtins.str value: builtins.bytes - def __init__( - self, - *, - key: builtins.str | None = ..., - value: builtins.bytes | None = ..., - ) -> None: ... + def __init__(self, *, key: builtins.str | None = ..., value: builtins.bytes | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["key", b"key", "value", b"value"]) -> None: ... global___KeyValueEntry = KeyValueEntry @@ -504,12 +467,7 @@ class InsertKeyValueRequest(google.protobuf.message.Message): allow_overwrite: builtins.bool @property def kv(self) -> global___KeyValueEntry: ... - def __init__( - self, - *, - kv: global___KeyValueEntry | None = ..., - allow_overwrite: builtins.bool | None = ..., - ) -> None: ... + def __init__(self, *, kv: global___KeyValueEntry | None = ..., allow_overwrite: builtins.bool | None = ...) -> None: ... def HasField(self, field_name: typing.Literal["kv", b"kv"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["allow_overwrite", b"allow_overwrite", "kv", b"kv"]) -> None: ... @@ -519,9 +477,7 @@ global___InsertKeyValueRequest = InsertKeyValueRequest class InsertKeyValueResponse(google.protobuf.message.Message): DESCRIPTOR: google.protobuf.descriptor.Descriptor - def __init__( - self, - ) -> None: ... + def __init__(self) -> None: ... global___InsertKeyValueResponse = InsertKeyValueResponse @@ -533,11 +489,7 @@ class GetKeyValueRequest(google.protobuf.message.Message): KEY_FIELD_NUMBER: builtins.int key: builtins.str - def __init__( - self, - *, - key: builtins.str | None = ..., - ) -> None: ... + def __init__(self, *, key: builtins.str | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["key", b"key"]) -> None: ... global___GetKeyValueRequest = GetKeyValueRequest @@ -549,11 +501,7 @@ class GetKeyValueResponse(google.protobuf.message.Message): KV_FIELD_NUMBER: builtins.int @property def kv(self) -> global___KeyValueEntry: ... - def __init__( - self, - *, - kv: global___KeyValueEntry | None = ..., - ) -> None: ... + def __init__(self, *, kv: global___KeyValueEntry | None = ...) -> None: ... def HasField(self, field_name: typing.Literal["kv", b"kv"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["kv", b"kv"]) -> None: ... @@ -565,11 +513,7 @@ class TryGetKeyValueRequest(google.protobuf.message.Message): KEY_FIELD_NUMBER: builtins.int key: builtins.str - def __init__( - self, - *, - key: builtins.str | None = ..., - ) -> None: ... + def __init__(self, *, key: builtins.str | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["key", b"key"]) -> None: ... global___TryGetKeyValueRequest = TryGetKeyValueRequest @@ -581,11 +525,7 @@ class TryGetKeyValueResponse(google.protobuf.message.Message): KV_FIELD_NUMBER: builtins.int @property def kv(self) -> global___KeyValueEntry: ... - def __init__( - self, - *, - kv: global___KeyValueEntry | None = ..., - ) -> None: ... + def __init__(self, *, kv: global___KeyValueEntry | None = ...) -> None: ... def HasField(self, field_name: typing.Literal["kv", b"kv"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["kv", b"kv"]) -> None: ... @@ -597,11 +537,7 @@ class GetKeyValueDirRequest(google.protobuf.message.Message): DIRECTORY_KEY_FIELD_NUMBER: builtins.int directory_key: builtins.str - def __init__( - self, - *, - directory_key: builtins.str | None = ..., - ) -> None: ... + def __init__(self, *, directory_key: builtins.str | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["directory_key", b"directory_key"]) -> None: ... global___GetKeyValueDirRequest = GetKeyValueDirRequest @@ -616,10 +552,7 @@ class GetKeyValueDirResponse(google.protobuf.message.Message): @property def kv(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___KeyValueEntry]: ... def __init__( - self, - *, - directory_key: builtins.str | None = ..., - kv: collections.abc.Iterable[global___KeyValueEntry] | None = ..., + self, *, directory_key: builtins.str | None = ..., kv: collections.abc.Iterable[global___KeyValueEntry] | None = ... ) -> None: ... def ClearField(self, field_name: typing.Literal["directory_key", b"directory_key", "kv", b"kv"]) -> None: ... @@ -637,12 +570,7 @@ class DeleteKeyValueRequest(google.protobuf.message.Message): IS_DIRECTORY_FIELD_NUMBER: builtins.int key: builtins.str is_directory: builtins.bool - def __init__( - self, - *, - key: builtins.str | None = ..., - is_directory: builtins.bool | None = ..., - ) -> None: ... + def __init__(self, *, key: builtins.str | None = ..., is_directory: builtins.bool | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["is_directory", b"is_directory", "key", b"key"]) -> None: ... global___DeleteKeyValueRequest = DeleteKeyValueRequest @@ -651,9 +579,7 @@ global___DeleteKeyValueRequest = DeleteKeyValueRequest class DeleteKeyValueResponse(google.protobuf.message.Message): DESCRIPTOR: google.protobuf.descriptor.Descriptor - def __init__( - self, - ) -> None: ... + def __init__(self) -> None: ... global___DeleteKeyValueResponse = DeleteKeyValueResponse @@ -688,7 +614,19 @@ class BarrierRequest(google.protobuf.message.Message): source_task: global___CoordinatedTask | None = ..., ) -> None: ... def HasField(self, field_name: typing.Literal["source_task", b"source_task"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["barrier_id", b"barrier_id", "barrier_timeout_in_ms", b"barrier_timeout_in_ms", "source_task", b"source_task", "tasks", b"tasks"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "barrier_id", + b"barrier_id", + "barrier_timeout_in_ms", + b"barrier_timeout_in_ms", + "source_task", + b"source_task", + "tasks", + b"tasks", + ], + ) -> None: ... global___BarrierRequest = BarrierRequest @@ -696,9 +634,7 @@ global___BarrierRequest = BarrierRequest class BarrierResponse(google.protobuf.message.Message): DESCRIPTOR: google.protobuf.descriptor.Descriptor - def __init__( - self, - ) -> None: ... + def __init__(self) -> None: ... global___BarrierResponse = BarrierResponse @@ -715,12 +651,7 @@ class CancelBarrierRequest(google.protobuf.message.Message): def source_task(self) -> global___CoordinatedTask: """Task that is making the request.""" - def __init__( - self, - *, - barrier_id: builtins.str | None = ..., - source_task: global___CoordinatedTask | None = ..., - ) -> None: ... + def __init__(self, *, barrier_id: builtins.str | None = ..., source_task: global___CoordinatedTask | None = ...) -> None: ... def HasField(self, field_name: typing.Literal["source_task", b"source_task"]) -> builtins.bool: ... def ClearField(self, field_name: typing.Literal["barrier_id", b"barrier_id", "source_task", b"source_task"]) -> None: ... @@ -730,8 +661,6 @@ global___CancelBarrierRequest = CancelBarrierRequest class CancelBarrierResponse(google.protobuf.message.Message): DESCRIPTOR: google.protobuf.descriptor.Descriptor - def __init__( - self, - ) -> None: ... + def __init__(self) -> None: ... global___CancelBarrierResponse = CancelBarrierResponse diff --git a/stubs/tensorflow/tensorflow/tsl/protobuf/distributed_runtime_payloads_pb2.pyi b/stubs/tensorflow/tensorflow/tsl/protobuf/distributed_runtime_payloads_pb2.pyi index 57202db5c159..54b11ce7c741 100644 --- a/stubs/tensorflow/tensorflow/tsl/protobuf/distributed_runtime_payloads_pb2.pyi +++ b/stubs/tensorflow/tensorflow/tsl/protobuf/distributed_runtime_payloads_pb2.pyi @@ -30,22 +30,13 @@ class GrpcPayloadContainer(google.protobuf.message.Message): VALUE_FIELD_NUMBER: builtins.int key: builtins.str value: builtins.bytes - def __init__( - self, - *, - key: builtins.str | None = ..., - value: builtins.bytes | None = ..., - ) -> None: ... + def __init__(self, *, key: builtins.str | None = ..., value: builtins.bytes | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["key", b"key", "value", b"value"]) -> None: ... PAYLOADS_FIELD_NUMBER: builtins.int @property def payloads(self) -> google.protobuf.internal.containers.ScalarMap[builtins.str, builtins.bytes]: ... - def __init__( - self, - *, - payloads: collections.abc.Mapping[builtins.str, builtins.bytes] | None = ..., - ) -> None: ... + def __init__(self, *, payloads: collections.abc.Mapping[builtins.str, builtins.bytes] | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["payloads", b"payloads"]) -> None: ... global___GrpcPayloadContainer = GrpcPayloadContainer @@ -59,9 +50,7 @@ class GrpcPayloadsLost(google.protobuf.message.Message): DESCRIPTOR: google.protobuf.descriptor.Descriptor - def __init__( - self, - ) -> None: ... + def __init__(self) -> None: ... global___GrpcPayloadsLost = GrpcPayloadsLost @@ -75,8 +64,6 @@ class WorkerPossiblyRestarted(google.protobuf.message.Message): DESCRIPTOR: google.protobuf.descriptor.Descriptor - def __init__( - self, - ) -> None: ... + def __init__(self) -> None: ... global___WorkerPossiblyRestarted = WorkerPossiblyRestarted diff --git a/stubs/tensorflow/tensorflow/tsl/protobuf/dnn_pb2.pyi b/stubs/tensorflow/tensorflow/tsl/protobuf/dnn_pb2.pyi index 929709311d04..c8d947994d74 100644 --- a/stubs/tensorflow/tensorflow/tsl/protobuf/dnn_pb2.pyi +++ b/stubs/tensorflow/tensorflow/tsl/protobuf/dnn_pb2.pyi @@ -63,7 +63,9 @@ class _DataLayout: ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType -class _DataLayoutEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_DataLayout.ValueType], builtins.type): +class _DataLayoutEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_DataLayout.ValueType], builtins.type +): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor kYXDepthBatch: _DataLayout.ValueType # 0 """Naming convention: @@ -121,7 +123,9 @@ class _FilterLayout: ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType -class _FilterLayoutEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_FilterLayout.ValueType], builtins.type): +class _FilterLayoutEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_FilterLayout.ValueType], builtins.type +): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor kOutputInputYX: _FilterLayout.ValueType # 0 """Naming convention: @@ -175,7 +179,9 @@ class _ActivationMode: ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType -class _ActivationModeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_ActivationMode.ValueType], builtins.type): +class _ActivationModeEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_ActivationMode.ValueType], builtins.type +): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor kNone: _ActivationMode.ValueType # 0 kSigmoid: _ActivationMode.ValueType # 1 @@ -229,7 +235,9 @@ class _ConvolutionMode: ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType -class _ConvolutionModeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_ConvolutionMode.ValueType], builtins.type): +class _ConvolutionModeEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_ConvolutionMode.ValueType], builtins.type +): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor CROSS_CORRELATION: _ConvolutionMode.ValueType # 0 CONVOLUTION: _ConvolutionMode.ValueType # 1 @@ -248,7 +256,9 @@ class _ConvolutionKind: ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType -class _ConvolutionKindEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_ConvolutionKind.ValueType], builtins.type): +class _ConvolutionKindEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_ConvolutionKind.ValueType], builtins.type +): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor INVALID: _ConvolutionKind.ValueType # 0 FORWARD: _ConvolutionKind.ValueType # 1 @@ -289,7 +299,9 @@ class _FusedMHAKind: ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType -class _FusedMHAKindEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_FusedMHAKind.ValueType], builtins.type): +class _FusedMHAKindEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_FusedMHAKind.ValueType], builtins.type +): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor BMM1_OUTPUT_UNKNOWN: _FusedMHAKind.ValueType # 0 BMM1_OUTPUT_INPUT_TYPE: _FusedMHAKind.ValueType # 1 @@ -307,7 +319,9 @@ class _FMHAMaskKind: ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType -class _FMHAMaskKindEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_FMHAMaskKind.ValueType], builtins.type): +class _FMHAMaskKindEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_FMHAMaskKind.ValueType], builtins.type +): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor NO_MASK: _FMHAMaskKind.ValueType # 0 PADDING: _FMHAMaskKind.ValueType # 1 @@ -348,9 +362,30 @@ class TensorDescriptorProto(google.protobuf.message.Message): data_layout: global___DataLayout.ValueType | None = ..., filter_layout: global___FilterLayout.ValueType | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["data_layout", b"data_layout", "filter_layout", b"filter_layout", "layout_oneof", b"layout_oneof"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["data_layout", b"data_layout", "data_type", b"data_type", "dimensions", b"dimensions", "filter_layout", b"filter_layout", "layout_oneof", b"layout_oneof"]) -> None: ... - def WhichOneof(self, oneof_group: typing.Literal["layout_oneof", b"layout_oneof"]) -> typing.Literal["data_layout", "filter_layout"] | None: ... + def HasField( + self, + field_name: typing.Literal[ + "data_layout", b"data_layout", "filter_layout", b"filter_layout", "layout_oneof", b"layout_oneof" + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "data_layout", + b"data_layout", + "data_type", + b"data_type", + "dimensions", + b"dimensions", + "filter_layout", + b"filter_layout", + "layout_oneof", + b"layout_oneof", + ], + ) -> None: ... + def WhichOneof( + self, oneof_group: typing.Literal["layout_oneof", b"layout_oneof"] + ) -> typing.Literal["data_layout", "filter_layout"] | None: ... global___TensorDescriptorProto = TensorDescriptorProto @@ -364,7 +399,9 @@ class AlgorithmProto(google.protobuf.message.Message): ValueType = typing.NewType("ValueType", builtins.int) V: typing_extensions.TypeAlias = ValueType - class _MathTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[AlgorithmProto._MathType.ValueType], builtins.type): + class _MathTypeEnumTypeWrapper( + google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[AlgorithmProto._MathType.ValueType], builtins.type + ): DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor DEFAULT_MATH: AlgorithmProto._MathType.ValueType # 0 TENSOR_OP_MATH: AlgorithmProto._MathType.ValueType # 1 @@ -387,12 +424,7 @@ class AlgorithmProto(google.protobuf.message.Message): VALUE_FIELD_NUMBER: builtins.int key: builtins.int value: builtins.int - def __init__( - self, - *, - key: builtins.int | None = ..., - value: builtins.int | None = ..., - ) -> None: ... + def __init__(self, *, key: builtins.int | None = ..., value: builtins.int | None = ...) -> None: ... def ClearField(self, field_name: typing.Literal["key", b"key", "value", b"value"]) -> None: ... ALGO_ID_FIELD_NUMBER: builtins.int @@ -433,7 +465,21 @@ class AlgorithmProto(google.protobuf.message.Message): workspace_size: google.protobuf.wrappers_pb2.UInt64Value | None = ..., ) -> None: ... def HasField(self, field_name: typing.Literal["workspace_size", b"workspace_size"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["algo_id", b"algo_id", "is_cudnn_frontend", b"is_cudnn_frontend", "math_type", b"math_type", "tuning_knobs", b"tuning_knobs", "workspace_size", b"workspace_size"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "algo_id", + b"algo_id", + "is_cudnn_frontend", + b"is_cudnn_frontend", + "math_type", + b"math_type", + "tuning_knobs", + b"tuning_knobs", + "workspace_size", + b"workspace_size", + ], + ) -> None: ... global___AlgorithmProto = AlgorithmProto @@ -461,14 +507,52 @@ class AlgorithmConfigProto(google.protobuf.message.Message): algorithm_no_scratch: global___AlgorithmProto | None = ..., scratch_size: builtins.int | None = ..., ) -> None: ... - def HasField(self, field_name: typing.Literal["algorithm", b"algorithm", "algorithm_no_scratch", b"algorithm_no_scratch", "optional_algorithm", b"optional_algorithm", "optional_algorithm_no_scratch", b"optional_algorithm_no_scratch", "optional_scratch_size", b"optional_scratch_size", "scratch_size", b"scratch_size"]) -> builtins.bool: ... - def ClearField(self, field_name: typing.Literal["algorithm", b"algorithm", "algorithm_no_scratch", b"algorithm_no_scratch", "optional_algorithm", b"optional_algorithm", "optional_algorithm_no_scratch", b"optional_algorithm_no_scratch", "optional_scratch_size", b"optional_scratch_size", "scratch_size", b"scratch_size"]) -> None: ... + def HasField( + self, + field_name: typing.Literal[ + "algorithm", + b"algorithm", + "algorithm_no_scratch", + b"algorithm_no_scratch", + "optional_algorithm", + b"optional_algorithm", + "optional_algorithm_no_scratch", + b"optional_algorithm_no_scratch", + "optional_scratch_size", + b"optional_scratch_size", + "scratch_size", + b"scratch_size", + ], + ) -> builtins.bool: ... + def ClearField( + self, + field_name: typing.Literal[ + "algorithm", + b"algorithm", + "algorithm_no_scratch", + b"algorithm_no_scratch", + "optional_algorithm", + b"optional_algorithm", + "optional_algorithm_no_scratch", + b"optional_algorithm_no_scratch", + "optional_scratch_size", + b"optional_scratch_size", + "scratch_size", + b"scratch_size", + ], + ) -> None: ... @typing.overload - def WhichOneof(self, oneof_group: typing.Literal["optional_algorithm", b"optional_algorithm"]) -> typing.Literal["algorithm"] | None: ... + def WhichOneof( + self, oneof_group: typing.Literal["optional_algorithm", b"optional_algorithm"] + ) -> typing.Literal["algorithm"] | None: ... @typing.overload - def WhichOneof(self, oneof_group: typing.Literal["optional_algorithm_no_scratch", b"optional_algorithm_no_scratch"]) -> typing.Literal["algorithm_no_scratch"] | None: ... + def WhichOneof( + self, oneof_group: typing.Literal["optional_algorithm_no_scratch", b"optional_algorithm_no_scratch"] + ) -> typing.Literal["algorithm_no_scratch"] | None: ... @typing.overload - def WhichOneof(self, oneof_group: typing.Literal["optional_scratch_size", b"optional_scratch_size"]) -> typing.Literal["scratch_size"] | None: ... + def WhichOneof( + self, oneof_group: typing.Literal["optional_scratch_size", b"optional_scratch_size"] + ) -> typing.Literal["scratch_size"] | None: ... global___AlgorithmConfigProto = AlgorithmConfigProto @@ -512,6 +596,24 @@ class ConvolutionDescriptorProto(google.protobuf.message.Message): convolution_mode: global___ConvolutionMode.ValueType | None = ..., name: builtins.str | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["compute_mode", b"compute_mode", "convolution_mode", b"convolution_mode", "dilations", b"dilations", "group_count", b"group_count", "name", b"name", "paddings", b"paddings", "strides", b"strides"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "compute_mode", + b"compute_mode", + "convolution_mode", + b"convolution_mode", + "dilations", + b"dilations", + "group_count", + b"group_count", + "name", + b"name", + "paddings", + b"paddings", + "strides", + b"strides", + ], + ) -> None: ... global___ConvolutionDescriptorProto = ConvolutionDescriptorProto diff --git a/stubs/tensorflow/tensorflow/tsl/protobuf/histogram_pb2.pyi b/stubs/tensorflow/tensorflow/tsl/protobuf/histogram_pb2.pyi index a33728dc47e4..986ebfcdc44f 100644 --- a/stubs/tensorflow/tensorflow/tsl/protobuf/histogram_pb2.pyi +++ b/stubs/tensorflow/tensorflow/tsl/protobuf/histogram_pb2.pyi @@ -55,6 +55,24 @@ class HistogramProto(google.protobuf.message.Message): bucket_limit: collections.abc.Iterable[builtins.float] | None = ..., bucket: collections.abc.Iterable[builtins.float] | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["bucket", b"bucket", "bucket_limit", b"bucket_limit", "max", b"max", "min", b"min", "num", b"num", "sum", b"sum", "sum_squares", b"sum_squares"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "bucket", + b"bucket", + "bucket_limit", + b"bucket_limit", + "max", + b"max", + "min", + b"min", + "num", + b"num", + "sum", + b"sum", + "sum_squares", + b"sum_squares", + ], + ) -> None: ... global___HistogramProto = HistogramProto diff --git a/stubs/tensorflow/tensorflow/tsl/protobuf/rpc_options_pb2.pyi b/stubs/tensorflow/tensorflow/tsl/protobuf/rpc_options_pb2.pyi index 6e08a2bf99ef..062ea4bd4610 100644 --- a/stubs/tensorflow/tensorflow/tsl/protobuf/rpc_options_pb2.pyi +++ b/stubs/tensorflow/tensorflow/tsl/protobuf/rpc_options_pb2.pyi @@ -64,6 +64,22 @@ class RPCOptions(google.protobuf.message.Message): disable_session_connection_sharing: builtins.bool | None = ..., num_channels_per_target: builtins.int | None = ..., ) -> None: ... - def ClearField(self, field_name: typing.Literal["cache_rpc_response", b"cache_rpc_response", "compression_algorithm", b"compression_algorithm", "compression_level", b"compression_level", "disable_session_connection_sharing", b"disable_session_connection_sharing", "num_channels_per_target", b"num_channels_per_target", "use_rpc_for_inprocess_master", b"use_rpc_for_inprocess_master"]) -> None: ... + def ClearField( + self, + field_name: typing.Literal[ + "cache_rpc_response", + b"cache_rpc_response", + "compression_algorithm", + b"compression_algorithm", + "compression_level", + b"compression_level", + "disable_session_connection_sharing", + b"disable_session_connection_sharing", + "num_channels_per_target", + b"num_channels_per_target", + "use_rpc_for_inprocess_master", + b"use_rpc_for_inprocess_master", + ], + ) -> None: ... global___RPCOptions = RPCOptions diff --git a/stubs/tensorflow/tensorflow/tsl/protobuf/status_pb2.pyi b/stubs/tensorflow/tensorflow/tsl/protobuf/status_pb2.pyi index 77e0f5b7bf39..b2df45b32086 100644 --- a/stubs/tensorflow/tensorflow/tsl/protobuf/status_pb2.pyi +++ b/stubs/tensorflow/tensorflow/tsl/protobuf/status_pb2.pyi @@ -27,10 +27,7 @@ class StatusProto(google.protobuf.message.Message): message: builtins.str """Detail error message.""" def __init__( - self, - *, - code: tensorflow.tsl.protobuf.error_codes_pb2.Code.ValueType | None = ..., - message: builtins.str | None = ..., + self, *, code: tensorflow.tsl.protobuf.error_codes_pb2.Code.ValueType | None = ..., message: builtins.str | None = ... ) -> None: ... def ClearField(self, field_name: typing.Literal["code", b"code", "message", b"message"]) -> None: ...