fix typo on initialization

RobinMagnet · Aug 29, 2023 · 4967f89 · 4967f89
1 parent 249ee9a
commit 4967f89
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/DiscreteOpt.py b/DiscreteOpt.py
@@ -182,7 +182,7 @@ def fit(self, p2p_21=None, p2p_12=None, n_jobs=1, verbose=False):
                 self.solve_p2p_12(n_jobs=n_jobs)
 
 
-class SmoothDiscreteOptimization2(DiscreteOptimization):
+class SmoothDiscreteOptimization(DiscreteOptimization):
     """
     This class extends the DiscreteOptimization class with primal energies
     """
@@ -311,7 +311,7 @@ def _initialize(self, p2p_21=None, p2p_12=None):
         p2p_21 : (n2,) array - initial map from mesh2 to mesh1
         """
         # Use the function from the DiscreteOptimization class
-        super()._initialize(p2p_21=None, p2p_12=None)
+        super()._initialize(p2p_21=p2p_21, p2p_12=p2p_12)
 
         self.Y_12 = None
         self.Y_21 = None