Flask-ML/tests/constants.py at 6541e886adbb862d78e5353e9732b9dfb6f2bfe5 · UMass-Rescue/Flask-ML · GitHub

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
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
from flask_ml.flask_ml_server.models import *

TEXT_INPUT_SCHEMA = InputSchema(key="text_input", label="Text Input", input_type=InputType.TEXT)
TEXTAREA_INPUT_SCHEMA = InputSchema(key="text_input", label="Text Area Input", input_type=InputType.TEXTAREA)
FILE_INPUT_SCHEMA = InputSchema(key="file_input", label="File Input", input_type=InputType.FILE)
BATCHTEXT_INPUT_SCHEMA = InputSchema(
    key="text_inputs", label="Batch Text Inputs", input_type=InputType.BATCHTEXT
)
BATCHFILE_INPUT_SCHEMA = InputSchema(
    key="file_inputs", label="Batch File Inputs", input_type=InputType.BATCHFILE
)
DIRECTORY_INPUT_SCHEMA = InputSchema(key="dir_input", label="Directory Input", input_type=InputType.DIRECTORY)
BATCHDIRECTORY_INPUT_SCHEMA = InputSchema(
    key="dir_inputs", label="Batch Directory Inputs", input_type=InputType.BATCHDIRECTORY
)
NEWFILEINPUT_INPUT_SCHEMA = InputSchema(
    key="file_input",
    label="New File Input",
    input_type=NewFileInputType(allowed_extensions="*", default_extension=".test"),
)

TEXT_PARAM_SCHEMA = ParameterSchema(
    key="param1",
    label="Text Parameter",
    value=TextParameterDescriptor(parameter_type=ParameterType.TEXT, default="default"),
)
ENUM_PARAM_SCHEMA = ParameterSchema(
    key="param1",
    label="Enum Parameter",
    value=EnumParameterDescriptor(
        parameter_type=ParameterType.ENUM,
        enum_vals=[EnumVal(label="Option 1", key="option_1"), EnumVal(label="Option 2", key="option_2")],
        default="option_1",
    ),
)
FLOAT_PARAM_SCHEMA = ParameterSchema(
    key="param1",
    label="Float Parameter",
    value=FloatParameterDescriptor(parameter_type=ParameterType.FLOAT, default=0.0),
)
INT_PARAM_SCHEMA = ParameterSchema(
    key="param1",
    label="Int Parameter",
    value=IntParameterDescriptor(parameter_type=ParameterType.INT, default=1),
)
RANGED_FLOAT_PARAM_SCHEMA = ParameterSchema(
    key="param1",
    label="Ranged Float Parameter",
    value=RangedFloatParameterDescriptor(
        parameter_type=ParameterType.RANGED_FLOAT,
        default=0.0,
        range=FloatRangeDescriptor(min=0.0, max=1.0),
    ),
)
RANGED_INT_PARAM_SCHEMA = ParameterSchema(
    key="param1",
    label="Ranged Int Parameter",
    value=RangedIntParameterDescriptor(
        parameter_type=ParameterType.RANGED_INT,
        default=0,
        range=IntRangeDescriptor(min=0, max=10),
    ),
)
BOOL_PARAM_SCHEMA = ParameterSchema(
    key="param1",
    label="Boolean Parameter",
    value=BoolParameterDescriptor(parameter_type=ParameterType.BOOLEAN, default=False),
)