Fix covariate and nested prints

Labo-Lacourse · Aug 11, 2023 · 4116629 · 4116629
1 parent 50d0dad
commit 4116629
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/stepmix/emission/covariate.py b/stepmix/emission/covariate.py
@@ -41,6 +41,7 @@ def __init__(
         self.lr = lr
         self.intercept = intercept
         self.method = method
+        self.model_str = "covariate"
 
     def check_parameters(self):
         super().check_parameters()

diff --git a/stepmix/emission/nested.py b/stepmix/emission/nested.py
@@ -124,6 +124,7 @@ def print_parameters(self, indent=1, feature_names=None):
             f_i = feature_names[i : i + range_]
             m.print_parameters(indent, model_name=name, feature_names=f_i)
             i += range_
+            print("\n")
 
     @property
     def n_parameters(self):