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

nerfactory/graphs/base.py

@@ -31,7 +31,11 @@ from nerfactory.data.structs import DatasetInputs, SceneBounds
 from nerfactory.graphs.modules.ray_generator import RayGenerator
 from nerfactory.utils import profiler
 from nerfactory.utils.config import GraphConfig
-from nerfactory.utils.misc import get_masked_dict, instantiate_from_dict_config, is_not_none
+from nerfactory.utils.misc import (
+    get_masked_dict,
+    instantiate_from_dict_config,
+    is_not_none,
+)
 
 
 @profiler.time_function

nerfactory/graphs/mipnerf.py

@@ -22,19 +22,22 @@ from typing import Dict, List
 import torch
 from torch.nn import Parameter
 from torchmetrics import PeakSignalNoiseRatio
-from torchmetrics.image.lpip import LearnedPerceptualImagePatchSimilarity
 from torchmetrics.functional import structural_similarity_index_measure
+from torchmetrics.image.lpip import LearnedPerceptualImagePatchSimilarity
 
+from nerfactory.cameras.rays import RayBundle
 from nerfactory.fields.modules.encoding import NeRFEncoding
 from nerfactory.fields.modules.field_heads import FieldHeadNames
 from nerfactory.fields.nerf_field import NeRFField
 from nerfactory.graphs.base import Graph
-from nerfactory.optimizers.loss import MSELoss
 from nerfactory.graphs.modules.ray_sampler import PDFSampler, UniformSampler
-from nerfactory.renderers.renderers import AccumulationRenderer, DepthRenderer, RGBRenderer
-from nerfactory.utils import colors
-from nerfactory.cameras.rays import RayBundle
-from nerfactory.utils import visualization, writer
+from nerfactory.optimizers.loss import MSELoss
+from nerfactory.renderers.renderers import (
+    AccumulationRenderer,
+    DepthRenderer,
+    RGBRenderer,
+)
+from nerfactory.utils import colors, visualization, writer
 
 
 class MipNerfGraph(Graph):

nerfactory/graphs/mipnerf_360.py

@@ -22,21 +22,24 @@ from typing import Dict, List
 import torch
 from torch.nn import Parameter
 from torchmetrics import PeakSignalNoiseRatio
-from torchmetrics.image.lpip import LearnedPerceptualImagePatchSimilarity
 from torchmetrics.functional import structural_similarity_index_measure
+from torchmetrics.image.lpip import LearnedPerceptualImagePatchSimilarity
 
+from nerfactory.cameras.rays import RayBundle
 from nerfactory.fields.modules.encoding import NeRFEncoding
 from nerfactory.fields.modules.field_heads import FieldHeadNames
 from nerfactory.fields.modules.spatial_distortions import SceneContraction
 from nerfactory.fields.nerf_field import NeRFField
 from nerfactory.graphs.base import Graph
 from nerfactory.graphs.modules.ray_losses import distortion_loss
-from nerfactory.optimizers.loss import MSELoss
 from nerfactory.graphs.modules.ray_sampler import PDFSampler, UniformSampler
-from nerfactory.renderers.renderers import AccumulationRenderer, DepthRenderer, RGBRenderer
-from nerfactory.utils import colors
-from nerfactory.cameras.rays import RayBundle
-from nerfactory.utils import visualization, writer
+from nerfactory.optimizers.loss import MSELoss
+from nerfactory.renderers.renderers import (
+    AccumulationRenderer,
+    DepthRenderer,
+    RGBRenderer,
+)
+from nerfactory.utils import colors, visualization, writer
 
 
 class MipNerf360Graph(Graph):

nerfactory/graphs/modules/ray_generator.py

