# coding: utf-8

"""
    node-server

    No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)  # noqa: E501

    The version of the OpenAPI document: 0.0.237
    Generated by: https://openapi-generator.tech
"""

from datetime import date, datetime  # noqa: F401
import decimal  # noqa: F401
import functools  # noqa: F401
import io  # noqa: F401
import re  # noqa: F401
import typing  # noqa: F401
import typing_extensions  # noqa: F401
import uuid  # noqa: F401

import frozendict  # noqa: F401

from tensorleap-openapi-client import schemas  # noqa: F401


class SampleSelectionParams(
    schemas.DictSchema
):
    """NOTE: This class is auto generated by OpenAPI Generator.
    Ref: https://openapi-generator.tech

    Do not edit the class manually.
    """


    class MetaOapg:
        required = {
            "fromEpoch",
            "fromDatasetSlice",
            "modelId",
            "numOfSamples",
            "batchSize",
            "sampleIdentity",
        }
        
        class properties:
            modelId = schemas.StrSchema
        
            @staticmethod
            def sampleIdentity() -> typing.Type['SampleIdentity']:
                return SampleIdentity
        
            @staticmethod
            def fromDatasetSlice() -> typing.Type['DataStateType']:
                return DataStateType
            batchSize = schemas.Float64Schema
            numOfSamples = schemas.Float64Schema
            fromEpoch = schemas.Float64Schema
            __annotations__ = {
                "modelId": modelId,
                "sampleIdentity": sampleIdentity,
                "fromDatasetSlice": fromDatasetSlice,
                "batchSize": batchSize,
                "numOfSamples": numOfSamples,
                "fromEpoch": fromEpoch,
            }
        additional_properties = schemas.NotAnyTypeSchema
    
    fromEpoch: MetaOapg.properties.fromEpoch
    fromDatasetSlice: 'DataStateType'
    modelId: MetaOapg.properties.modelId
    numOfSamples: MetaOapg.properties.numOfSamples
    batchSize: MetaOapg.properties.batchSize
    sampleIdentity: 'SampleIdentity'
    
    @typing.overload
    def __getitem__(self, name: typing_extensions.Literal["fromEpoch"]) -> MetaOapg.properties.fromEpoch: ...
    
    @typing.overload
    def __getitem__(self, name: typing_extensions.Literal["fromDatasetSlice"]) -> 'DataStateType': ...
    
    @typing.overload
    def __getitem__(self, name: typing_extensions.Literal["modelId"]) -> MetaOapg.properties.modelId: ...
    
    @typing.overload
    def __getitem__(self, name: typing_extensions.Literal["numOfSamples"]) -> MetaOapg.properties.numOfSamples: ...
    
    @typing.overload
    def __getitem__(self, name: typing_extensions.Literal["batchSize"]) -> MetaOapg.properties.batchSize: ...
    
    @typing.overload
    def __getitem__(self, name: typing_extensions.Literal["sampleIdentity"]) -> 'SampleIdentity': ...
    
    def __getitem__(self, name: typing.Union[typing_extensions.Literal["fromEpoch"], typing_extensions.Literal["fromDatasetSlice"], typing_extensions.Literal["modelId"], typing_extensions.Literal["numOfSamples"], typing_extensions.Literal["batchSize"], typing_extensions.Literal["sampleIdentity"], ]):
        # dict_instance[name] accessor
        return super().__getitem__(name)
    
    @typing.overload
    def get_item_oapg(self, name: typing_extensions.Literal["fromEpoch"]) -> MetaOapg.properties.fromEpoch: ...
    
    @typing.overload
    def get_item_oapg(self, name: typing_extensions.Literal["fromDatasetSlice"]) -> 'DataStateType': ...
    
    @typing.overload
    def get_item_oapg(self, name: typing_extensions.Literal["modelId"]) -> MetaOapg.properties.modelId: ...
    
    @typing.overload
    def get_item_oapg(self, name: typing_extensions.Literal["numOfSamples"]) -> MetaOapg.properties.numOfSamples: ...
    
    @typing.overload
    def get_item_oapg(self, name: typing_extensions.Literal["batchSize"]) -> MetaOapg.properties.batchSize: ...
    
    @typing.overload
    def get_item_oapg(self, name: typing_extensions.Literal["sampleIdentity"]) -> 'SampleIdentity': ...
    
    def get_item_oapg(self, name: typing.Union[typing_extensions.Literal["fromEpoch"], typing_extensions.Literal["fromDatasetSlice"], typing_extensions.Literal["modelId"], typing_extensions.Literal["numOfSamples"], typing_extensions.Literal["batchSize"], typing_extensions.Literal["sampleIdentity"], ]):
        return super().get_item_oapg(name)

    def __new__(
        cls,
        *args: typing.Union[dict, frozendict.frozendict, ],
        fromEpoch: typing.Union[MetaOapg.properties.fromEpoch, decimal.Decimal, int, float, ],
        fromDatasetSlice: 'DataStateType',
        modelId: typing.Union[MetaOapg.properties.modelId, str, ],
        numOfSamples: typing.Union[MetaOapg.properties.numOfSamples, decimal.Decimal, int, float, ],
        batchSize: typing.Union[MetaOapg.properties.batchSize, decimal.Decimal, int, float, ],
        sampleIdentity: 'SampleIdentity',
        _configuration: typing.Optional[schemas.Configuration] = None,
    ) -> 'SampleSelectionParams':
        return super().__new__(
            cls,
            *args,
            fromEpoch=fromEpoch,
            fromDatasetSlice=fromDatasetSlice,
            modelId=modelId,
            numOfSamples=numOfSamples,
            batchSize=batchSize,
            sampleIdentity=sampleIdentity,
            _configuration=_configuration,
        )

from tensorleap-openapi-client.model.data_state_type import DataStateType
from tensorleap-openapi-client.model.sample_identity import SampleIdentity
