diff --git a/data_pipeline/online_pipeline/simulate_stream.py b/data_pipeline/online_pipeline/simulate_stream.py index 041b17c..dfde282 100644 --- a/data_pipeline/online_pipeline/simulate_stream.py +++ b/data_pipeline/online_pipeline/simulate_stream.py @@ -1,9 +1,9 @@ import os, time, random from pydub import AudioSegment -BACKGROUND_DIR = "/mnt/persist/train_soundscapes" -CALLS_DIR = "/mnt/persist/train_audio" -OUTPUT_DIR = "/mnt/persist/online_input" +BACKGROUND_DIR = os.getenv("BACKGROUND_DIR", "./tmp/train_soundscapes") +CALLS_DIR = os.getenv("CALLS_DIR", "./tmp/train_audio") +OUTPUT_DIR = os.getenv("OUTPUT_DIR", "./tmp/online_input") INTERVAL = 10 # seconds OVERLAY_PROB = 0.3 # 30% chance to overlay a bird call diff --git a/pyproject.toml b/pyproject.toml new file mode 100644 index 0000000..3d02d85 --- /dev/null +++ b/pyproject.toml @@ -0,0 +1,7 @@ +[tool.pytest.ini_options] +testpaths = ["data_pipeline"] +norecursedirs = [ + "continous_X_pipeline/ansible/k8s/kubespray", + "*/lib/python3.11/site-packages", + "node_modules", +]