fixed most linting errors

scripts/run_train_nerf.py

@@ -1,21 +1,23 @@
-import os
-from typing import Dict
-from mattport.nerf.trainer import Trainer
+"""
+run_train_nerf.py
+"""
 import logging
 import hydra
 from omegaconf import DictConfig
 
+from mattport.nerf.trainer import Trainer
+
+
 logging.basicConfig(format="[%(filename)s:%(lineno)d] %(message)s", level=logging.DEBUG)
 
 
 @hydra.main(config_path="../configs", config_name="config.yaml")
 def main(config: DictConfig):
+    """Main function."""
     trainer = Trainer(config)
-    print("trainer!")
-
     trainer.setup_dataset()
-    # trainer.setup_graph()
-    # trainer.setup_optimizer()
+    trainer.setup_graph()
+    trainer.setup_optimizer()
 
 
 if __name__ == "__main__":