From 356c4b18b991e0386644a4816202df431859ab4d Mon Sep 17 00:00:00 2001 From: Dacheng Xu Date: Wed, 25 Sep 2024 05:25:08 -0400 Subject: [PATCH] Set `request_cpus` as integer (#220) --- alea/submitters/htcondor.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/alea/submitters/htcondor.py b/alea/submitters/htcondor.py index 156f0d2..257558f 100644 --- a/alea/submitters/htcondor.py +++ b/alea/submitters/htcondor.py @@ -62,7 +62,7 @@ def __init__(self, *args, **kwargs): # Resources configurations self.request_cpus = self.htcondor_configurations.pop("request_cpus", 1) - self.request_memory = self.htcondor_configurations.pop("request_memory", 2000) + self.request_memory = self.htcondor_configurations.pop("request_memory", 2_000) self.request_disk = self.htcondor_configurations.pop("request_disk", 2_000) self.combine_disk = self.htcondor_configurations.pop("combine_disk", 20_000) @@ -432,7 +432,7 @@ def _initialize_job( # no other attributes on a local job return job - job.add_profiles(Namespace.CONDOR, "request_cpus", f"{cores}") + job.add_profiles(Namespace.CONDOR, "request_cpus", cores) # Set memory and disk requirements # If the job fails, retry with more memory and disk