pySTEL: bugfix in THRIFT class

PrincetonUniversity · Jan 9, 2025 · 916a309 · 916a309
1 parent ea42847
commit 916a309
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pySTEL/libstell/thrift.py b/pySTEL/libstell/thrift.py
@@ -191,7 +191,7 @@ def get_vars(self,var,time=0.0):
 
         plot_var = getattr(self,var)
 
-        time = np.array(time)
+        time = np.atleast_1d(time)
 
         idx = [np.argmin(np.abs(self.THRIFT_T-t)) for t in time]
         #idx = np.argmin(np.abs(self.THRIFT_T-time))