From 1dca0b844c71e93ea4ecfdf0f9adc3fbadd86be4 Mon Sep 17 00:00:00 2001 From: Oliver Kinch Date: Thu, 11 Jul 2024 10:46:51 +0200 Subject: [PATCH] Simplify import --- src/domsdatabasen/__init__.py | 3 +++ src/scripts/finalize.py | 2 +- src/scripts/process.py | 2 +- src/scripts/scrape.py | 2 +- 4 files changed, 6 insertions(+), 3 deletions(-) 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__)