File size: 434 Bytes
fd01725
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
from abc import abstractmethod
from typing import Optional


class DataBase():
    def __init__(self) -> None:
        raise NotImplementedError

    @abstractmethod
    def prepare_data(self) -> None:
        raise NotImplementedError
    
    @abstractmethod
    def setup(self, stage: Optional[str] = None):
        raise NotImplementedError

    @abstractmethod
    def dataset(self, stage: str):
        raise NotImplementedError