`tqdm` Experimental warning is raised when package is imported

Upon import of `trainax`, the following warning is raised:

```python
TqdmExperimentalWarning: Using `tqdm.autonotebook.tqdm` in notebook mode. Use `tqdm.tqdm` instead to force console mode (e.g. in jupyter console)
  from tqdm.autonotebook import tqdm
```