sorting imports on save; cleanup imports sorting; add isort check (#207)

nerfactory/utils/colors.py

@@ -14,8 +14,9 @@
 
 """Common Colors"""
 from typing import Union
-from omegaconf import ListConfig
+
 import torch
+from omegaconf import ListConfig
 from torchtyping import TensorType
 
 WHITE = torch.tensor([1.0, 1.0, 1.0])

nerfactory/utils/config.py

@@ -16,7 +16,8 @@
 
 from dataclasses import dataclass
 from typing import Any, Dict, Optional
-from omegaconf import DictConfig, MISSING
+
+from omegaconf import MISSING, DictConfig
 
 
 @dataclass

nerfactory/utils/io.py

@@ -19,7 +19,7 @@ import json
 import logging
 import os
 import pickle
-from typing import Any, Tuple, Optional
+from typing import Any, Optional, Tuple
 
 
 def load_from_json(filename: str):

nerfactory/utils/misc.py

@@ -18,10 +18,10 @@ Miscellaneous helper code.
 
 import hashlib
 import json
+from math import floor, log
 from pydoc import locate
 from typing import Any, Callable, Dict
 
-from math import floor, log
 import torch
 from omegaconf import DictConfig
 

nerfactory/utils/profiler.py

@@ -21,7 +21,11 @@ from typing import Callable
 
 from nerfactory.utils import comms
 from nerfactory.utils.config import LoggingConfig
-from nerfactory.utils.decorators import check_main_thread, check_profiler_enabled, decorate_all
+from nerfactory.utils.decorators import (
+    check_main_thread,
+    check_profiler_enabled,
+    decorate_all,
+)
 
 PROFILER = []
 

nerfactory/utils/writer.py

@@ -27,11 +27,11 @@ from typing import Dict
 import imageio
 import numpy as np
 import torch
+import wandb
 from omegaconf import ListConfig
 from torch.utils.tensorboard import SummaryWriter
 from torchtyping import TensorType
 
-import wandb
 from nerfactory.utils.config import LoggingConfig
 from nerfactory.utils.decorators import check_main_thread, decorate_all
 from nerfactory.utils.misc import human_format