michael/modrec-workflow
1
0
Fork 0
forked from qoherent/modrec-workflow
Code Pull Requests Workflows Activity

fixed the workflow to detect the helpers folder

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
.riahub/workflows/workflow.yaml
View File

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

6
scripts/produce_dataset.py
View File

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