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

nerfactory/fields/modules/__init__.py

@@ -13,6 +13,6 @@
 # limitations under the License.
 
 """init field modules"""
+from .base import FieldModule
 from .encoding import Encoding, ScalingAndOffset
 from .mlp import MLP
-from .base import FieldModule

nerfactory/fields/modules/embedding.py

@@ -19,6 +19,7 @@ Code for embeddings.
 
 import torch
 from torchtyping import TensorType
+
 from nerfactory.fields.modules.base import FieldModule
 
 

nerfactory/fields/modules/mlp.py

@@ -15,7 +15,7 @@
 """
 Multi Layer Perceptron
 """
-from typing import Tuple, Optional
+from typing import Optional, Tuple
 
 import torch
 from torch import nn

nerfactory/fields/nerf_field.py

@@ -21,11 +21,16 @@ import torch
 from torch import nn
 from torchtyping import TensorType
 
-from nerfactory.fields.modules.encoding import Encoding, Identity
-from nerfactory.fields.modules.field_heads import DensityFieldHead, FieldHead, FieldHeadNames, RGBFieldHead
-from nerfactory.fields.modules.mlp import MLP
-from nerfactory.fields.base import Field
 from nerfactory.cameras.rays import RaySamples
+from nerfactory.fields.base import Field
+from nerfactory.fields.modules.encoding import Encoding, Identity
+from nerfactory.fields.modules.field_heads import (
+    DensityFieldHead,
+    FieldHead,
+    FieldHeadNames,
+    RGBFieldHead,
+)
+from nerfactory.fields.modules.mlp import MLP
 from nerfactory.fields.modules.spatial_distortions import SpatialDistortion
 
 

nerfactory/fields/nerfw_field.py

@@ -20,6 +20,8 @@ import torch
 from torch import nn
 from torchtyping import TensorType
 
+from nerfactory.cameras.rays import RaySamples
+from nerfactory.fields.base import Field
 from nerfactory.fields.modules.embedding import Embedding
 from nerfactory.fields.modules.encoding import Encoding, Identity
 from nerfactory.fields.modules.field_heads import (
@@ -31,8 +33,6 @@ from nerfactory.fields.modules.field_heads import (
     UncertaintyFieldHead,
 )
 from nerfactory.fields.modules.mlp import MLP
-from nerfactory.fields.base import Field
-from nerfactory.cameras.rays import RaySamples
 
 
 class VanillaNerfWField(Field):