diff --git a/wetb/utils/cluster_tools/cluster_resource.py b/wetb/utils/cluster_tools/cluster_resource.py index 78bc9eb9605f33e2133108500b98044201da71a9..4c5cc1512319d1ba47fbdf7354787f9c0ad3dd91 100644 --- a/wetb/utils/cluster_tools/cluster_resource.py +++ b/wetb/utils/cluster_tools/cluster_resource.py @@ -69,9 +69,9 @@ class LocalResource(Resource): try: return psutil.Process(i).name except psutil._error.AccessDenied: - pass + return "" no_cpu = multiprocessing.cpu_count() cpu_free = (1 - psutil.cpu_percent(.5) / 100) * no_cpu - no_current_process = len([i for i in psutil.get_pid_list() if name(i) == self.process_name]) + no_current_process = len([i for i in psutil.get_pid_list() if name(i).lower().startswith(self.process_name.lower())]) return no_cpu, cpu_free, no_current_process