[htcondor] Fix in config hook interface.

riga · Jan 11, 2025 · 3403ef8 · 3403ef8
1 parent fb764f6
commit 3403ef8
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/law/contrib/htcondor/workflow.py b/law/contrib/htcondor/workflow.py
@@ -177,7 +177,7 @@ def cast_dir(
 
         # task hook
         if grouped_submission:
-            c = task.htcondor_job_config(c, list(submit_jobs.keys()), list(submit_jobs.values()))  # type: ignore[arg-type] # noqa
+            c = task.htcondor_job_config(c, list(submit_jobs.keys()), list(submit_jobs.values()))
         else:
             c = task.htcondor_job_config(c, job_num, branches)
 
@@ -383,8 +383,8 @@ def htcondor_create_job_file_factory(self, **kwargs) -> HTCondorJobFileFactory:
     def htcondor_job_config(
         self,
         config: HTCondorJobFileFactory.Config,
-        job_num: int,
-        branches: list[int],
+        job_num: int | list[int],
+        branches: list[int] | list[list[int]],
     ) -> HTCondorJobFileFactory.Config:
         return config