From f4c601f501b5152f8ee55050a704a597edfcfdca Mon Sep 17 00:00:00 2001 From: Shuhei Iitsuka Date: Mon, 14 Aug 2023 19:22:24 +0900 Subject: [PATCH] Make scripts run without install --- scripts/encode_data.py | 7 ++++++- scripts/prepare_knbc.py | 7 ++++++- scripts/tests/test_encode_data.py | 3 +-- 3 files changed, 13 insertions(+), 4 deletions(-) diff --git a/scripts/encode_data.py b/scripts/encode_data.py index 025c84e9..fcb5714c 100644 --- a/scripts/encode_data.py +++ b/scripts/encode_data.py @@ -17,10 +17,15 @@ import functools import itertools import multiprocessing +import os import sys import typing -from budoux import utils +# module hack +LIB_PATH = os.path.join(os.path.dirname(__file__), '..') +sys.path.insert(0, os.path.abspath(LIB_PATH)) + +from budoux import utils # noqa (module hack) ArgList = typing.Optional[typing.List[str]] DEFAULT_OUTPUT_FILENAME = 'encoded_data.txt' diff --git a/scripts/prepare_knbc.py b/scripts/prepare_knbc.py index c6e79162..554d9a58 100644 --- a/scripts/prepare_knbc.py +++ b/scripts/prepare_knbc.py @@ -26,10 +26,15 @@ import argparse import os +import sys import typing from html.parser import HTMLParser -from budoux import utils +# module hack +LIB_PATH = os.path.join(os.path.dirname(__file__), '..') +sys.path.insert(0, os.path.abspath(LIB_PATH)) + +from budoux import utils # noqa (module hack) class KNBCHTMLParser(HTMLParser): diff --git a/scripts/tests/test_encode_data.py b/scripts/tests/test_encode_data.py index 2e713664..2bdc14ed 100644 --- a/scripts/tests/test_encode_data.py +++ b/scripts/tests/test_encode_data.py @@ -18,12 +18,11 @@ import typing import unittest -from budoux import utils - # module hack LIB_PATH = os.path.join(os.path.dirname(__file__), '..', '..') sys.path.insert(0, os.path.abspath(LIB_PATH)) +from budoux import utils # noqa (module hack) from scripts import encode_data # noqa (module hack)