fixed the workflow to detect the helpers folder

This commit is contained in:
liyuxiao2 2025-05-20 10:50:55 -04:00
parent e822d68b53
commit 3df27cf012
2 changed files with 7 additions and 6 deletions

View File

@ -39,9 +39,10 @@ jobs:
- name: 1. Build HDF5 Dataset
run: |
mkdir -p data
python scripts/produce_dataset.py
echo "datasets produced successfully"
mkdir -p data
PYTHONPATH=. python scripts/produce_dataset.py
echo "datasets produced successfully"
shell: bash
- name: 2. Train Model
run: |

View File

@ -116,9 +116,9 @@ def generate_datasets(cfg):
dset (h5py.Dataset): The created dataset object
"""
#always create the output directory
os.makedirs(cfg.output_dir, exist_ok=True)
parent = os.path.dirname(cfg.output_dir)
if not parent:
os.makedirs(cfg.output_dir, exist_ok=True)
# we assume the recordings are in .npy format
files = os.listdir(cfg.input_dir)