beginning dataset building

landoskape · Apr 6, 2024 · 5951b50 · 5951b50
1 parent b5e0abe
commit 5951b50
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/dominoes/datasets.py b/dominoes/datasets.py
@@ -1,10 +1,17 @@
+from abc import ABC, abstractmethod
 import itertools
+from multiprocessing import Pool
+from functools import partial
+
 import numpy as np
 import scipy as sp
 import torch
+
 from . import utils
-from multiprocessing import Pool
-from functools import partial
+
+class dominoe(ABC):
+    def __init__(self):
+        pass
 
 
 def dominoeUnevenBatch(batchSize, minSeq, maxSeq, listDominoes, dominoeValue, highestDominoe, ignoreIndex=-1, return_full=False):