Skip to content

Conversation

@Nikhita-14
Copy link

@Nikhita-14 Nikhita-14 commented Jan 7, 2026

Add root-level requirements.txt for CEBRA pipeline

Adds a root-level requirements.txt for the CEBRA pipeline.

Reason:
The README instructs pip install -r requirements.txt, but the only existing requirements file is inside PreprocessingPipeline/.
Having a root-level file allows users to install dependencies directly from the repo root, as intended.

Affected files (previously missing imports without root-level requirements):

  • prepare_cebra_input.py
  • preprocessing_functions.py
  • train_cebra.py
  • train_cebra_cut_supervised.py
  • train_cebra_cut_unsupervised.py
  • cebra-running-notebook-2.ipynb
  • eeg_preprocessing.ipynb

Included core dependencies:

  • Numerical & EEG processing: numpy, scipy, mne
  • Machine learning / deep learning: torch>=2.0, cebra==0.6.0a2, scikit-learn
  • Visualization & utilities: matplotlib, seaborn, tqdm

Feedback and suggestions are welcome!

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.

1 participant