[tests] Fix docformatting

src/ptbench/configs/datasets/__init__.py

src/ptbench/data/dataset.py

src/ptbench/data/loader.py

src/ptbench/data/transforms.py

src/ptbench/data/utils.py

src/ptbench/engine/evaluator.py

src/ptbench/engine/predictor.py

src/ptbench/engine/trainer.py

src/ptbench/models/alexnet.py

src/ptbench/models/densenet.py

src/ptbench/models/densenet_rs.py

src/ptbench/models/logistic_regression.py

src/ptbench/models/pasa.py

src/ptbench/models/signs_to_tb.py

src/ptbench/scripts/aggregpred.py

src/ptbench/scripts/compare.py

src/ptbench/scripts/config.py

src/ptbench/scripts/dataset.py

src/ptbench/scripts/evaluate.py

src/ptbench/scripts/predtojson.py