Skip to content

Commit 50a98b9

Browse files
fix: auto-detect similarity workers (default 0 = use cpu_cap)
1 parent 45e0084 commit 50a98b9

2 files changed

Lines changed: 11 additions & 6 deletions

File tree

extropy/cli/commands/network.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -76,10 +76,10 @@ def network_command(
7676
help="Blocking attribute (repeatable). If omitted, auto-selects top attributes",
7777
),
7878
similarity_workers: int = typer.Option(
79-
1,
79+
0,
8080
"--similarity-workers",
81-
min=1,
82-
help="Worker processes for similarity computation",
81+
min=0,
82+
help="Worker processes for similarity computation (0 = auto-detect)",
8383
),
8484
similarity_chunk_size: int = typer.Option(
8585
64,

extropy/utils/resource_governor.py

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -90,9 +90,9 @@ def recommend_workers(
9090
requested_workers: int,
9191
memory_per_worker_gb: float,
9292
) -> int:
93-
requested_workers = max(1, int(requested_workers))
93+
requested_workers = max(0, int(requested_workers))
9494
if self.resource_mode != "auto":
95-
return requested_workers
95+
return max(1, requested_workers)
9696

9797
snap = self.snapshot()
9898
cpu_cap = (
@@ -103,7 +103,12 @@ def recommend_workers(
103103
if self.safe_auto_workers:
104104
cpu_cap = min(cpu_cap, 8)
105105

106-
return max(1, min(requested_workers, cpu_cap, mem_cap))
106+
auto_cap = max(1, min(cpu_cap, mem_cap))
107+
108+
if requested_workers == 0:
109+
return auto_cap
110+
111+
return max(1, min(requested_workers, auto_cap))
107112

108113
def recommend_chunk_size(
109114
self,

0 commit comments

Comments
 (0)