Fix imports jumpy and formatting issues

[Lookatator]

• Fixes the bugs of imports in the develop branch.
• Fix formatting issues in develop.

[codecov-commenter]

Codecov Report

Attention: 107 lines in your changes are missing coverage. Please review.

Comparison is base (f6a0b71) 92.31% compared to head (17bec85) 92.28%. Report is 26 commits behind head on develop.

Files	Patch %	Lines
qdax/environments/humanoidtrap.py	18.30%	58 Missing :warning:
qdax/core/containers/mels_repertoire.py	76.47%	16 Missing :warning:
qdax/baselines/pbt.py	76.47%	12 Missing :warning:
qdax/baselines/sac_pbt.py	96.49%	4 Missing :warning:
qdax/core/emitters/pbt_me_emitter.py	96.82%	4 Missing :warning:
.../core_test/containers_test/mels_repertoire_test.py	93.75%	4 Missing :warning:
qdax/environments/__init__.py	78.57%	3 Missing :warning:
qdax/environments/exploration_wrappers.py	70.00%	3 Missing :warning:
tests/default_tasks_test/jumanji_envs_test.py	96.42%	2 Missing :warning:
tests/core_test/mels_test.py	98.36%	1 Missing :warning:

:exclamation: Your organization needs to install the Codecov GitHub app to enable full functionality.

Additional details and impacted files

```diff @@ Coverage Diff @@ ## develop #163 +/- ## =========================================== - Coverage 92.31% 92.28% -0.03% =========================================== Files 103 120 +17 Lines 5816 7055 +1239 =========================================== + Hits 5369 6511 +1142 - Misses 447 544 +97 ```

:umbrella: View full report in Codecov by Sentry.
:loudspeaker: Have feedback on the report? Share it here.