blob: 3a726b383d7ca1f518ef8fc825e2064ebb52f8eb (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
|
# ---------------------------------------------------------
# Copyright (c) Microsoft Corporation. All rights reserved.
# ---------------------------------------------------------
from typing import Any
from azure.ai.ml._schema.component.input_output import SUPPORTED_PARAM_TYPES
from azure.ai.ml.entities._mixins import DictMixin, RestTranslatableMixin
class _InputOutputBase(DictMixin, RestTranslatableMixin):
def __init__(
self,
*,
# pylint: disable=redefined-builtin
type: Any,
# pylint: disable=unused-argument
**kwargs: Any,
) -> None:
"""Base class for Input & Output class.
This class is introduced to support literal output in the future.
:param type: The type of the Input/Output.
:type type: str
"""
self.type = type
def _is_literal(self) -> bool:
"""Check whether input is a literal
:return: True if this input is literal input.
:rtype: bool
"""
return self.type in SUPPORTED_PARAM_TYPES
|