Skip to content

Commit c3abba0

Browse files
committed
remove print during import
1 parent 3b1477f commit c3abba0

28 files changed

+124
-52
lines changed

.gitignore

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,3 +4,4 @@ python_temp.mat
44
python_temp_reformated.mat
55
*.asv
66
.venv/
7+
build/

Dockerfile

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -47,4 +47,7 @@ RUN chown -R octave:octave /home/octave/
4747
USER octave
4848
WORKDIR /home/octave
4949

50-
VOLUME ["/source"]
50+
VOLUME ["/source"]
51+
52+
#https://wiki.ubuntu.com/DashAsBinSh
53+
#RUN rm /bin/sh && ln -s /bin/bash /bin/sh

dist/eegprep-0.0.2-py3-none-any.whl

-31.2 KB
Binary file not shown.

dist/eegprep-0.0.2.tar.gz

-21.7 KB
Binary file not shown.

dist/eegprep-0.0.3-py3-none-any.whl

31.4 KB
Binary file not shown.

notebooks/test.ipynb

Lines changed: 59 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,59 @@
1+
{
2+
"cells": [
3+
{
4+
"cell_type": "code",
5+
"execution_count": 5,
6+
"metadata": {},
7+
"outputs": [],
8+
"source": [
9+
"import sys\n"
10+
]
11+
},
12+
{
13+
"cell_type": "code",
14+
"execution_count": 6,
15+
"metadata": {},
16+
"outputs": [],
17+
"source": [
18+
"sys.path.insert(0, '/Users/dtyoung/Documents/EEGLAB/eegprep/src/')"
19+
]
20+
},
21+
{
22+
"cell_type": "code",
23+
"execution_count": 9,
24+
"metadata": {},
25+
"outputs": [],
26+
"source": [
27+
"import eegprep"
28+
]
29+
},
30+
{
31+
"cell_type": "code",
32+
"execution_count": null,
33+
"metadata": {},
34+
"outputs": [],
35+
"source": []
36+
}
37+
],
38+
"metadata": {
39+
"kernelspec": {
40+
"display_name": ".venv",
41+
"language": "python",
42+
"name": "python3"
43+
},
44+
"language_info": {
45+
"codemirror_mode": {
46+
"name": "ipython",
47+
"version": 3
48+
},
49+
"file_extension": ".py",
50+
"mimetype": "text/x-python",
51+
"name": "python",
52+
"nbconvert_exporter": "python",
53+
"pygments_lexer": "ipython3",
54+
"version": "3.11.9"
55+
}
56+
},
57+
"nbformat": 4,
58+
"nbformat_minor": 2
59+
}

pyproject.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
44

