# @generated by generate_proto_mypy_stubs.py.  Do not edit!
import sys
from google.protobuf.descriptor import (
    Descriptor as google___protobuf___descriptor___Descriptor,
    EnumDescriptor as google___protobuf___descriptor___EnumDescriptor,
    FileDescriptor as google___protobuf___descriptor___FileDescriptor,
)

from google.protobuf.internal.containers import (
    RepeatedScalarFieldContainer as google___protobuf___internal___containers___RepeatedScalarFieldContainer,
)

from google.protobuf.internal.enum_type_wrapper import (
    _EnumTypeWrapper as google___protobuf___internal___enum_type_wrapper____EnumTypeWrapper,
)

from google.protobuf.message import (
    Message as google___protobuf___message___Message,
)

from typing import (
    Iterable as typing___Iterable,
    NewType as typing___NewType,
    Optional as typing___Optional,
    cast as typing___cast,
)

from typing_extensions import (
    Literal as typing_extensions___Literal,
)


builtin___bool = bool
builtin___bytes = bytes
builtin___float = float
builtin___int = int


DESCRIPTOR: google___protobuf___descriptor___FileDescriptor = ...

CompressionTypeProtoValue = typing___NewType('CompressionTypeProtoValue', builtin___int)
type___CompressionTypeProtoValue = CompressionTypeProtoValue
CompressionTypeProto: _CompressionTypeProto
class _CompressionTypeProto(google___protobuf___internal___enum_type_wrapper____EnumTypeWrapper[CompressionTypeProtoValue]):
    DESCRIPTOR: google___protobuf___descriptor___EnumDescriptor = ...
    NONE = typing___cast(CompressionTypeProtoValue, 0)
    PNG = typing___cast(CompressionTypeProtoValue, 1)
NONE = typing___cast(CompressionTypeProtoValue, 0)
PNG = typing___cast(CompressionTypeProtoValue, 1)
type___CompressionTypeProto = CompressionTypeProto

SensorTypeProtoValue = typing___NewType('SensorTypeProtoValue', builtin___int)
type___SensorTypeProtoValue = SensorTypeProtoValue
SensorTypeProto: _SensorTypeProto
class _SensorTypeProto(google___protobuf___internal___enum_type_wrapper____EnumTypeWrapper[SensorTypeProtoValue]):
    DESCRIPTOR: google___protobuf___descriptor___EnumDescriptor = ...
    OBSERVATION = typing___cast(SensorTypeProtoValue, 0)
    PARAMETERIZATION = typing___cast(SensorTypeProtoValue, 1)
    REWARD = typing___cast(SensorTypeProtoValue, 2)
OBSERVATION = typing___cast(SensorTypeProtoValue, 0)
PARAMETERIZATION = typing___cast(SensorTypeProtoValue, 1)
REWARD = typing___cast(SensorTypeProtoValue, 2)
type___SensorTypeProto = SensorTypeProto

class ObservationProto(google___protobuf___message___Message):
    DESCRIPTOR: google___protobuf___descriptor___Descriptor = ...
    class FloatData(google___protobuf___message___Message):
        DESCRIPTOR: google___protobuf___descriptor___Descriptor = ...
        data: google___protobuf___internal___containers___RepeatedScalarFieldContainer[builtin___float] = ...

        def __init__(self,
            *,
            data : typing___Optional[typing___Iterable[builtin___float]] = None,
            ) -> None: ...
        def ClearField(self, field_name: typing_extensions___Literal[u"data",b"data"]) -> None: ...
    type___FloatData = FloatData

    shape: google___protobuf___internal___containers___RepeatedScalarFieldContainer[builtin___int] = ...
    compression_type: type___CompressionTypeProtoValue = ...
    compressed_data: builtin___bytes = ...
    sensor_type: type___SensorTypeProtoValue = ...

    @property
    def float_data(self) -> type___ObservationProto.FloatData: ...

    def __init__(self,
        *,
        shape : typing___Optional[typing___Iterable[builtin___int]] = None,
        compression_type : typing___Optional[type___CompressionTypeProtoValue] = None,
        compressed_data : typing___Optional[builtin___bytes] = None,
        float_data : typing___Optional[type___ObservationProto.FloatData] = None,
        sensor_type : typing___Optional[type___SensorTypeProtoValue] = None,
        ) -> None: ...
    def HasField(self, field_name: typing_extensions___Literal[u"compressed_data",b"compressed_data",u"float_data",b"float_data",u"observation_data",b"observation_data"]) -> builtin___bool: ...
    def ClearField(self, field_name: typing_extensions___Literal[u"compressed_data",b"compressed_data",u"compression_type",b"compression_type",u"float_data",b"float_data",u"observation_data",b"observation_data",u"sensor_type",b"sensor_type",u"shape",b"shape"]) -> None: ...
    def WhichOneof(self, oneof_group: typing_extensions___Literal[u"observation_data",b"observation_data"]) -> typing_extensions___Literal["compressed_data","float_data"]: ...
type___ObservationProto = ObservationProto