🔀 Fix typo & fix return syntax (#56)

* Fix typo in `make_arbitrary_rf.py` * Fix return syntax in `make_arbitrary_rf.py` Co-authored-by: Keerthi Sravan Ravi <[email protected]> Co-authored-by: veldmannm <[email protected]>
imr-framework · Jun 22, 2021 · 2548f0d · 2548f0d
1 parent de4b3eb
commit 2548f0d
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/doc/readthedocs_requirements.txt b/doc/readthedocs_requirements.txt
@@ -0,0 +1 @@
+furo==2021.4.11b34
diff --git a/pypulseq/make_arbitrary_rf.py b/pypulseq/make_arbitrary_rf.py
@@ -69,7 +69,7 @@ def make_arbitrary_rf(signal: np.ndarray, flip_angle: float, bandwidth: float =
     signal = np.squeeze(signal)
     if signal.ndim > 1:
         raise ValueError(f'signal should have ndim=1. Passed ndim={signal.ndim}')
-    signal = signal / bp.abs(np.sum(signal * system.rf_raster_time)) * flip_angle / (2 * np.pi)
+    signal = signal / np.abs(np.sum(signal * system.rf_raster_time)) * flip_angle / (2 * np.pi)
 
     N = len(signal)
     duration = N * system.rf_raster_time
@@ -121,4 +121,7 @@ def make_arbitrary_rf(signal: np.ndarray, flip_angle: float, bandwidth: float =
         rf.t = np.concatenate((rf.t, rf.t[-1] + t_fill))
         rf.signal = np.concatenate((rf.signal, np.zeros(len(t_fill))))
 
-    return rf, gz if return_gz else rf
+    if return_gz:
+        return rf, gz
+    else:
+        return rf