Clean up import list of learning config

JMGaljaard · Sep 4, 2022 · dcd518e · dcd518e
1 parent c884499
commit dcd518e
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/fltk/util/config/learning_config.py b/fltk/util/config/learning_config.py
@@ -14,12 +14,8 @@
 # noinspection PyProtectedMember
 from torch.nn.modules.loss import _Loss
 
-from fltk.util.config.definitions import DataSampler, Loss, get_loss_function
-from fltk.util.config.definitions.aggregate import Aggregations
-from fltk.util.config.definitions.dataset import Dataset
-from fltk.util.config.definitions.logging import LogLevel
-from fltk.util.config.definitions.net import Nets
-from fltk.util.config.definitions.optim import Optimizations
+from fltk.util.config.definitions import (DataSampler, Loss, get_loss_function, Aggregations, Dataset, LogLevel, Nets,
+                                          Optimizations)
 
 
 def _eval_decoder(obj: Union[str, T]) -> Union[Any, T]: