convert triang output to scalar

segasai · Nov 16, 2024 · ae91e7d · ae91e7d
1 parent 72ab328
commit ae91e7d
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/py/rvspecfit/spec_inter.py b/py/rvspecfit/spec_inter.py
@@ -54,6 +54,8 @@ def __call__(self, p):
                 b1[:, None]).sum(axis=0)
         if self.exp:
             spec = np.exp(spec)
+        if spec.size == 1:
+            spec = float(spec[0])
         return spec