From 4b7696430637bfca3033a103fa865fec1e16568d Mon Sep 17 00:00:00 2001 From: troyyyyy Date: Tue, 9 Jan 2024 09:39:42 +0800 Subject: [PATCH] [FIX] minor changes --- ablkit/learning/basic_nn.py | 2 +- ablkit/learning/model_converter.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) 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.