diff --git a/dominoes/datasets/base.py b/dominoes/datasets/base.py
index 3e840c3..61a514a 100644
--- a/dominoes/datasets/base.py
+++ b/dominoes/datasets/base.py
@@ -1,13 +1,6 @@
 from abc import ABC, abstractmethod
-from copy import copy
-from itertools import repeat
 import torch
 
-from multiprocessing import Pool, cpu_count
-
-from .support import get_dominoes, get_best_line, pad_best_lines
-from ..utils import named_transpose
-
 
 class Dataset(ABC):
     """
diff --git a/dominoes/datasets/dominoe_dataset.py b/dominoes/datasets/dominoe_dataset.py
index cbce1a9..f6f576d 100644
--- a/dominoes/datasets/dominoe_dataset.py
+++ b/dominoes/datasets/dominoe_dataset.py
@@ -1,4 +1,3 @@
-from abc import ABC, abstractmethod
 from copy import copy
 from itertools import repeat
 import torch