From 87eeda765bc4f4bed6b0b2b4fc98ee63fd142232 Mon Sep 17 00:00:00 2001
From: Wei Lin Liu <108709566+willy-liu@users.noreply.github.com>
Date: Thu, 22 May 2025 11:17:59 +0800
Subject: [PATCH] Update nerfstudio_collate.py

Fix removed class usage "torch._six" which is dropped in later Pytorch 2.x releases.
---
 nerfstudio/data/utils/nerfstudio_collate.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/nerfstudio/data/utils/nerfstudio_collate.py b/nerfstudio/data/utils/nerfstudio_collate.py
index 65917d85..b10f2985 100644
--- a/nerfstudio/data/utils/nerfstudio_collate.py
+++ b/nerfstudio/data/utils/nerfstudio_collate.py
@@ -23,7 +23,12 @@
 
 import torch
 import torch.utils.data
-from torch._six import string_classes
+try:
+    # old PyTorch compatibility
+    from torch._six import string_classes
+except ImportError:
+    # fallback for torch versions that dropped _six
+    string_classes = (str,)
 
 from nerfstudio.cameras.cameras import Cameras
 from nerfstudio.utils.images import BasicImages