|
3 | 3 |
|
4 | 4 | from abc import ABC, abstractmethod |
5 | 5 | from enum import Enum |
6 | | -from typing import Literal, TypeVar, overload |
| 6 | +from typing import Literal, TypeVar |
7 | 7 |
|
8 | 8 | from ..models import ModelConfig, ModelProvider |
9 | 9 | from ..sampler_params import SamplerType |
@@ -40,12 +40,6 @@ def __init__(self, model_configs: list[ModelConfig]): |
40 | 40 | self._sampler_params = get_sampler_params() |
41 | 41 | self._model_configs = model_configs |
42 | 42 |
|
43 | | - @overload |
44 | | - def display(self, info_type: Literal[InfoType.SAMPLERS], **kwargs) -> None: ... |
45 | | - |
46 | | - @overload |
47 | | - def display(self, info_type: Literal[InfoType.MODEL_CONFIGS], **kwargs) -> None: ... |
48 | | - |
49 | 43 | def display(self, info_type: ConfigBuilderInfoType, **kwargs) -> None: |
50 | 44 | """Display information based on the provided info type. |
51 | 45 |
|
@@ -77,9 +71,6 @@ class InterfaceInfo(InfoDisplay): |
77 | 71 | def __init__(self, model_providers: list[ModelProvider]): |
78 | 72 | self._model_providers = model_providers |
79 | 73 |
|
80 | | - @overload |
81 | | - def display(self, info_type: Literal[InfoType.MODEL_PROVIDERS], **kwargs) -> None: ... |
82 | | - |
83 | 74 | def display(self, info_type: DataDesignerInfoType, **kwargs) -> None: |
84 | 75 | """Display information based on the provided info type. |
85 | 76 |
|
|
0 commit comments