diff --git a/ablkit/learning/basic_nn.py b/ablkit/learning/basic_nn.py index 03d2a74..b783b49 100644 --- a/ablkit/learning/basic_nn.py +++ b/ablkit/learning/basic_nn.py @@ -54,7 +54,7 @@ class BasicNN: in the ``fit`` and ``train_epoch`` methods. Defaults to None. test_transform : Callable[..., Any], optional A function/transform that takes an object and returns a transformed version in the - ``predict``, ``predict_proba`` and ``score`` methods.Defaults to None. + ``predict``, ``predict_proba`` and ``score`` methods. Defaults to None. collate_fn : Callable[[List[T]], Any], optional The function used to collate data. Defaults to None. """ diff --git a/ablkit/learning/model_converter.py b/ablkit/learning/model_converter.py index 99eb56f..5237efe 100644 --- a/ablkit/learning/model_converter.py +++ b/ablkit/learning/model_converter.py @@ -71,7 +71,7 @@ class ModelConverter: in the `fit` and `train_epoch` methods. Defaults to None. test_transform : Callable[..., Any], optional A function/transform that takes an object and returns a transformed version in the - `predict`, `predict_proba` and `score` methods.Defaults to None. + `predict`, `predict_proba` and `score` methods. Defaults to None. collate_fn : Callable[[List[T]], Any], optional The function used to collate data. Defaults to None. @@ -161,7 +161,7 @@ class ModelConverter: in the `fit` and `train_epoch` methods. Defaults to None. test_transform : Callable[..., Any], optional A function/transform that takes an object and returns a transformed version in the - `predict`, `predict_proba` and `score` methods.Defaults to None. + `predict`, `predict_proba` and `score` methods. Defaults to None. collate_fn : Callable[[List[T]], Any], optional The function used to collate data. Defaults to None.