add make_it_personal

Author: nerdai

fl4health/mixins/personalized/__init__.py

@@ -1,11 +1,10 @@
-from .ditto import DittoPersonalizedMixin
-from fl4health.clients.basic_client import BasicClient
-
 from abc import ABC, abstractmethod
 from enum import Enum
 
 from fl4health.clients.basic_client import BasicClient
 
+from .ditto import DittoPersonalizedMixin
+
 
 class PersonalizedModes(str, Enum):
     DITTO = "ditto"

fl4health/mixins/personalized/ditto.py

@@ -6,9 +6,9 @@
 
 import torch
 import torch.nn as nn
-from torch.optim import Optimizer
 from flwr.common.logger import log
 from flwr.common.typing import Config, NDArrays, Scalar
+from torch.optim import Optimizer
 
 from fl4health.mixins.adaptive_drift_contrained import AdaptiveDriftConstrainedMixin
 from fl4health.mixins.personalized.base import BasePersonalizedMixin