diff --git a/src/decentralizepy/sharing/FFT.py b/src/decentralizepy/sharing/FFT.py index 1cc8382722baef3a605b81e1a7eb959b48d27b83..1cdf701f51fb3ae205b0e6016e26f7859104d163 100644 --- a/src/decentralizepy/sharing/FFT.py +++ b/src/decentralizepy/sharing/FFT.py @@ -4,9 +4,9 @@ import os from pathlib import Path from time import time +import numpy as np import torch import torch.fft as fft -import numpy as np from decentralizepy.sharing.Sharing import Sharing diff --git a/src/decentralizepy/sharing/SubSampling.py b/src/decentralizepy/sharing/SubSampling.py index 1e956cd0615f0b93493e8c2f0a57cae1ef7a2dd3..5ec0c44f054eb44debf056189513ab814f620a9f 100644 --- a/src/decentralizepy/sharing/SubSampling.py +++ b/src/decentralizepy/sharing/SubSampling.py @@ -4,7 +4,6 @@ import os from pathlib import Path import torch -import numpy as np from decentralizepy.sharing.Sharing import Sharing @@ -202,7 +201,7 @@ class SubSampling(Sharing): m["seed"] = seed m["alpha"] = alpha - m["params"] = subsample.numpy().astype(np.int32) + m["params"] = subsample.numpy() # logging.info("Converted dictionary to json") self.total_data += len(self.communication.encrypt(m["params"]))