Fix 'messaging' imports under 'training' module.

declearn/training/_manager.py

@@ -23,8 +23,8 @@ from typing import Any, Dict, List, Optional, Tuple, Union
 import numpy as np
 import tqdm
 
+from declearn import messaging
 from declearn.aggregator import Aggregator
-from declearn.communication import messaging
 from declearn.dataset import Dataset
 from declearn.metrics import (
     MeanMetric,

declearn/training/dp/_manager.py

@@ -23,8 +23,8 @@ from typing import List, Optional, Tuple, Union
 from opacus.accountants import IAccountant, create_accountant  # type: ignore
 from opacus.accountants.utils import get_noise_multiplier  # type: ignore
 
+from declearn import messaging
 from declearn.aggregator import Aggregator
-from declearn.communication import messaging
 from declearn.dataset import Dataset
 from declearn.metrics import MetricInputType, MetricSet
 from declearn.model.api import Model

test/training/test_train_manager.py

@@ -22,8 +22,8 @@ from typing import Any, Iterator, Optional
 
 import numpy
 
+from declearn import messaging
 from declearn.aggregator import Aggregator
-from declearn.communication import messaging
 from declearn.dataset import Dataset
 from declearn.metrics import Metric, MetricSet
 from declearn.model.api import Model, Vector

test/training/test_train_manager_dp.py

@@ -28,7 +28,7 @@ try:
 except ModuleNotFoundError:
     pytest.skip("Opacus is unavailable", allow_module_level=True)
 
-from declearn.communication import messaging
+from declearn import messaging
 from declearn.dataset import DataSpecs
 from declearn.optimizer.modules import GaussianNoiseModule
 from declearn.training.dp import DPTrainingManager