fix again

yuxiangw · Nov 6, 2023 · 270f452 · 270f452
1 parent ebd5a24
commit 270f452
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/autodp/transformer_zoo.py b/autodp/transformer_zoo.py
@@ -141,14 +141,14 @@ def update_params(self, mechanism_list):
 
 
 # composition of only Gaussian mechanisms
-class ComposeGaussian(Composition):
+class ComposeGaussian(Transformer):
     """ CompositionGaussian is a specialized composation function of ONLY Guassian mechanisms
     output a Mechanism that represents the composed mechanism"""
     def __init__(self):
         Composition.__init__(self)
         self.name = 'ComposeGaussian'
 
-    def compose(self, mechanism_list, coeff_list, RDP_compose_only=True, BBGHS_conversion=True, fDP_based_conversion=True):
+    def compose(self, mechanism_list, coeff_list):
         # Make sure that the list contains only Gaussian mechanisms
         for mech in mechanism_list:
             assert(isinstance(mech, mechanism_zoo.GaussianMechanism)