From 98e938bc23ea41fea79f8a2a21eb2715e6e8c621 Mon Sep 17 00:00:00 2001
From: Rishi Sharma <rishi.sharma@epfl.ch>
Date: Wed, 9 Mar 2022 15:39:59 +0100
Subject: [PATCH] Remove unnecessary imports

---
 src/decentralizepy/datasets/Celeba.py  | 1 -
 src/decentralizepy/datasets/Femnist.py | 1 -
 src/decentralizepy/sharing/Sharing.py  | 2 --
 3 files changed, 4 deletions(-)

diff --git a/src/decentralizepy/datasets/Celeba.py b/src/decentralizepy/datasets/Celeba.py
index 61a77c6..ab077b3 100644
--- a/src/decentralizepy/datasets/Celeba.py
+++ b/src/decentralizepy/datasets/Celeba.py
@@ -3,7 +3,6 @@ import logging
 import os
 from collections import defaultdict
 
-import matplotlib.pyplot as plt
 import numpy as np
 import torch
 import torch.nn.functional as F
diff --git a/src/decentralizepy/datasets/Femnist.py b/src/decentralizepy/datasets/Femnist.py
index 23e9704..221e6f2 100644
--- a/src/decentralizepy/datasets/Femnist.py
+++ b/src/decentralizepy/datasets/Femnist.py
@@ -3,7 +3,6 @@ import logging
 import os
 from collections import defaultdict
 
-import matplotlib.pyplot as plt
 import numpy as np
 import torch
 import torch.nn.functional as F
diff --git a/src/decentralizepy/sharing/Sharing.py b/src/decentralizepy/sharing/Sharing.py
index 66ecae3..85fc07b 100644
--- a/src/decentralizepy/sharing/Sharing.py
+++ b/src/decentralizepy/sharing/Sharing.py
@@ -1,8 +1,6 @@
 import logging
-import pickle
 from collections import deque
 
-import numpy
 import torch
 
 
-- 
GitLab