Mismatch between doc strings and code in log_prob function

[rakaar]

Here in doc strings of log_prob function method - https://github.com/sbi-dev/sbi/blob/main/sbi/neural_nets/estimators/nflows_flow.py#L84C46-L84C81

Args:
            input: Inputs to evaluate the log probability on. Of shape
                `(sample_dim, batch_dim, *event_shape)`.
            condition: Conditions of shape `(sample_dim, batch_dim, *event_shape)`.

Condition variable is expected to be of format (sample_dim, batch_dim, event_shape)
Dimension 0 - Sample
Dimension 1 - Batch
Dimension 2 - Event

But in function definition in this line - https://github.com/sbi-dev/sbi/blob/main/sbi/neural_nets/estimators/nflows_flow.py#L94

...
condition_batch_dim = condition.shape[0]
...

0th dimension is considered as batch dimension, while it should be 1st dimension as per the doc strings.

The mismatch might be lead to usage of incorrect tensor shapes.

[janfb]

Hi @rakaar , thanks for reporting this. It's likely a typo in the docs, as we usually assume that the condition has shape (batch, *event_shape).
We will fix it in the upcoming release.