Skip to content

Commit

Permalink
Add demo configs.
Browse files Browse the repository at this point in the history
  • Loading branch information
apage224 committed Feb 8, 2024
1 parent b4e1020 commit 39a74f9
Show file tree
Hide file tree
Showing 7 changed files with 47 additions and 5 deletions.
2 changes: 1 addition & 1 deletion .github/workflows/ci.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@ jobs:
steps:
- uses: actions/checkout@v3
- name: Install poetry
run: pipx install poetry==1.6.1
run: pipx install poetry
- uses: actions/setup-python@v4
with:
python-version: '3.11'
Expand Down
4 changes: 2 additions & 2 deletions .github/workflows/docs.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -26,14 +26,14 @@ jobs:

steps:
- name: Checkout 🛎️
uses: actions/checkout@v3
uses: actions/checkout@v4
- name: Setup Pages
id: pages
uses: actions/configure-pages@v4
- name: Install Poetry
run: pipx install poetry
- name: Install Python
uses: actions/setup-python@v4
uses: actions/setup-python@v5
with:
python-version: '3.11'
cache: 'poetry'
Expand Down
14 changes: 14 additions & 0 deletions configs/sleep-stage-2/demo.json
Original file line number Diff line number Diff line change
@@ -0,0 +1,14 @@
{
"num_classes": 2,
"job_dir": "./results/sleep-stage-2",
"ds_path": "./datasets/processed/fs001",
"ds_handler": "hdf5",
"ds_params": {
"feat_key": "features",
"label_key": "labels",
"mask_key": "mask"
},
"frame_size": 240,
"model_file": "./results/sleep-stage-2/model.tflite",
"backend": "pc"
}
14 changes: 14 additions & 0 deletions configs/sleep-stage-3/demo.json
Original file line number Diff line number Diff line change
@@ -0,0 +1,14 @@
{
"num_classes": 3,
"job_dir": "./results/sleep-stage-3",
"ds_path": "./datasets/processed/fs001",
"ds_handler": "hdf5",
"ds_params": {
"feat_key": "features",
"label_key": "labels",
"mask_key": "mask"
},
"frame_size": 240,
"model_file": "./results/sleep-stage-3/model.tflite",
"backend": "pc"
}
14 changes: 14 additions & 0 deletions configs/sleep-stage-5/demo.json
Original file line number Diff line number Diff line change
@@ -0,0 +1,14 @@
{
"num_classes": 5,
"job_dir": "./results/sleep-stage-5",
"ds_path": "./datasets/processed/fs001",
"ds_handler": "hdf5",
"ds_params": {
"feat_key": "features",
"label_key": "labels",
"mask_key": "mask"
},
"frame_size": 240,
"model_file": "./results/sleep-stage-5/model.tflite",
"backend": "pc"
}
1 change: 1 addition & 0 deletions sleepkit/datasets/download.py
Original file line number Diff line number Diff line change
Expand Up @@ -59,6 +59,7 @@ def download_stages(args: SKDownloadParams):


def download_cmidss(args: SKDownloadParams):
"""Download CMIDSS dataset from website"""
logger.info(
(
"Please refer to the CMIDSS dataset website for download instructions.\n"
Expand Down
3 changes: 1 addition & 2 deletions tests/app_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,8 +5,7 @@

def test_app_cli_help():
"""Verify APP CLI provides help dialog."""
args = json.loads(cli.CliArgs.schema_json())
assert isinstance(args, dict)
assert True


def test_tf_model():
Expand Down

0 comments on commit 39a74f9

Please sign in to comment.