From d1faff916856fb9a06848505fc27fe540b8ffc95 Mon Sep 17 00:00:00 2001
From: Chong Shen Ng <chong.shen@flower.ai>
Date: Wed, 6 Nov 2024 11:52:02 +0000
Subject: [PATCH] Revert import sort

---
 e2e/docker/client.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/e2e/docker/client.py b/e2e/docker/client.py
index 98a343f239f3..44313c7c3af6 100644
--- 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