summaryrefslogtreecommitdiffstats
path: root/python/astra/astra.py
diff options
context:
space:
mode:
Diffstat (limited to 'python/astra/astra.py')
-rw-r--r--python/astra/astra.py24
1 files changed, 22 insertions, 2 deletions
diff --git a/python/astra/astra.py b/python/astra/astra.py
index 26b1ff0..61c26ee 100644
--- a/python/astra/astra.py
+++ b/python/astra/astra.py
@@ -49,10 +49,30 @@ def version(printToScreen=False):
"""
return a.version(printToScreen)
-def set_gpu_index(idx):
+def set_gpu_index(idx, memory=0):
"""Set default GPU index to use.
:param idx: GPU index
:type idx: :class:`int`
"""
- a.set_gpu_index(idx)
+ a.set_gpu_index(idx, memory)
+
+def delete(ids):
+ """Delete an astra object.
+
+ :param ids: ID or list of ID's to delete.
+ :type ids: :class:`int` or :class:`list`
+
+ """
+ return a.delete(ids)
+
+def info(ids):
+ """Print info about an astra object.
+
+ :param ids: ID or list of ID's to show.
+ :type ids: :class:`int` or :class:`list`
+
+ """
+ return a.info(ids)
+
+