diff --git a/src/calculate_validities.py b/src/calculate_validities.py index e3955ce..174964e 100644 --- a/src/calculate_validities.py +++ b/src/calculate_validities.py @@ -54,11 +54,13 @@ def calculate_validities( return vd + if __name__ == "__main__": import sys import load_data from pathlib import Path from io import StringIO + if len(sys.argv) == 2: df = load_data.from_yml(Path(sys.argv[1])) else: diff --git a/src/load_data.py b/src/load_data.py index ece9f88..7cb5d8f 100644 --- a/src/load_data.py +++ b/src/load_data.py @@ -1,12 +1,14 @@ import io -from pathlib import Path import sys -try: - import src.yml as yaml # for quarto document scripts -except ModuleNotFoundError: - import yml as yaml # for directly running the package +from pathlib import Path + from pandas import DataFrame, read_csv +try: + import src.yml as yaml # for quarto document scripts +except ModuleNotFoundError: + import yml as yaml # for directly running the package + DEFAULT_YAML_PATH = Path("02-data/processed") @@ -43,6 +45,7 @@ if __name__ == "__main__": res = from_yml() from io import StringIO + output = StringIO() res.to_csv(output) output.seek(0)