Fixed Python API

It4innovations · Dec 12, 2023 · 1657ec0 · 1657ec0
1 parent a7caf08
commit 1657ec0
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/crates/pyhq/src/client/job.rs b/crates/pyhq/src/client/job.rs
@@ -233,9 +233,8 @@ pub fn wait_for_jobs_impl(
         let mut response: JobInfoResponse;
 
         loop {
-            let selector = IdSelector::Specific(IntArray::from_ids(
-                remaining_job_ids.iter().copied().collect(),
-            ));
+            let selector =
+                IdSelector::Specific(IntArray::from_ids(remaining_job_ids.iter().copied()));
 
             response = hyperqueue::rpc_call!(
                 ctx.session.connection(),
@@ -311,7 +310,7 @@ pub fn get_failed_tasks_impl(
 ) -> PyResult<FailedTaskMap> {
     run_future(async move {
         let message = FromClientMessage::JobDetail(JobDetailRequest {
-            job_id_selector: IdSelector::Specific(IntArray::from_ids(job_ids)),
+            job_id_selector: IdSelector::Specific(IntArray::from_ids(job_ids.into_iter())),
             task_selector: Some(TaskSelector {
                 id_selector: TaskIdSelector::All,
                 status_selector: TaskStatusSelector::Specific(vec![Status::Failed]),