Skip to content

Commit bcdcfdc

Browse files
committed
Add --short option in check your install
1 parent 1820280 commit bcdcfdc

2 files changed

Lines changed: 12 additions & 5 deletions

File tree

.github/workflows/installation-tips-test.yml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
name: Checks uv instalation from Installation Tips
1+
name: Checks uv installation from Installation Tips
22

33
on:
44
workflow_dispatch:
@@ -30,7 +30,7 @@ jobs:
3030
- name: Install requirements
3131
run: uv pip install -r ./installation_tips/beginner_requirements_stable.txt
3232
- name: Check Installation Tips
33-
run: python ./installation_tips/check_your_install.py --ci # ci flag turns off gui
33+
run: python ./installation_tips/check_your_install.py --ci --short # ci flag turns off gui
3434
- name: Windows cleanup
3535
if: ${{ matrix.os == 'windows-latest' }}
3636
run: python ./installation_tips/cleanup_for_windows.py

installation_tips/check_your_install.py

Lines changed: 10 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -18,11 +18,16 @@ def check_import_si_full():
1818
import spikeinterface.full as si
1919

2020

21-
def _create_recording():
21+
def _create_recording(short=False):
2222
import spikeinterface.full as si
2323

24+
if short:
25+
durations = [30.0]
26+
else:
27+
durations = [200.0]
28+
2429
rec, _ = si.generate_ground_truth_recording(
25-
durations=[200.0], sampling_frequency=30_000.0, num_channels=16, num_units=10, seed=2205
30+
durations=durations, sampling_frequency=30_000.0, num_channels=16, num_units=10, seed=2205
2631
)
2732
rec.save(folder="./toy_example_recording", verbose=False, **job_kwargs)
2833

@@ -85,13 +90,15 @@ def _clean():
8590
# add ci flag so that gui will not be used in ci
8691
# end user can ignore
8792
parser.add_argument("--ci", action="store_false")
93+
parser.add_argument("--short", action="store_false")
94+
8895

8996
if __name__ == "__main__":
9097

9198
args = parser.parse_args()
9299

93100
_clean()
94-
_create_recording()
101+
_create_recording(short=args.short)
95102

96103
steps = [
97104
("Import spikeinterface", check_import_si),

0 commit comments

Comments
 (0)