Added endpoint to linspace, logspace and geomspace.

pulse_lib/segments/utility/looping.py

@@ -220,21 +220,30 @@ class loop_obj():
 
 class linspace(loop_obj):
     """docstring for linspace"""
-    def __init__(self, start, stop, n_steps = 50, name = None, unit = None, axis = -1, setvals = None):
+    def __init__(self, start, stop, n_steps = 50,
+                 name = None, unit = None, axis = -1, setvals = None,
+                 endpoint=True):
         super().__init__()
-        super().add_data(np.linspace(start, stop, n_steps), axis = axis, labels = name, units = unit, setvals= setvals)
+        super().add_data(np.linspace(start, stop, n_steps, endpoint=endpoint),
+                         axis = axis, labels = name, units = unit, setvals= setvals)
 
 class logspace(loop_obj):
     """docstring for logspace"""
-    def __init__(self, start, stop, n_steps = 50, name = None, unit = None, axis = -1, setvals = None):
+    def __init__(self, start, stop, n_steps = 50,
+                 name = None, unit = None, axis = -1, setvals = None,
+                 endpoint=True):
         super().__init__()
-        super().add_data(np.logspace(start, stop, n_steps), axis = axis, labels = name, units = unit, setvals = setvals)
+        super().add_data(np.logspace(start, stop, n_steps, endpoint=endpoint),
+                         axis = axis, labels = name, units = unit, setvals = setvals)
 
 class geomspace(loop_obj):
     """docstring for geomspace"""
-    def __init__(self, start, stop, n_steps = 50, name = None, unit = None, axis = -1, setvals = None):
+    def __init__(self, start, stop, n_steps = 50,
+                 name = None, unit = None, axis = -1, setvals = None,
+                 endpoint=True):
         super().__init__()
-        super().add_data(np.geomspace(start, stop, n_steps), axis = axis, labels = name, units = unit, setvals= setvals)
+        super().add_data(np.geomspace(start, stop, n_steps, endpoint=endpoint),
+                         axis = axis, labels = name, units = unit, setvals= setvals)
 
 if __name__ == '__main__':
     lp = loop_obj()