# 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 GradsItem(
    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 = {
            "connection_name",
            "encoder_names",
            "grads",
            "node_name",
            "label",
            "visualizer_name",
        }
        
        class properties:
            node_name = schemas.StrSchema
            label = schemas.StrSchema
        
            @staticmethod
            def grads() -> typing.Type['VisData']:
                return VisData
            visualizer_name = schemas.StrSchema
            
            
            class encoder_names(
                schemas.ListSchema
            ):
            
            
                class MetaOapg:
                    items = schemas.StrSchema
            
                def __new__(
                    cls,
                    arg: typing.Union[typing.Tuple[typing.Union[MetaOapg.items, str, ]], typing.List[typing.Union[MetaOapg.items, str, ]]],
                    _configuration: typing.Optional[schemas.Configuration] = None,
                ) -> 'encoder_names':
                    return super().__new__(
                        cls,
                        arg,
                        _configuration=_configuration,
                    )
            
                def __getitem__(self, i: int) -> MetaOapg.items:
                    return super().__getitem__(i)
            connection_name = schemas.StrSchema
        
            @staticmethod
            def channel_importance() -> typing.Type['HorizontalBarData']:
                return HorizontalBarData
            __annotations__ = {
                "node_name": node_name,
                "label": label,
                "grads": grads,
                "visualizer_name": visualizer_name,
                "encoder_names": encoder_names,
                "connection_name": connection_name,
                "channel_importance": channel_importance,
            }
        additional_properties = schemas.NotAnyTypeSchema
    
    connection_name: MetaOapg.properties.connection_name
    encoder_names: MetaOapg.properties.encoder_names
    grads: 'VisData'
    node_name: MetaOapg.properties.node_name
    label: MetaOapg.properties.label
    visualizer_name: MetaOapg.properties.visualizer_name
    
    @typing.overload
    def __getitem__(self, name: typing_extensions.Literal["connection_name"]) -> MetaOapg.properties.connection_name: ...
    
    @typing.overload
    def __getitem__(self, name: typing_extensions.Literal["encoder_names"]) -> MetaOapg.properties.encoder_names: ...
    
    @typing.overload
    def __getitem__(self, name: typing_extensions.Literal["grads"]) -> 'VisData': ...
    
    @typing.overload
    def __getitem__(self, name: typing_extensions.Literal["node_name"]) -> MetaOapg.properties.node_name: ...
    
    @typing.overload
    def __getitem__(self, name: typing_extensions.Literal["label"]) -> MetaOapg.properties.label: ...
    
    @typing.overload
    def __getitem__(self, name: typing_extensions.Literal["visualizer_name"]) -> MetaOapg.properties.visualizer_name: ...
    
    @typing.overload
    def __getitem__(self, name: typing_extensions.Literal["channel_importance"]) -> 'HorizontalBarData': ...
    
    def __getitem__(self, name: typing.Union[typing_extensions.Literal["connection_name"], typing_extensions.Literal["encoder_names"], typing_extensions.Literal["grads"], typing_extensions.Literal["node_name"], typing_extensions.Literal["label"], typing_extensions.Literal["visualizer_name"], typing_extensions.Literal["channel_importance"], ]):
        # dict_instance[name] accessor
        return super().__getitem__(name)
    
    @typing.overload
    def get_item_oapg(self, name: typing_extensions.Literal["connection_name"]) -> MetaOapg.properties.connection_name: ...
    
    @typing.overload
    def get_item_oapg(self, name: typing_extensions.Literal["encoder_names"]) -> MetaOapg.properties.encoder_names: ...
    
    @typing.overload
    def get_item_oapg(self, name: typing_extensions.Literal["grads"]) -> 'VisData': ...
    
    @typing.overload
    def get_item_oapg(self, name: typing_extensions.Literal["node_name"]) -> MetaOapg.properties.node_name: ...
    
    @typing.overload
    def get_item_oapg(self, name: typing_extensions.Literal["label"]) -> MetaOapg.properties.label: ...
    
    @typing.overload
    def get_item_oapg(self, name: typing_extensions.Literal["visualizer_name"]) -> MetaOapg.properties.visualizer_name: ...
    
    @typing.overload
    def get_item_oapg(self, name: typing_extensions.Literal["channel_importance"]) -> typing.Union['HorizontalBarData', schemas.Unset]: ...
    
    def get_item_oapg(self, name: typing.Union[typing_extensions.Literal["connection_name"], typing_extensions.Literal["encoder_names"], typing_extensions.Literal["grads"], typing_extensions.Literal["node_name"], typing_extensions.Literal["label"], typing_extensions.Literal["visualizer_name"], typing_extensions.Literal["channel_importance"], ]):
        return super().get_item_oapg(name)

    def __new__(
        cls,
        *args: typing.Union[dict, frozendict.frozendict, ],
        connection_name: typing.Union[MetaOapg.properties.connection_name, str, ],
        encoder_names: typing.Union[MetaOapg.properties.encoder_names, list, tuple, ],
        grads: 'VisData',
        node_name: typing.Union[MetaOapg.properties.node_name, str, ],
        label: typing.Union[MetaOapg.properties.label, str, ],
        visualizer_name: typing.Union[MetaOapg.properties.visualizer_name, str, ],
        channel_importance: typing.Union['HorizontalBarData', schemas.Unset] = schemas.unset,
        _configuration: typing.Optional[schemas.Configuration] = None,
    ) -> 'GradsItem':
        return super().__new__(
            cls,
            *args,
            connection_name=connection_name,
            encoder_names=encoder_names,
            grads=grads,
            node_name=node_name,
            label=label,
            visualizer_name=visualizer_name,
            channel_importance=channel_importance,
            _configuration=_configuration,
        )

from tensorleap_openapi_client.model.horizontal_bar_data import HorizontalBarData
from tensorleap_openapi_client.model.vis_data import VisData
