Fix import for depreciated torch._six

See pytorch/pytorch#94709 (comment)

Author: alex391

bindsnet/datasets/collate.py

@@ -8,7 +8,14 @@
 """
 
 import torch
-from torch._six import container_abcs, string_classes, int_classes
+
+# https://github.com/pytorch/pytorch/pull/94709#issuecomment-1461471006
+try:
+    from torch._six import container_abcs, string_classes, int_classes
+except ImportError:
+    int_classes = int
+    string_classes = str
+    import collections.abc as container_abcs
 
 from torch.utils.data._utils import collate as pytorch_collate
 

bindsnet/pipeline/base_pipeline.py

@@ -2,7 +2,13 @@
 from typing import Tuple, Dict, Any
 
 import torch
-from torch._six import container_abcs, string_classes
+
+# https://github.com/pytorch/pytorch/pull/94709#issuecomment-1461471006
+try:
+    from torch._six import container_abcs, string_classes
+except ImportError:
+    string_classes = str
+    import collections.abc as container_abcs
 
 from ..network import Network
 from ..network.monitors import Monitor