multi-pilot changes

radical-cybertools · Jul 16, 2024 · 0c97a91 · 0c97a91
1 parent 211ef98
commit 0c97a91
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/examples/pq_multi_dask.py b/examples/pq_multi_dask.py
@@ -51,11 +51,11 @@ def circuit(parameters):
         # Submit tasks to pcs
         tasks = []
         for i in range(1000):
-            k = pcs.submit_task(f"task_pennylane-{i}", pennylane_quantum_circuit)
+            k = pcs.submit_task(pennylane_quantum_circuit, task_name = f"task_pennylane-{i}" )
             tasks.append(k)
 
         for i in range(1000):
-            k = pcs.submit_task(f"task_pennylane-{i}", pennylane_quantum_circuit, pilot=pilots[0])
+            k = pcs.submit_task(pennylane_quantum_circuit, task_name = f"task_pennylane-{i}", pilot=pilots[0])
             tasks.append(k)
 
 

diff --git a/pilot/pilot_compute_service.py b/pilot/pilot_compute_service.py
@@ -137,7 +137,7 @@ def get_client(self):
         """
         return self.cluster_manager.get_client()
 
-    def submit_task(self, task_name, func, *args, **kwargs):
+    def submit_task(self, func, *args, **kwargs):
         pilot_scheduled = 'ANY'
 
         if kwargs.get("pilot"):
@@ -146,6 +146,11 @@ def submit_task(self, task_name, func, *args, **kwargs):
             pilot_scheduled = kwargs["pilot"]
             del kwargs["pilot"]
 
+        task_name = kwargs.get("task_name", f"task-{uuid.uuid4()}")
+        if kwargs.get("task_name"):
+            del kwargs["task_name"]
+
+
         if not self.client:
             self.client = self.get_client()