Linting

[Saran-nns]

Analyze source code for bugs and style errors.
If you lint any module from any branch, please mark them done below

• visualizer
• optimizer
• losses
• train
• utils
• inference
• predictive_model/ae
• predictive_models/lstm
• generative_models/vae
• generative_models/vaegan
• base_models/MLPClassifier
• base_models/MLPRegressor
• dataset/dataset
• dataset/example
• dataset/generator
• dataset/geolife
• dataset/pedestrian

[JustinShenk]

I added 'pre-commit' to requirements/dev.txt with flake8 and black, and ran pre-commit run --all-files. I skipped PR process and pushed it to master at b145684, though PR is the standard procedure. There are several errors worth attending to, like variable name complexity:

traja/tests/test_dataset.py:71:1: C901 'test_time_based_sampling_dataloaders_with_short_stride_do_not_overlap' is too complex (20)

Installing pre-commit locally will now run black and flake8 on staged files.

Also highly recommended is to add flake8 linter (and black) to Visual Studio Code or whichever IDE you use.