Skip to content

Commit 2c22432

Browse files
committed
Fix cli arguments
1 parent c9f07b8 commit 2c22432

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

giskard/commands/cli_worker.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -109,7 +109,7 @@ def start_command(url: AnyHttpUrl, is_server, api_key, is_daemon, hf_token, nb_w
109109
)
110110
api_key = initialize_api_key(api_key, is_server)
111111
hf_token = initialize_hf_token(hf_token, is_server)
112-
_start_command(is_server, url, api_key, is_daemon, hf_token, nb_workers)
112+
_start_command(is_server, url, api_key, is_daemon, hf_token, int(nb_workers) if nb_workers is not None else None)
113113

114114

115115
def initialize_api_key(api_key, is_server):

giskard/utils/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -31,8 +31,8 @@ def __init__(self):
3131
def start(self, max_workers: int = None):
3232
if self.pool is not None:
3333
return
34-
LOGGER.info("Starting worker pool...")
3534
self.max_workers = max(max_workers, settings.min_workers) if max_workers is not None else os.cpu_count()
35+
LOGGER.info("Starting worker pool with %s workers...", self.max_workers)
3636
self.pool = ProcessPoolExecutor(max_workers=self.max_workers)
3737
LOGGER.info("Pool is started")
3838

0 commit comments

Comments
 (0)