From 9a4c6e2cc41859f44cabaa57e3089f17843c6008 Mon Sep 17 00:00:00 2001 From: jiaruiyu99 Date: Thu, 5 Sep 2024 17:20:53 +0200 Subject: [PATCH] Remove unecessary noqa --- scripts/eval_script.py | 5 ++--- scripts/images_to_nerf_dataset.py | 2 +- scripts/process_thermal/visualise_temperature.py | 14 +------------- scripts/render_video_script.py | 3 +-- scripts/train_eval_script.py | 13 +++++-------- scripts/train_script.py | 2 +- 6 files changed, 11 insertions(+), 28 deletions(-) diff --git a/scripts/eval_script.py b/scripts/eval_script.py index 6c5b84d..82576f6 100644 --- a/scripts/eval_script.py +++ b/scripts/eval_script.py @@ -2,9 +2,8 @@ from pathlib import Path import tyro - -from thermo_nerf.evaluator.evaluator import Evaluator # noqa: E402 -from thermo_nerf.render.renderer import RenderedImageModality, Renderer # noqa: E402 +from thermo_nerf.evaluator.evaluator import Evaluator +from thermo_nerf.render.renderer import RenderedImageModality, Renderer @dataclass diff --git a/scripts/images_to_nerf_dataset.py b/scripts/images_to_nerf_dataset.py index ac6c5ce..9c330e6 100644 --- a/scripts/images_to_nerf_dataset.py +++ b/scripts/images_to_nerf_dataset.py @@ -3,7 +3,7 @@ """ import tyro -from nerfstudio.process_data.images_to_nerfstudio_dataset import ( # noqa: E402 +from nerfstudio.process_data.images_to_nerfstudio_dataset import ( ImagesToNerfstudioDataset, ) diff --git a/scripts/process_thermal/visualise_temperature.py b/scripts/process_thermal/visualise_temperature.py index e68b9ba..af38589 100644 --- a/scripts/process_thermal/visualise_temperature.py +++ b/scripts/process_thermal/visualise_temperature.py @@ -1,22 +1,10 @@ import json -import sys from dataclasses import dataclass import matplotlib.pyplot as plt import tyro -# This is a hack so that the script work with azure sdkv2. -# The root folder is _not_ added to the python path as it was in sdkv1 -# and thus, local imports of rebel_nerf are not found. -# Remove when https://github.com/Azure/azure-sdk-for-python/issues/29724 is fixed - -sys.path.append(".") -sys.path.append("./nerfstudio") - - -from rebel_nerf.flir_themral_images.thermal_visualiser import ( # noqa: E402 - ThermalVisualiser, -) +from rebel_nerf.flir_themral_images.thermal_visualiser import ThermalVisualiser @dataclass diff --git a/scripts/render_video_script.py b/scripts/render_video_script.py index 53dc98b..ff13892 100644 --- a/scripts/render_video_script.py +++ b/scripts/render_video_script.py @@ -7,8 +7,7 @@ from typing import Optional import tyro - -from thermo_nerf.render.renderer import RenderedImageModality, Renderer # noqa: E402 +from thermo_nerf.render.renderer import RenderedImageModality, Renderer @dataclass diff --git a/scripts/train_eval_script.py b/scripts/train_eval_script.py index 99ed884..d53c0c9 100644 --- a/scripts/train_eval_script.py +++ b/scripts/train_eval_script.py @@ -3,14 +3,11 @@ import tyro from nerfstudio.engine.trainer import TrainerConfig -from nerfstudio.scripts.train import main # noqa: E402 - -from thermo_nerf.evaluator.evaluator import Evaluator # noqa: E402 -from thermo_nerf.render.renderer import RenderedImageModality, Renderer # noqa: E402 -from thermo_nerf.rgb_concat.config_concat_nerfacto import ConcatNerfConfig # noqa: E402 -from thermo_nerf.thermal_nerf.config_thermal_nerf import ( # noqa: E402 - ThermalNeRFTrackConfig, -) +from nerfstudio.scripts.train import main +from thermo_nerf.evaluator.evaluator import Evaluator +from thermo_nerf.render.renderer import RenderedImageModality, Renderer +from thermo_nerf.rgb_concat.config_concat_nerfacto import ConcatNerfConfig +from thermo_nerf.thermal_nerf.config_thermal_nerf import ThermalNeRFTrackConfig @dataclass diff --git a/scripts/train_script.py b/scripts/train_script.py index 9387c34..980f96c 100644 --- a/scripts/train_script.py +++ b/scripts/train_script.py @@ -2,7 +2,7 @@ Train a radiance field with nerfstudio. """ -from nerfstudio.scripts.train import entrypoint # noqa: E402 +from nerfstudio.scripts.train import entrypoint if __name__ == "__main__": entrypoint()