BUG: change config.

Author: msafari18

barcodebert/datasets.py

@@ -8,10 +8,9 @@
 import numpy as np
 import pandas as pd
 import torch
-from torch.utils.data import Dataset
+from torch.utils.data import Dataset, IterableDataset
 from torchtext.vocab import vocab as build_vocab_from_dict
 from transformers import AutoTokenizer
-from torch.utils.data import IterableDataset
 
 
 class KmerTokenizer(object):