88 lines
3.1 KiB
Python
88 lines
3.1 KiB
Python
def common_input_types():
|
|
"""
|
|
Defines the input types for a plot function.
|
|
|
|
Returns:
|
|
Dict[str, Any]: A dictionary specifying required input types.
|
|
"""
|
|
return {
|
|
"required": {
|
|
"dataframe": ("DATAFRAME", {}),
|
|
"x_column_name": ("STRING", {"default": ""}),
|
|
"y_column_name": ("STRING", {"default": ""}),
|
|
"title": ("STRING", {"default": ""}),
|
|
"x_axis_label": ("STRING", {"default": ""}),
|
|
"y_axis_label": ("STRING", {"default": ""}),
|
|
"legend_label": ("STRING", {"default": ""}),
|
|
"x_tick_as_int": ("BOOLEAN", {"default": False}), # To show 2002 instead of 2002.5 on x-axis
|
|
"style": ("STRING", {"default": ""}),
|
|
"palette": ("STRING", {"default": ""})
|
|
}
|
|
}
|
|
|
|
def common_input_types_pt():
|
|
"""
|
|
Defines the input types for a plot function that takes PyTorch tensor.
|
|
|
|
Returns:
|
|
Dict[str, Any]: A dictionary specifying required input types.
|
|
"""
|
|
return {
|
|
"required": {
|
|
"tens": ("TENSOR", {}),
|
|
"x_axis_dim": ("INT", {"default": 0, "min":-100, "max":1e6}),
|
|
"y_axis_dims": ("STRING", {"default": ""}),
|
|
"title": ("STRING", {"default": ""}),
|
|
"x_axis_label": ("STRING", {"default": ""}),
|
|
"y_axis_label": ("STRING", {"default": ""}),
|
|
"legend_label": ("STRING", {"default": ""}),
|
|
"x_tick_as_int": ("BOOLEAN", {"default": False}), # To show 2002 instead of 2002.5 on x-axis
|
|
"style": ("STRING", {"default": ""}),
|
|
"palette": ("STRING", {"default": ""})
|
|
}
|
|
}
|
|
|
|
|
|
def common_input_types_y_only_bin():
|
|
"""
|
|
Defines the input types for a plot function for types of charts that do
|
|
not require x-series data as input, but takes bins.
|
|
|
|
Returns:
|
|
Dict[str, Any]: A dictionary specifying required input types.
|
|
"""
|
|
return {
|
|
"required": {
|
|
"dataframe": ("DATAFRAME", {}),
|
|
"y_column_name": ("STRING", {"default": ""}),
|
|
"title": ("STRING", {"default": ""}),
|
|
"y_axis_label": ("STRING", {"default": ""}),
|
|
# "legend_label": ("STRING", {"default": ""}),
|
|
"bins": ("INT", {"default": 10, "min": 2, "max": 200}),
|
|
"style": ("STRING", {"default": ""}),
|
|
"palette": ("STRING", {"default": ""})
|
|
}
|
|
}
|
|
|
|
|
|
def common_input_types_y_only_bin_pt():
|
|
"""
|
|
Defines the input types for a plot function for types of charts that do
|
|
not require x-series data as input, but takes bins.
|
|
|
|
Returns:
|
|
Dict[str, Any]: A dictionary specifying required input types.
|
|
"""
|
|
return {
|
|
"required": {
|
|
"tens": ("TENSOR", {}),
|
|
"y_axis_dims": ("STRING", {"default": ""}),
|
|
"title": ("STRING", {"default": ""}),
|
|
"y_axis_label": ("STRING", {"default": ""}),
|
|
# "legend_label": ("STRING", {"default": ""}),
|
|
"bins": ("INT", {"default": 10, "min": 2, "max": 200}),
|
|
"style": ("STRING", {"default": ""}),
|
|
"palette": ("STRING", {"default": ""})
|
|
}
|
|
}
|