File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -158,7 +158,7 @@ def fit( # noqa: C901 # Complexity is bad.
158158 X_val : list [str ] | None = None ,
159159 y_val : LabelType | None = None ,
160160 class_weight : torch .Tensor | None = None ,
161- seed : int = _DEFAULT_RANDOM_SEED ,
161+ random_seed : int = _DEFAULT_RANDOM_SEED ,
162162 ) -> StaticModelForClassification :
163163 """
164164 Fit a model.
@@ -188,11 +188,11 @@ def fit( # noqa: C901 # Complexity is bad.
188188 :param y_val: The labels to be used for validation.
189189 :param class_weight: The weight of the classes. If None, all classes are weighted equally. Must
190190 have the same length as the number of classes.
191- :param seed : The random seed to use. Defaults to 42.
191+ :param random_seed : The random seed to use. Defaults to 42.
192192 :return: The fitted model.
193193 :raises ValueError: If either X_val or y_val are provided, but not both.
194194 """
195- pl .seed_everything (seed )
195+ pl .seed_everything (random_seed )
196196 logger .info ("Re-initializing model." )
197197
198198 # Determine whether the task is multilabel based on the type of y.
You can’t perform that action at this time.
0 commit comments