yonikremer commited on
Commit
4b6c061
1 Parent(s): d102e03

added a type hint

Browse files
Files changed (1) hide show
  1. 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.