Update har_bal.py

josephcslater · Aug 1, 2021 · 165ff16 · 165ff16
1 parent b48172a
commit 165ff16
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/mousai/har_bal.py b/mousai/har_bal.py
@@ -622,7 +622,7 @@ def hb_err(X):
         e = hb_err(X)
         if mask_constant is True:
             X = np.hstack((np.zeros_like(X[:, 0]).reshape(-1, 1), X))
-        amps = np.sqrt(X[:, 1]**2+X[:, 2]**2)*2/X.shape[1]
+        amps = np.sqrt(X[:, 1]**2 + X[:, 2]**2) * 2 / X.shape[1]
         phases = np.arctan2(X[:, 1], -X[:, 2])
     except:  # Catches and raises errors- needs actual error listed.
         print(
@@ -634,7 +634,7 @@ def hb_err(X):
         e = hb_err(X)
         if mask_constant is True:
             X = np.hstack((np.zeros_like(X[:, 0]).reshape(-1, 1), X))
-        amps = np.sqrt(X[:, 1]**2+X[:, 2]**2)*2/X.shape[1]
+        amps = np.sqrt(X[:, 1]**2 + X[:, 2]**2) * 2 / X.shape[1]
         phases = np.arctan2(X[:, 1], -X[:, 2])
 
         raise