Skip to content

Speed up tutorial data generation#1064

Merged
bendichter merged 2 commits intomainfrom
faster-tutorial-data
Mar 4, 2026
Merged

Speed up tutorial data generation#1064
bendichter merged 2 commits intomainfrom
faster-tutorial-data

Conversation

@bendichter
Copy link
Collaborator

Summary

  • Remove expensive principal_components (PCA) and spike_amplitudes computations from generate_test_data()
  • Pass compute_pc_features=False and compute_amplitudes=False to export_to_phy instead
  • Reduces tutorial data generation from ~10 minutes to ~3 seconds (206x speedup)

Based on @CodyCBakerPhD's #1063, rebased onto main.

Test plan

  • Verify tutorial data generation completes successfully
  • Verify Phy test data still works in E2E pipeline tests

🤖 Generated with Claude Code

Remove expensive principal_components and spike_amplitudes computations
from generate_test_data(). Pass compute_pc_features=False and
compute_amplitudes=False to export_to_phy instead. This reduces
generation time from ~10min to ~3s.

Based on CodyCBakerPhD's PR #1063.

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
@bendichter bendichter disabled auto-merge March 4, 2026 19:28
@bendichter bendichter requested a review from rly March 4, 2026 19:29
@bendichter bendichter enabled auto-merge March 4, 2026 19:29
@rly rly disabled auto-merge March 4, 2026 19:53
@bendichter bendichter merged commit 19684ae into main Mar 4, 2026
67 of 69 checks passed
@bendichter bendichter deleted the faster-tutorial-data branch March 4, 2026 19:55
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants