diff --git a/src/domsdatabasen/__init__.py b/src/domsdatabasen/__init__.py index 6a7970e6..8bb4f824 100644 --- a/src/domsdatabasen/__init__.py +++ b/src/domsdatabasen/__init__.py @@ -1 +1,4 @@ """__init__.py file for the domsdatabasen package.""" +from .dataset_builder import DatasetBuilder +from .processor import Processor +from .scraper import Scraper diff --git a/src/scripts/finalize.py b/src/scripts/finalize.py index 7a55397c..62dd6a42 100644 --- a/src/scripts/finalize.py +++ b/src/scripts/finalize.py @@ -11,7 +11,7 @@ import hydra -from domsdatabasen.dataset_builder import DatasetBuilder +from domsdatabasen import DatasetBuilder from omegaconf import DictConfig diff --git a/src/scripts/process.py b/src/scripts/process.py index f9818e68..44726f0b 100644 --- a/src/scripts/process.py +++ b/src/scripts/process.py @@ -17,7 +17,7 @@ import logging import hydra -from domsdatabasen.processor import Processor +from domsdatabasen import Processor from omegaconf import DictConfig logger = logging.getLogger(__name__) diff --git a/src/scripts/scrape.py b/src/scripts/scrape.py index 2134dd3a..8939616c 100644 --- a/src/scripts/scrape.py +++ b/src/scripts/scrape.py @@ -17,7 +17,7 @@ import logging import hydra -from domsdatabasen.scraper import Scraper +from domsdatabasen import Scraper from omegaconf import DictConfig logger = logging.getLogger(__name__)