From fd6cc84808f38affed4b76978c28602b20178571 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Jon=C3=A1=C5=A1=20Kulh=C3=A1nek?= Date: Wed, 12 Apr 2023 19:17:43 +0200 Subject: [PATCH] Add support for torch v2.0.0 (#1713) Fix missing torch._six namespace --- nerfstudio/data/utils/nerfstudio_collate.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/nerfstudio/data/utils/nerfstudio_collate.py b/nerfstudio/data/utils/nerfstudio_collate.py index ffcd989b4..91774343a 100644 --- a/nerfstudio/data/utils/nerfstudio_collate.py +++ b/nerfstudio/data/utils/nerfstudio_collate.py @@ -23,7 +23,6 @@ from typing import Callable, Dict, Union import torch import torch.utils.data -from torch._six import string_classes from nerfstudio.cameras.cameras import Cameras @@ -119,7 +118,7 @@ def nerfstudio_collate( return torch.tensor(batch, dtype=torch.float64) elif isinstance(elem, int): return torch.tensor(batch) - elif isinstance(elem, string_classes): + elif isinstance(elem, (str, bytes)): return batch elif isinstance(elem, collections.abc.Mapping): try: