Compatibility for numpy 2.0

nerfstudio/cameras/cameras.py

@@ -15,6 +15,7 @@
 """
 Camera Models
 """
+
 import base64
 import math
 from dataclasses import dataclass

nerfstudio/cameras/lie_groups.py

@@ -15,6 +15,7 @@
 """
 Helper for Lie group operations. Currently only used for pose optimization.
 """
+
 import torch
 from jaxtyping import Float
 from torch import Tensor

nerfstudio/cameras/rays.py

@@ -15,6 +15,7 @@
 """
 Some ray datastructures.
 """
+
 import random
 from dataclasses import dataclass, field
 from typing import Callable, Dict, Literal, Optional, Tuple, Union, overload
@@ -153,15 +154,13 @@ def get_weights(self, densities: Float[Tensor, "*batch num_samples 1"]) -> Float
     @staticmethod
     def get_weights_and_transmittance_from_alphas(
         alphas: Float[Tensor, "*batch num_samples 1"], weights_only: Literal[True]
-    ) -> Float[Tensor, "*batch num_samples 1"]:
-        ...
+    ) -> Float[Tensor, "*batch num_samples 1"]: ...
 
     @overload
     @staticmethod
     def get_weights_and_transmittance_from_alphas(
         alphas: Float[Tensor, "*batch num_samples 1"], weights_only: Literal[False] = False
-    ) -> Tuple[Float[Tensor, "*batch num_samples 1"], Float[Tensor, "*batch num_samples 1"]]:
-        ...
+    ) -> Tuple[Float[Tensor, "*batch num_samples 1"], Float[Tensor, "*batch num_samples 1"]]: ...
 
     @staticmethod
     def get_weights_and_transmittance_from_alphas(

nerfstudio/configs/base_config.py

@@ -14,7 +14,6 @@
 
 """Base Configs"""
 
-
 from __future__ import annotations
 
 from dataclasses import dataclass, field

nerfstudio/data/datamanagers/parallel_datamanager.py

@@ -15,6 +15,7 @@
 """
 Parallel data manager that generates training data in multiple python processes.
 """
+
 from __future__ import annotations
 
 import concurrent.futures

nerfstudio/data/dataparsers/arkitscenes_dataparser.py

@@ -13,6 +13,7 @@
 # limitations under the License.
 
 """Data parser for ARKitScenes dataset"""
+
 import math
 from dataclasses import dataclass, field
 from pathlib import Path

nerfstudio/data/dataparsers/blender_dataparser.py

@@ -13,6 +13,7 @@
 # limitations under the License.
 
 """Data parser for blender dataset"""
+
 from __future__ import annotations
 
 from dataclasses import dataclass, field

nerfstudio/data/dataparsers/colmap_dataparser.py

@@ -11,7 +11,7 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-""" Data parser for nerfstudio datasets. """
+"""Data parser for nerfstudio datasets."""
 
 from __future__ import annotations
 

nerfstudio/data/dataparsers/dnerf_dataparser.py

@@ -13,6 +13,7 @@
 # limitations under the License.
 
 """Data parser for blender dataset"""
+
 from __future__ import annotations
 
 from dataclasses import dataclass, field

nerfstudio/data/dataparsers/dycheck_dataparser.py

@@ -13,6 +13,7 @@
 # limitations under the License.
 
 """Data parser for DyCheck (https://arxiv.org/abs/2210.13445) dataset of `iphone` subset"""
+
 from __future__ import annotations
 
 import math

nerfstudio/data/dataparsers/nerfosr_dataparser.py

@@ -11,9 +11,9 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-""" Data parser for NeRF-OSR datasets
+"""Data parser for NeRF-OSR datasets
 
-    Presented in the paper: https://4dqv.mpi-inf.mpg.de/NeRF-OSR/
+Presented in the paper: https://4dqv.mpi-inf.mpg.de/NeRF-OSR/
 
 """
 

nerfstudio/data/dataparsers/nerfstudio_dataparser.py

@@ -11,7 +11,7 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-""" Data parser for nerfstudio datasets. """
+"""Data parser for nerfstudio datasets."""
 
 from __future__ import annotations
 

nerfstudio/data/dataparsers/nuscenes_dataparser.py

@@ -13,6 +13,7 @@
 # limitations under the License.
 
 """Data parser for NuScenes dataset"""
+
 import math
 import os
 from dataclasses import dataclass, field

nerfstudio/data/dataparsers/phototourism_dataparser.py

@@ -13,6 +13,7 @@
 # limitations under the License.
 
 """Phototourism dataset parser. Datasets and documentation here: http://phototour.cs.washington.edu/datasets/"""
+
 from __future__ import annotations
 
 import math

nerfstudio/data/dataparsers/scannet_dataparser.py

@@ -13,6 +13,7 @@
 # limitations under the License.
 
 """Data parser for ScanNet dataset"""
+
 import math
 from dataclasses import dataclass, field
 from pathlib import Path

nerfstudio/data/dataparsers/scannetpp_dataparser.py

@@ -11,7 +11,7 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-""" Data parser for ScanNet++ datasets. """
+"""Data parser for ScanNet++ datasets."""
 
 from __future__ import annotations
 

nerfstudio/data/datasets/base_dataset.py

@@ -15,6 +15,7 @@
 """
 Dataset.
 """
+
 from __future__ import annotations
 
 from copy import deepcopy

nerfstudio/data/utils/data_utils.py

@@ -13,6 +13,7 @@
 # limitations under the License.
 
 """Utility functions to allow easy re-use of common operations across dataloaders"""
+
 from pathlib import Path
 from typing import List, Tuple, Union
 

nerfstudio/data/utils/dataparsers_utils.py

@@ -11,7 +11,7 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-""" Data parser utils for nerfstudio datasets. """
+"""Data parser utils for nerfstudio datasets."""
 
 import math
 import os

nerfstudio/engine/callbacks.py

@@ -15,6 +15,7 @@
 """
 Callback code used for training iterations
 """
+
 from __future__ import annotations
 
 from dataclasses import dataclass

nerfstudio/engine/optimizers.py

@@ -15,6 +15,7 @@
 """
 Optimizers class.
 """
+
 from __future__ import annotations
 
 from dataclasses import dataclass

nerfstudio/engine/trainer.py

@@ -15,6 +15,7 @@
 """
 Code to train model.
 """
+
 from __future__ import annotations
 
 import dataclasses

nerfstudio/exporter/exporter_utils.py

@@ -16,7 +16,6 @@
 Export utils such as structs, point cloud generation, and rendering code.
 """
 
-
 from __future__ import annotations
 
 import sys

nerfstudio/exporter/texture_utils.py

@@ -16,7 +16,6 @@
 Texture utils.
 """
 
-
 from __future__ import annotations
 
 import math

nerfstudio/exporter/tsdf_utils.py

@@ -16,7 +16,6 @@
 TSDF utils.
 """
 
-
 from __future__ import annotations
 
 from dataclasses import dataclass, field

nerfstudio/field_components/__init__.py

@@ -13,6 +13,7 @@
 # limitations under the License.
 
 """init field modules"""
+
 from .base_field_component import FieldComponent as FieldComponent
 from .encodings import Encoding as Encoding, ScalingAndOffset as ScalingAndOffset
 from .mlp import MLP as MLP

nerfstudio/field_components/base_field_component.py

@@ -15,6 +15,7 @@
 """
 The field module baseclass.
 """
+
 from abc import abstractmethod
 from typing import Optional
 

nerfstudio/field_components/embedding.py

@@ -16,7 +16,6 @@
 Code for embeddings.
 """
 
-
 import torch
 from jaxtyping import Shaped
 from torch import Tensor

nerfstudio/field_components/field_heads.py

@@ -15,6 +15,7 @@
 """
 Collection of render heads
 """
+
 from enum import Enum
 from typing import Callable, Optional, Union
 

nerfstudio/field_components/mlp.py

@@ -15,6 +15,7 @@
 """
 Multi Layer Perceptron
 """
+
 from typing import Literal, Optional, Set, Tuple, Union
 
 import numpy as np

nerfstudio/fields/density_fields.py

@@ -16,7 +16,6 @@
 Proposal network field.
 """
 
-
 from typing import Literal, Optional, Tuple
 
 import torch

nerfstudio/fields/generfacto_field.py

@@ -16,7 +16,6 @@
 Field for Generfacto model
 """
 
-
 from typing import Dict, Literal, Optional, Tuple
 
 import numpy as np

nerfstudio/fields/nerfacto_field.py

@@ -16,7 +16,6 @@
 Field for compound nerf model, adds scene contraction and image embeddings to instant ngp
 """
 
-
 from typing import Dict, Literal, Optional, Tuple
 
 import torch

nerfstudio/fields/semantic_nerf_field.py

@@ -15,6 +15,7 @@
 """
 Semantic NeRF field implementation.
 """
+
 from typing import Dict, Optional, Tuple
 
 import torch

nerfstudio/fields/tensorf_field.py

@@ -14,7 +14,6 @@
 
 """TensoRF Field"""
 
-
 from typing import Dict, Optional
 
 import torch

nerfstudio/fields/vanilla_nerf_field.py

@@ -14,7 +14,6 @@
 
 """Classic NeRF field"""
 
-
 from typing import Dict, Optional, Tuple, Type
 
 import torch

nerfstudio/model_components/losses.py

@@ -15,6 +15,7 @@
 """
 Collection of Losses.
 """
+
 from enum import Enum
 from typing import Dict, Literal, Optional, Tuple, cast
 

nerfstudio/model_components/ray_generators.py

@@ -15,6 +15,7 @@
 """
 Ray generator.
 """
+
 from jaxtyping import Int
 from torch import Tensor, nn
 

nerfstudio/model_components/ray_samplers.py

@@ -379,8 +379,7 @@ class DensityFn(Protocol):
 
     def __call__(
         self, positions: Float[Tensor, "*batch 3"], times: Optional[Float[Tensor, "*batch 1"]] = None
-    ) -> Float[Tensor, "*batch 1"]:
-        ...
+    ) -> Float[Tensor, "*batch 1"]: ...
 
 
 class VolumetricSampler(Sampler):

nerfstudio/model_components/renderers.py

@@ -26,6 +26,7 @@
     rgb = rgb_renderer(rgb=field_outputs[FieldHeadNames.RGB], weights=weights)
 
 """
+
 import contextlib
 import math
 from typing import Generator, Literal, Optional, Tuple, Union

nerfstudio/model_components/shaders.py

@@ -13,6 +13,7 @@
 # limitations under the License.
 
 """Shaders for rendering."""
+
 from typing import Optional
 
 from jaxtyping import Float

nerfstudio/models/mipnerf.py

@@ -15,6 +15,7 @@
 """
 Implementation of mip-NeRF.
 """
+
 from __future__ import annotations
 
 from typing import Dict, List, Tuple

nerfstudio/pipelines/base_pipeline.py

@@ -15,6 +15,7 @@
 """
 Abstracts for the Pipeline class.
 """
+
 from __future__ import annotations
 
 import typing

nerfstudio/plugins/types.py

@@ -15,6 +15,7 @@
 """
 This package contains specifications used to register plugins.
 """
+
 from dataclasses import dataclass
 
 from nerfstudio.engine.trainer import TrainerConfig

nerfstudio/scripts/docs/build_docs.py

@@ -14,6 +14,7 @@
 
 #!/usr/bin/env python
 """Simple yaml debugger"""
+
 import subprocess
 import sys
 

nerfstudio/scripts/downloads/download_data.py

@@ -13,6 +13,7 @@
 # limitations under the License.
 
 """Download datasets and specific captures from the datasets."""
+
 from __future__ import annotations
 
 import json