Improve 'declearn.model' docstring and add some missing '__all__'.

declearn/model/__init__.py

@@ -19,8 +19,25 @@
 
 This declearn submodule provides with:
 * Model and Vector abstractions, used as an API to design FL algorithms
-* Submodules implementing interfaces to curretnly supported frameworks 
-and models.
+* Submodules implementing interfaces to various frameworks and models.
+
+The automatically-imported submodules implemented here are:
+* api: Model and Vector abstractions' defining module.
+    - Model: abstract API to interface framework-specific models.
+    - Vector: abstract API for data tensors containers.
+* sklearn: scikit-learn based or oriented tools
+    - NumpyVector: Vector for numpy array data structures.
+    - SklearnSGDModel: Model for scikit-learn's SGDClassifier and SGDRegressor.
+
+The optional-dependency-based submodules that may be manually imported are:
+* tensorflow: tensorflow-interfacing tools
+    - TensorflowModel: Model to wrap any tensorflow-keras Layer model.
+    - TensorflowOptiModule: Hacky OptiModule to wrap a keras Optimizer.
+    - TensorflowVector: Vector for tensorflow Tensor and IndexedSlices.
+* torch: pytorch-interfacing tools
+    - TorchModel: Model to wrap any torch Module model.
+    - TorchOptiModule: Hacky OptiModule to wrap a torch Optimizer.
+    - TorchVector: Vector for torch Tensor objects.
 """
 
 from . import api

declearn/model/api/__init__.py

@@ -15,7 +15,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-"""Model Vector abstractions submodule."""
+"""Model and Vector abstractions submodule."""
 
 from ._vector import Vector, register_vector_type
 from ._model import Model

declearn/model/sklearn/_np_vec.py

@@ -25,6 +25,7 @@ from typing_extensions import Self  # future: import from typing (Py>=3.11)
 
 from declearn.model.api._vector import Vector, register_vector_type
 
+
 __all__ = [
     "NumpyVector",
 ]

declearn/model/tensorflow/_vector.py

@@ -37,6 +37,11 @@ from declearn.model.tensorflow.utils import (
 from declearn.utils import get_device_policy
 
 
+__all__ = [
+    "TensorflowVector",
+]
+
+
 @register_vector_type(tf.Tensor, EagerTensor, tf.IndexedSlices)
 class TensorflowVector(Vector):
     """Vector subclass to store tensorflow tensors.

declearn/model/torch/_vector.py

@@ -29,6 +29,11 @@ from declearn.model.torch.utils import select_device
 from declearn.utils import get_device_policy
 
 
+__all__ = [
+    "TorchVector",
+]
+
+
 @register_vector_type(torch.Tensor)
 class TorchVector(Vector):
     """Vector subclass to store PyTorch tensors.