Skip to content

Commit 095c8fd

Browse files
authored
Merge pull request #118 from web-genie-ai/feat/logs
chore: add logs
2 parents 7d53064 + c574ca1 commit 095c8fd

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

webgenie/rewards/visual_reward/visual_reward.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -80,6 +80,7 @@ async def reward(self, task: Task, solutions: List[Solution]) -> np.ndarray:
8080
current_work_dir = f"{WORK_DIR}/task_{timestamp}_{task.task_id}"
8181
os.makedirs(current_work_dir, exist_ok=True)
8282

83+
bt.logging.info(f"The number of cpu cores: {os.cpu_count()}")
8384
# Use ProcessPoolExecutor for parallel processing
8485
with multiprocessing.Pool(processes=os.cpu_count()) as pool:
8586
# Convert solutions into chunks for parallel processing

0 commit comments

Comments
 (0)