resource updates for coloc run

Author: akhanf

spimquant/workflow/rules/counts.smk

@@ -90,7 +90,7 @@ rule coloc_per_voxel_template:
         ),
     threads: 16
     resources:
-        mem_mb=15000,
+        mem_mb=64000,
         runtime=30,
     script:
         "../scripts/coloc_per_voxel_template.py"

spimquant/workflow/rules/regionprops.smk

@@ -134,10 +134,10 @@ rule colocalize_regionprops:
             suffix="coloc.parquet",
             **inputs["spim"].wildcards,
         ),
-    threads: 1
+    threads: 16
     resources:
-        mem_mb=1500,
-        runtime=30,
+        mem_mb=64000,
+        runtime=180,
     script:
         "../scripts/compute_colocalization.py"
 
@@ -180,9 +180,9 @@ rule sample_at_vessel_sdt:
             suffix="regionprops.parquet",
             **inputs["spim"].wildcards,
         ),
-    threads: 1
+    threads: 8
     resources:
         mem_mb=32000,
-        runtime=30,
+        runtime=360,
     script:
         "../scripts/sample_at_points.py"

spimquant/workflow/rules/segstats.smk

@@ -39,9 +39,9 @@ rule map_regionprops_to_atlas_rois:
                 **inputs["spim"].wildcards,
             )
         ),
-    threads: 1
+    threads: 4
     resources:
-        mem_mb=1500,
+        mem_mb=32000,
         runtime=15,
     script:
         "../scripts/map_atlas_to_regionprops.py"
@@ -84,9 +84,9 @@ rule map_coloc_to_atlas_rois:
                 **inputs["spim"].wildcards,
             )
         ),
-    threads: 1
+    threads: 4
     resources:
-        mem_mb=1500,
+        mem_mb=32000,
         runtime=15,
     script:
         "../scripts/map_atlas_to_coloc.py"