Invoke wheel cleaner after the install step #13

Merged
Sergey Sharybin merged 2 commits from wheel_cleanup into main 2024-11-21 09:54:08 +01:00
Showing only changes of commit b5172b29a1 - Show all commits

View File

@ -39,7 +39,7 @@ def disk_free_in_gb(builder: worker.utils.Builder) -> float:
def get_thread_count(thread_memory_in_GB: float) -> int:
num_threads = psutil.cpu_count()
num_threads = psutil.cpu_count() or 1
memory_in_GB = psutil.virtual_memory().total / (1024**3)
return min(int(memory_in_GB / thread_memory_in_GB), num_threads)