From fba0e93acae9d8b0e44c88486fbe1fc3aa254c0f Mon Sep 17 00:00:00 2001 From: Matt Tancik Date: Wed, 4 May 2022 16:12:49 -0700 Subject: [PATCH] Fix typing --- mattport/nerf/dataset/utils.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/mattport/nerf/dataset/utils.py b/mattport/nerf/dataset/utils.py index 3e5c67455..69031846b 100644 --- a/mattport/nerf/dataset/utils.py +++ b/mattport/nerf/dataset/utils.py @@ -98,12 +98,15 @@ def get_dataset_inputs_dict( """ dataset_inputs_dict = {} + if alpha_color is not None: + alpha_color = get_color(alpha_color) + if dataset_type == "blender": for split in splits: dataset_inputs = load_blender_data( get_absolute_path(data_directory), downscale_factor=downscale_factor, - alpha_color=get_color(alpha_color), + alpha_color=alpha_color, ) dataset_inputs_dict[split] = dataset_inputs else: