add max_iter_harmony to self

This fixes a bug in pull request #8 that caused this error: AttributeError: 'Harmony' object has no attribute 'max_iter_harmony'
slowkow · Aug 11, 2020 · d30739a · d30739a
1 parent 61d6fa1
commit d30739a
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/harmonypy/harmony.py b/harmonypy/harmony.py
@@ -156,6 +156,7 @@ def __init__(
         self.sigma_prior     = sigma
         self.block_size      = block_size
         self.K               = K                # number of clusters
+        self.max_iter_harmony = max_iter_harmony
         self.max_iter_kmeans = max_iter_kmeans
         self.verbose         = verbose
         self.theta           = theta