From aeafdafa57d9a01423373b4c942a76feb07342cb Mon Sep 17 00:00:00 2001 From: Aaryan Chandna Date: Sun, 15 Jun 2025 19:10:32 -0700 Subject: [PATCH] reformat --- nerfstudio/data/dataparsers/nerfstudio_dataparser.py | 2 +- nerfstudio/data/utils/dataparsers_utils.py | 6 ++++-- 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/nerfstudio/data/dataparsers/nerfstudio_dataparser.py b/nerfstudio/data/dataparsers/nerfstudio_dataparser.py index d92282755..2539a4aa0 100644 --- a/nerfstudio/data/dataparsers/nerfstudio_dataparser.py +++ b/nerfstudio/data/dataparsers/nerfstudio_dataparser.py @@ -31,8 +31,8 @@ from nerfstudio.data.utils.dataparsers_utils import ( get_train_eval_split_all, get_train_eval_split_filename, get_train_eval_split_fraction, + get_train_eval_split_indices, get_train_eval_split_interval, - get_train_eval_split_indices ) from nerfstudio.utils.io import load_from_json from nerfstudio.utils.rich_utils import CONSOLE diff --git a/nerfstudio/data/utils/dataparsers_utils.py b/nerfstudio/data/utils/dataparsers_utils.py index 12ac73ee0..7e8e7bfff 100644 --- a/nerfstudio/data/utils/dataparsers_utils.py +++ b/nerfstudio/data/utils/dataparsers_utils.py @@ -100,7 +100,10 @@ def get_train_eval_split_all(image_filenames: List) -> Tuple[np.ndarray, np.ndar i_eval = i_all return i_train, i_eval -def get_train_eval_split_indices(image_filenames: List, eval_image_indices: Tuple[int, ...]) -> Tuple[np.ndarray, np.ndarray]: + +def get_train_eval_split_indices( + image_filenames: List, eval_image_indices: Tuple[int, ...] +) -> Tuple[np.ndarray, np.ndarray]: """ Get the train/eval split based on specified indices in the config. @@ -116,4 +119,3 @@ def get_train_eval_split_indices(image_filenames: List, eval_image_indices: Tupl train_indices = all_indices - eval_indices return np.array(sorted(train_indices)), np.array(sorted(eval_indices)) -