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 ⚠️
qdax/core/containers/mels_repertoire.py	76.47%	16 Missing ⚠️
qdax/baselines/pbt.py	76.47%	12 Missing ⚠️
qdax/baselines/sac_pbt.py	96.49%	4 Missing ⚠️
qdax/core/emitters/pbt_me_emitter.py	96.82%	4 Missing ⚠️
.../core_test/containers_test/mels_repertoire_test.py	93.75%	4 Missing ⚠️
qdax/environments/__init__.py	78.57%	3 Missing ⚠️
qdax/environments/exploration_wrappers.py	70.00%	3 Missing ⚠️
tests/default_tasks_test/jumanji_envs_test.py	96.42%	2 Missing ⚠️
tests/core_test/mels_test.py	98.36%	1 Missing ⚠️

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

Additional details and impacted files

@@             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     

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.