From bfe2a89c40bd029cfefe977321e98d6ad13e45b9 Mon Sep 17 00:00:00 2001 From: terrance Date: Thu, 25 May 2023 22:04:34 +0000 Subject: [PATCH] add Tensor to randomcamerasdatamanager --- nerfstudio/data/datamanagers/random_cameras_datamanager.py | 1 + 1 file changed, 1 insertion(+) diff --git a/nerfstudio/data/datamanagers/random_cameras_datamanager.py b/nerfstudio/data/datamanagers/random_cameras_datamanager.py index a5868f54c..391e6a503 100644 --- a/nerfstudio/data/datamanagers/random_cameras_datamanager.py +++ b/nerfstudio/data/datamanagers/random_cameras_datamanager.py @@ -25,6 +25,7 @@ from typing import Dict, List, Tuple, Type, Union import torch from rich.progress import Console from torch.nn import Parameter +from torch import Tensor from typing_extensions import Literal from nerfstudio.cameras.cameras import Cameras