@@ -17,8 +17,9 @@ Ray generator.
 """
 from torch import nn
 from torchtyping import TensorType
-from nerfactory.cameras.rays import RayBundle
+
 from nerfactory.cameras.cameras import get_camera_model
+from nerfactory.cameras.rays import RayBundle
 
 
 class RayGenerator(nn.Module):

nerfactory/graphs/modules/scene_colliders.py

@@ -18,8 +18,9 @@ Ray generator.
 import torch
 from torch import nn
 from torchtyping import TensorType
-from nerfactory.data.structs import SceneBounds
+
 from nerfactory.cameras.rays import RayBundle
+from nerfactory.data.structs import SceneBounds
 
 
 class SceneBoundsCollider(nn.Module):

nerfactory/graphs/nerfw.py

@@ -18,20 +18,24 @@ NeRF-W (NeRF in the wild) implementation.
 
 import torch
 from torchmetrics import PeakSignalNoiseRatio
-from torchmetrics.image.lpip import LearnedPerceptualImagePatchSimilarity
 from torchmetrics.functional import structural_similarity_index_measure
+from torchmetrics.image.lpip import LearnedPerceptualImagePatchSimilarity
 
+from nerfactory.cameras.rays import RayBundle
 from nerfactory.fields.modules.encoding import NeRFEncoding
 from nerfactory.fields.modules.field_heads import FieldHeadNames
 from nerfactory.fields.nerf_field import NeRFField
 from nerfactory.fields.nerfw_field import VanillaNerfWField
 from nerfactory.graphs.base import Graph
-from nerfactory.optimizers.loss import MSELoss
 from nerfactory.graphs.modules.ray_sampler import PDFSampler, UniformSampler
-from nerfactory.renderers.renderers import AccumulationRenderer, DepthRenderer, RGBRenderer, UncertaintyRenderer
-from nerfactory.utils import colors
-from nerfactory.cameras.rays import RayBundle
-from nerfactory.utils import visualization, writer
+from nerfactory.optimizers.loss import MSELoss
+from nerfactory.renderers.renderers import (
+    AccumulationRenderer,
+    DepthRenderer,
+    RGBRenderer,
+    UncertaintyRenderer,
+)
+from nerfactory.utils import colors, visualization, writer
 
 
 class NerfWGraph(Graph):

nerfactory/graphs/semantic_nerf.py

@@ -21,16 +21,21 @@ from typing import Dict, Optional, Tuple
 import torch
 from torch import nn
 from torchtyping import TensorType
-from nerfactory.data.structs import Semantics
 
-from nerfactory.fields.modules.encoding import Encoding, Identity, NeRFEncoding
-from nerfactory.fields.modules.field_heads import DensityFieldHead, FieldHeadNames, RGBFieldHead, SemanticFieldHead
-from nerfactory.fields.modules.mlp import MLP
+from nerfactory.cameras.rays import RayBundle, RaySamples
+from nerfactory.data.structs import Semantics
 from nerfactory.fields.base import Field
+from nerfactory.fields.modules.encoding import Encoding, Identity, NeRFEncoding
+from nerfactory.fields.modules.field_heads import (
+    DensityFieldHead,
+    FieldHeadNames,
+    RGBFieldHead,
+    SemanticFieldHead,
+)
+from nerfactory.fields.modules.mlp import MLP
 from nerfactory.fields.nerf_field import NeRFField
 from nerfactory.graphs.vanilla_nerf import NeRFGraph
 from nerfactory.renderers.renderers import SemanticRenderer
-from nerfactory.cameras.rays import RaySamples, RayBundle
 from nerfactory.utils import writer
 
 

nerfactory/graphs/vanilla_nerf.py

@@ -22,19 +22,22 @@ from typing import Dict, List
 import torch
 from torch.nn import Parameter
 from torchmetrics import PeakSignalNoiseRatio
-from torchmetrics.image.lpip import LearnedPerceptualImagePatchSimilarity
 from torchmetrics.functional import structural_similarity_index_measure
+from torchmetrics.image.lpip import LearnedPerceptualImagePatchSimilarity
 
+from nerfactory.cameras.rays import RayBundle
 from nerfactory.fields.modules.encoding import NeRFEncoding
 from nerfactory.fields.modules.field_heads import FieldHeadNames
 from nerfactory.fields.nerf_field import NeRFField
 from nerfactory.graphs.base import Graph
-from nerfactory.optimizers.loss import MSELoss
 from nerfactory.graphs.modules.ray_sampler import PDFSampler, UniformSampler
-from nerfactory.renderers.renderers import AccumulationRenderer, DepthRenderer, RGBRenderer
-from nerfactory.utils import colors
-from nerfactory.cameras.rays import RayBundle
-from nerfactory.utils import visualization, writer
+from nerfactory.optimizers.loss import MSELoss
+from nerfactory.renderers.renderers import (
+    AccumulationRenderer,
+    DepthRenderer,
+    RGBRenderer,
+)
+from nerfactory.utils import colors, visualization, writer
 
 
 class NeRFGraph(Graph):