diff --git a/src/decentralizepy/sharing/Choco.py b/src/decentralizepy/sharing/Choco.py index 05de209d3cd80e1e0e8d8c2d2bed8f60c74b6d5a..bf1d77fa26376a379c06f2a93acf53d87a21defe 100644 --- a/src/decentralizepy/sharing/Choco.py +++ b/src/decentralizepy/sharing/Choco.py @@ -310,9 +310,9 @@ class Choco(Sharing): model, dataset, log_dir, - compress=False, - compression_package=None, - compression_class=None + compress, + compression_package, + compression_class ) self.step_size = step_size self.alpha = alpha