Skip to content

Conversation

@sfluegel05
Copy link
Collaborator

This came up for the ensemble - if someone selects 0 models, the PredictionSmoother will fail:

chebai/result/analyse_sem.py", line 224, in __call__
    preds = preds_masked_succ.max(dim=2).values
            ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
IndexError: max(): Expected reduction dim 2 to have non-zero size.

This can be easily avoided by skipping predictions where the label dimension is 0

@sfluegel05 sfluegel05 merged commit 8e12448 into dev Jul 18, 2025
6 checks passed
@sfluegel05 sfluegel05 deleted the fix/prediction-smoothing-empty-labels branch July 18, 2025 16:32
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants