Add flwr_datasets as known_first_party

adap · Nov 6, 2024 · 6021a0f · 6021a0f
1 parent 9c60283
commit 6021a0f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/e2e/docker/client.py b/e2e/docker/client.py
@@ -4,12 +4,12 @@
 import torch
 import torch.nn as nn
 import torch.nn.functional as F
-from flwr_datasets import FederatedDataset
 from torch.utils.data import DataLoader, Subset
 from torchvision.transforms import Compose, Normalize, ToTensor
 
 from flwr.client import ClientApp, NumPyClient
 from flwr.common import Context
+from flwr_datasets import FederatedDataset
 
 # #############################################################################
 # 1. Regular PyTorch pipeline: nn.Module, train, test, and DataLoader

diff --git a/pyproject.toml b/pyproject.toml
@@ -147,7 +147,7 @@ extend_exclude = [
 
 [tool.isort]
 profile = "black"
-known_first_party = ["flwr", "flwr_tool"]
+known_first_party = ["flwr", "flwr_tool", "flwr_datasets"]
 
 [tool.black]
 line-length = 88