55
[project]
66
name = "eegprep"
7-
version = "0.0.2"
7+
version = "0.0.3"
88
authors = [
99
{ name="Arnaud Delorme", email="[email protected]" },
1010
{ name="Young Truong", email="[email protected]" },

src/eegprep.egg-info/PKG-INFO

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
Metadata-Version: 2.1
22
Name: eegprep
3-
Version: 0.0.2
3+
Version: 0.0.3
44
Summary: EEG preprocessing pipeline on Python
55
Author-email: Arnaud Delorme <[email protected]>, Young Truong <[email protected]>
66
License: GNU General Public License
@@ -40,7 +40,7 @@ Requires-Dist: oct2py>=5.5.0
4040
Requires-Dist: torch>=1.10.0
4141
Requires-Dist: h5py>=3.3.0
4242

43-
## how to duplicate a function
43+
## How to convert a function from MATLAB to Python
4444

4545
1. Get a MATLAB file to load and process an EEG file
4646

@@ -54,3 +54,10 @@ Requires-Dist: h5py>=3.3.0
5454

5555
6. Write the function to compare (see example) and the helper Python function to load the file (note that there could be a general Python helper function)
5656

57+
## Install package
58+
Packaging was done following the tutorial: https://packaging.python.org/en/latest/tutorials/packaging-projects/ with setuptools
59+
60+
To install the package, run:
61+
```
62+
pip install eegprep==0.0.2
63+
```

src/eegprep/ICL_feature_extractor_compare_helper.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
1-
from ICL_feature_extractor import ICL_feature_extractor
2-
from pop_loadset import pop_loadset
1+
from .ICL_feature_extractor import ICL_feature_extractor
2+
from .pop_loadset import pop_loadset
33
import sys
44

55
# check if a parameter is present and if it is assign eeglab_file_path to it

src/eegprep/__init__.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1,3 @@
1-
from .iclabel import iclabel
1+
from .iclabel import iclabel
2+
from .pop_saveset import pop_saveset
3+
from .pop_loadset import pop_loadset

src/eegprep/eeg_autocorr.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -50,12 +50,12 @@ def test_eeg_autocorr():
5050
psdmed = eeg_autocorr(EEG, 100)
5151

5252
# print information about psdmed
53-
print(psdmed.shape)
53+
# print(psdmed.shape)
5454

5555
#print(psdmed)
5656

5757

5858
#assert psdmed.shape == (10, 100)
5959
#assert np.all(np.isfinite(psdmed))
6060

61-
test_eeg_autocorr()
61+
# test_eeg_autocorr()

src/eegprep/eeg_autocorr_compare_helper.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
1-
from eeg_autocorr import eeg_autocorr
2-
from pop_loadset import pop_loadset
1+
from .eeg_autocorr import eeg_autocorr
2+
from .pop_loadset import pop_loadset
33

44
eeglab_file_path = './eeglab_data_with_ica_tmp.set'
55
EEG = pop_loadset(eeglab_file_path)

src/eegprep/eeg_autocorr_fftw_compare_helper.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
1-
from eeg_autocorr_fftw import eeg_autocorr_fftw
2-
from pop_loadset import pop_loadset
3-
from pop_reref import pop_reref
1+
from .eeg_autocorr_fftw import eeg_autocorr_fftw
2+
from .pop_loadset import pop_loadset
3+
from .pop_reref import pop_reref
44
import sys
55

66
# check if a parameter is present and if it is assign eeglab_file_path to it

src/eegprep/eeg_autocorr_welch.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -55,7 +55,7 @@ def test_eeg_autocorr_welch():
5555
psdmed = eeg_autocorr_welch(EEG, 100)
5656

5757
# print information about psdmed
58-
print(psdmed.shape)
59-
print(psdmed)
58+
# print(psdmed.shape)
59+
# print(psdmed)
6060

6161
# test_eeg_autocorr_welch()

src/eegprep/eeg_autocorr_welch_compare_helper.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
1-
from eeg_autocorr_welch import eeg_autocorr_welch
2-
from pop_loadset import pop_loadset
1+
from .eeg_autocorr_welch import eeg_autocorr_welch
2+
from .pop_loadset import pop_loadset
33

44
eeglab_file_path = './eeglab_data_with_ica_tmp.set'
55
EEG = pop_loadset(eeglab_file_path)

src/eegprep/eeg_checkset.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
import numpy as np
22
import os
3-
from pop_loadset import pop_loadset
3+
from .pop_loadset import pop_loadset
44

55
def eeg_checkset(EEG):
66

src/eegprep/eeg_eeglab2mne.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,10 @@
1-
from eeg_autocorr import eeg_autocorr
2-
from pop_loadset import pop_loadset
1+
from .eeg_autocorr import eeg_autocorr
2+
from .pop_loadset import pop_loadset
33
import mne
44
import tempfile
55
import os
66
from mne.export import export_raw
7-
from pop_saveset import pop_saveset # in development
7+
from .pop_saveset import pop_saveset # in development
88

99
# write a funtion that converts a MNE raw object to an EEGLAB set file
1010
def eeg_eeglab2mne(EEG):

src/eegprep/eeg_mne2eeglab.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
1-
from eeg_autocorr import eeg_autocorr
2-
from pop_loadset import pop_loadset
1+
from .eeg_autocorr import eeg_autocorr
2+
from .pop_loadset import pop_loadset
33
import mne
44
import tempfile
55
import os

src/eegprep/eeg_rpsd_compare_helper.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
1-
from eeg_rpsd import eeg_rpsd
2-
from pop_loadset import pop_loadset
1+
from .eeg_rpsd import eeg_rpsd
2+
from .pop_loadset import pop_loadset
33
import sys
44

55
if len(sys.argv) > 1:

src/eegprep/eeglabcompat.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
from oct2py import Oct2Py, get_log
2-
from pop_saveset import pop_saveset
3-
from pop_loadset import pop_loadset
2+
from .pop_saveset import pop_saveset
3+
from .pop_loadset import pop_loadset
44
import logging
55
import os
66
import sys

src/eegprep/iclabel.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -15,9 +15,9 @@ def iclabel(EEG, algorithm='default'):
1515
features[0] = np.single(np.concatenate([features[0],-features[0],features[0][:, ::-1, :, :],-features[0][:, ::-1, :, :]], axis=3))
1616
features[1] = np.single(np.tile(features[1], (1, 1, 1, 4)))
1717
features[2] = np.single(np.tile(features[2], (1, 1, 1, 4)))
18-
print('Feature 0 shape:', features[0].shape)
19-
print('Feature 1 shape:', features[1].shape)
20-
print('Feature 2 shape:', features[2].shape)
18+
# print('Feature 0 shape:', features[0].shape)
19+
# print('Feature 1 shape:', features[1].shape)
20+
# print('Feature 2 shape:', features[2].shape)
2121

2222
# Load the ICLabelNet model
2323
model = ICLabelNet('netICL.mat')

src/eegprep/iclabel_helper.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
from iclabel import iclabel
2-
from pop_loadset import pop_loadset
3-
from pop_saveset import pop_saveset
2+
from .pop_loadset import pop_loadset
3+
from .pop_saveset import pop_saveset
44
import sys
55

66
# check if there are 2 arguments otherwise issue an error
@@ -19,7 +19,7 @@
1919
# EEG['trials'] = 1
2020

2121
# Print the loaded data
22-
print('Data shape:', EEG['data'].shape)
22+
# print('Data shape:', EEG['data'].shape)
2323
EEG = iclabel(EEG, 'default')
2424

2525
# save dataset

src/eegprep/iclabel_net.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -25,10 +25,10 @@ def __init__(self, mat_path):
2525
super().__init__()
2626
iclabel_matlab = scipy.io.loadmat(mat_path)
2727
params = iclabel_matlab['params'][0]
28-
i = 11
29-
print('shape of param', i, torch.tensor(params[i][1]).shape)
28+
# i = 11
29+
# print('shape of param', i, torch.tensor(params[i][1]).shape)
3030
self.discriminator_image_layer1_conv = torch.nn.Conv2d(in_channels=1, out_channels=128, kernel_size=4, stride=2, padding=1, dilation=1)
31-
print(self.discriminator_image_layer1_conv.weight.shape)
31+
# print(self.discriminator_image_layer1_conv.weight.shape)
3232
self.discriminator_image_layer1_conv.weight = torch.nn.Parameter(torch.tensor(params[0][1]).permute(3, 2, 0, 1))
3333
self.discriminator_image_layer1_conv.bias = torch.nn.Parameter(torch.tensor(params[1][1]).squeeze())
3434
self.discriminator_image_layer1_relu = torch.nn.LeakyReLU(0.2)
@@ -83,7 +83,7 @@ def forward(self, image, psdmed, autocorr):
8383
x_image = self.discriminator_image_layer2_relu(x_image)
8484
x_image = self.discriminator_image_layer3_conv(x_image)
8585
x_image = self.discriminator_image_layer3_relu(x_image)
86-
print('x_image', x_image.shape)
86+
# print('x_image', x_image.shape)
8787

8888
x_psdmed = self.discriminator_psdmed_layer1_conv_conv(psdmed)
8989
x_psdmed = self.discriminator_psdmed_layer1_conv_relu(x_psdmed)
@@ -94,7 +94,7 @@ def forward(self, image, psdmed, autocorr):
9494
x_psdmed = self.discriminator_psdmed_reshape(x_psdmed)
9595
x_psdmed = self.discriminator_psdmed_concat1([x_psdmed]*4)
9696
x_psdmed = self.discriminator_psdmed_concat2([x_psdmed]*4)
97-
print('x_psdmed', x_psdmed.shape)
97+
# print('x_psdmed', x_psdmed.shape)
9898

9999
x_autocorr = self.discriminator_autocorr_layer1_conv_conv(autocorr)
100100
x_autocorr = self.discriminator_autocorr_layer1_conv_relu(x_autocorr)
@@ -105,11 +105,11 @@ def forward(self, image, psdmed, autocorr):
105105
x_autocorr = self.discriminator_autocorr_reshape(x_autocorr)
106106
x_autocorr = self.discriminator_autocorr_concat1([x_autocorr]*4)
107107
x_autocorr = self.discriminator_autocorr_concat2([x_autocorr]*4)
108-
print('x_autocorr', x_autocorr.shape)
108+
# print('x_autocorr', x_autocorr.shape)
109109

110110
x = self.discriminator_concat([x_image, x_psdmed, x_autocorr])
111111
x = self.discriminator_conv(x)
112-
print('x', x.shape)
112+
# print('x', x.shape)
113113
# subtract max value to avoid overflow
114114
x = x - torch.max(x, dim=1, keepdim=True).values
115115
x = self.discriminator_softmax(x)

src/eegprep/pop_reref_compare_helper.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
1-
from pop_reref import pop_reref
2-
from pop_loadset import pop_loadset
1+
from .pop_reref import pop_reref
2+
from .pop_loadset import pop_loadset
33
import sys
44
import os
55
import scipy.io

src/eegprep/pop_reref_helper.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
1-
from ICL_feature_extractor import ICL_feature_extractor
2-
from pop_loadset import pop_loadset
3-
from pop_saveset import pop_saveset
4-
from pop_reref import pop_reref
1+
from .ICL_feature_extractor import ICL_feature_extractor
2+
from .pop_loadset import pop_loadset
3+
from .pop_saveset import pop_saveset
4+
from .pop_reref import pop_reref
55
import sys
66

77
# check if a parameter is present and if it is assign eeglab_file_path to it

src/eegprep/pop_resample.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,8 +2,8 @@
22
import os
33
from scipy.io import savemat
44
import tempfile # need to do something with tempfile
5-
from pop_loadset import pop_loadset
6-
from pop_saveset import pop_saveset
5+
from .pop_loadset import pop_loadset
6+
from .pop_saveset import pop_saveset
77
from oct2py import octave as eeglab
88

99
def pop_resample( EEG, freq): # 2 additional parameters in MATLAB (never used)

src/eegprep/pop_saveset.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -183,7 +183,7 @@ def pop_saveset2(EEG, file_name):
183183
# # Step 4: Save the EEGLAB dataset as a .mat file
184184
savemat(file_name, eeglab_dict)
185185

186-
from pop_loadset import pop_loadset
186+
from .pop_loadset import pop_loadset
187187

188188
def test_pop_saveset():
189189
file_path = './eeglab_data_with_ica_tmp.set'

src/eegprep/topoplot_compare_helper.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
1-
from topoplot import topoplot
2-
from pop_loadset import pop_loadset
1+
from .topoplot import topoplot
2+
from .pop_loadset import pop_loadset
33
import sys
44

55
print('Topoplot compare helper')

0 commit comments

Comments
 (0)