Features.freeze: fix starting where boolean array

PAHFIT · Nov 30, 2024 · 6defaa9 · 6defaa9
1 parent 353e862
commit 6defaa9
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 8 deletions.
diff --git a/pahfit/features/features.py b/pahfit/features/features.py
@@ -349,7 +349,7 @@ def freeze(self, param: str, where=None, thaw=False, **keys):
             return
         if where is None:
             if keys:
-                where = np.zeros_like(self.colnames, dtype=np.bool)
+                where = np.zeros((len(self[param])), dtype=np.bool)
                 for k, v in keys.items():
                     if k in self.colnames:
                         col = self[k]

diff --git a/pahfit/model.py b/pahfit/model.py
@@ -138,13 +138,8 @@ def __repr__(self):
     def _repr_html_(self):
         return self._status_message() + self.features._repr_html_()
 
-    def guess(
-        self,
-        spec: Spectrum1D,
-        redshift=None,
-        integrate_line_flux=False,
-        calc_line_fwhm=True,
-    ):
+    def guess(self, spec: Spectrum1D, redshift=None, integrate_line_flux=False,
+              calc_line_fwhm=True):
         """Make an initial guess of the physics, based on the given
         observational data.