Update torchvision import in dataset wrapper

bindsnet/datasets/torchvision_wrapper.py

@@ -1,7 +1,7 @@
 from typing import Dict, Optional
 
 import torch
-import torchvision
+from torchvision import datasets as torchDB
 
 from bindsnet.encoding import Encoder, NullEncoder
 
@@ -13,7 +13,7 @@ def create_torchvision_dataset_wrapper(ds_type):
     ``__getitem__``. This applies to all of the datasets inside of ``torchvision``.
     """
     if type(ds_type) == str:
-        ds_type = getattr(torchvision.datasets, ds_type)
+        ds_type = getattr(torchDB, ds_type)
 
     class TorchvisionDatasetWrapper(ds_type):
         __doc__ = (