Skip to content

Comments

Optimize prob() and cdf() methods#17

Merged
famura merged 4 commits intomainfrom
feat/optimizations
Jan 5, 2026
Merged

Optimize prob() and cdf() methods#17
famura merged 4 commits intomainfrom
feat/optimizations

Conversation

@famura
Copy link
Owner

@famura famura commented Jan 5, 2026

The new solution uses torch.searchsorted together with torch.gather to avoid creating the biog boolean intermediate tensors

@famura famura self-assigned this Jan 5, 2026
@famura famura added the type: refactor Refactoring code without functional changes label Jan 5, 2026
@github-actions github-actions bot added area: source Core source code size: xs labels Jan 5, 2026
@github-actions
Copy link
Contributor

github-actions bot commented Jan 5, 2026

Coverage

Coverage Report
FileStmtsMissCoverMissing
__init__.py20100% 
binned_logit_cdf.py138298%98, 317
TOTAL140298% 

Tests Skipped Failures Errors Time
492 226 💤 0 ❌ 0 🔥 10.545s ⏱️

@famura famura merged commit fbb09e7 into main Jan 5, 2026
3 checks passed
@famura famura deleted the feat/optimizations branch January 5, 2026 16:29
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

area: source Core source code size: xs type: refactor Refactoring code without functional changes

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant