fix: count sampled tokens with np.count_nonzero - #4251

Bugbot Analysis Progress (53s elapsed)

✅ Gathered PR context (2s)
✅ Completed bug detection (51s)
✅ Posted analysis results (0s)

Final Result: Bugbot completed review - no issues found! ✅

Request ID: 8ae2929e-1bfc-4eed-a3ae-b7b57ddeb139