From d24d64587cbfa2e61ac51bd8c5ad80ee009f03d5 Mon Sep 17 00:00:00 2001
From: Gihan Jayatilaka <gihan@umd.edu>
Date: Thu, 27 Jul 2023 15:23:55 -0400
Subject: [PATCH] Minor bugfix

---
 utils/collate.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/utils/collate.py b/utils/collate.py
index a3f6ad0e..55a9db13 100755
--- a/utils/collate.py
+++ b/utils/collate.py
@@ -5,7 +5,7 @@
 import torch
 import numpy as np
 import collections
-from torch._six import string_classes
+string_classes = str
 
 
 """ Custom collate function """