compute args

danionella · Sep 4, 2024 · d3d03c6 · d3d03c6
1 parent 4145a0d
commit d3d03c6
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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: