Client bug fix

src/maggma/cli/__init__.py

@@ -57,6 +57,7 @@
     " mrun will find an open port if None is provided to the manager",
 )
 @click.option("-N", "--num-chunks", "num_chunks", default=0, type=int)
+@click.option("-w", "--num-workers", "num_workers", default=0, type=int)
 @click.option(
     "--no_bars", is_flag=True, help="Turns of Progress Bars for headless operations"
 )
@@ -100,7 +101,7 @@ def run(
                 root.critical(f"Using random port for mrun manager: {port}")
             loop.run_until_complete(
                 manager(
-                    url=url, port=port, builders=builder_objects, num_chunks=num_chunks
+                    url=url, port=port, builders=builder_objects, num_chunks=num_chunks, num_workers=num_workers
                 )
             )
         else:

src/maggma/cli/distributed.py

@@ -32,6 +32,10 @@ async def manager(
     """
     logger = getLogger("Manager")
 
+    if not num_chunks and num_workers:
+        raise ValueError("Both num_chunks and num_workers must be non-zero")
+
+
     logger.info(f"Binding to Manager URL {url}:{port}")
     context = zmq.Context()
     socket = context.socket(REP)