From 72e11a67d7e6e60d21fa2904efd2590106b3ebb6 Mon Sep 17 00:00:00 2001 From: Marty Oehme Date: Wed, 14 Feb 2024 16:02:21 +0100 Subject: [PATCH] chore(code): Format code --- src/calculate_validities.py | 2 ++ src/load_data.py | 13 ++++++++----- 2 files changed, 10 insertions(+), 5 deletions(-) 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)