minor formatting fixes

adap · Oct 22, 2023 · b86a5ef · b86a5ef
1 parent dbce27a
commit b86a5ef
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/baselines/hfedxgboost/hfedxgboost/main.py b/baselines/hfedxgboost/hfedxgboost/main.py
@@ -9,14 +9,14 @@
 import flwr as fl
 import hydra
 import torch
+import wandb
 from flwr.common import Scalar
 from flwr.server.app import ServerConfig
 from flwr.server.client_manager import SimpleClientManager
 from hydra.utils import instantiate
 from omegaconf import DictConfig, OmegaConf
 from torch.utils.data import TensorDataset
 
-import wandb
 from hfedxgboost.client import FlClient
 from hfedxgboost.dataset import divide_dataset_between_clients, load_single_dataset
 from hfedxgboost.server import FlServer, serverside_eval

diff --git a/baselines/hfedxgboost/hfedxgboost/server.py b/baselines/hfedxgboost/hfedxgboost/server.py
@@ -11,6 +11,7 @@
 from typing import Dict, List, Optional, Tuple, Union
 
 import flwr as fl
+import wandb
 from flwr.common import EvaluateRes, FitRes, Parameters, Scalar, parameters_to_ndarrays
 from flwr.common.logger import log
 from flwr.common.typing import GetParametersIns
@@ -23,7 +24,6 @@
 from torch.utils.data import DataLoader
 from xgboost import XGBClassifier, XGBRegressor
 
-import wandb
 from hfedxgboost.models import CNN
 from hfedxgboost.utils import EarlyStop, single_tree_preds_from_each_client, test