Hervé BREDIN

Or ``` pip install diart # does not install pyannote pip install diart[pretrained] # installs pyannote ```

In case you are wondering how this is done (I am sure you know already :))... The following `setup.py` section will allow `pip install pyannote.core[notebook]` for instance... https://github.com/pyannote/pyannote-core/blob/5399294a65e6ad81ae4da5a906d84d515b832822/setup.py#L67-L78

Alternative: [`rich`](https://github.com/Textualize/rich/discussions/884#discussioncomment-269200)

As mentioned [here](https://github.com/openjournals/joss-reviews/issues/4584#issuecomment-1224234397), I cannot start working on this review before the end of September. I apologize if my initial statement was not clear enough...

## Review checklist for @hbredin ### Conflict of interest - [x] I confirm that I have read the [JOSS conflict of interest (COI) policy](https://github.com/openjournals/joss/blob/master/COI.md) and that: I have no COIs...

Did you do something to solve this issue?

Thanks for reporting this. This is indeed expected but should definitely be fixed -- but unfortunately not at the top of my priority list (as I have been using speechbrain...

> Sure, I can give it a try! I suppose restoring the previous behavior relying on `default_collate` should be enough? "Enough", I am not 100% sure. But it definitely is...

Installing `torch-audiomentations` from the master branch should fix this.

`pip install https://github.com/asteroid-team/torch-audiomentations/archive/master.zip` You might also need to pass option `output_type="dict"` to `AddBackgroundNoise`. This will become the default behavior in future version of `torch-audiomentations`.