yonikremer
commited on
Commit
•
4b6c061
1
Parent(s):
d102e03
added a type hint
Browse files- hanlde_form_submit.py +1 -1
hanlde_form_submit.py
CHANGED
@@ -8,7 +8,7 @@ from supported_models import get_supported_model_names
|
|
8 |
SUPPORTED_MODEL_NAMES = get_supported_model_names()
|
9 |
|
10 |
|
11 |
-
def create_pipeline(model_name: str, group_size) -> GroupedSamplingPipeLine:
|
12 |
"""
|
13 |
Creates a pipeline with the given model name and group size.
|
14 |
:param model_name: The name of the model to use.
|
|
|
8 |
SUPPORTED_MODEL_NAMES = get_supported_model_names()
|
9 |
|
10 |
|
11 |
+
def create_pipeline(model_name: str, group_size: int) -> GroupedSamplingPipeLine:
|
12 |
"""
|
13 |
Creates a pipeline with the given model name and group size.
|
14 |
:param model_name: The name of the model to use.
|