From d3d03c6df03683217ddf8480d26b138612cd7cd8 Mon Sep 17 00:00:00 2001
From: Benjamin <bjudkewitz@users.noreply.github.com>
Date: Wed, 4 Sep 2024 11:57:00 +0200
Subject: [PATCH] compute args

---
 dejaq/parallel.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/dejaq/parallel.py b/dejaq/parallel.py
index 0047da8..7753d8a 100644
--- a/dejaq/parallel.py
+++ b/dejaq/parallel.py
@@ -65,10 +65,11 @@ def close(self):
         self.in_queue = None
 
     def compute(self, progress=True, ndarray=True, **kwargs):
-        ''' Computes the results of the lazymap and returns them as a list.
+        ''' Computes the results of the lazymap and returns them as a list or ndarray
 
         Args:
             progress (bool): whether to show a tqdm progress bar (default: True)
+            ndarray (bool): whether to try to return the results as a numpy array (default: True)
             **kwargs: optional, being passed to tqqdm
 
         Returns: