Skip to content

Commit 5cd24d7

Browse files
committed
Update tests
1 parent c8ceefb commit 5cd24d7

File tree

3 files changed

+4
-4
lines changed

3 files changed

+4
-4
lines changed

tests/test_MhaMlpComparator.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,10 +8,11 @@
88
import numpy as np
99
from metaperceptron import MhaMlpComparator
1010

11+
1112
# Sample optimizer dictionary for testing
1213
optim_dict = {
13-
'BaseGA': {"epoch": 10, "pop_size": 20},
14-
"OriginalPSO": {"epoch": 10, "pop_size": 20},
14+
'BaseGA': {"epoch": 10, "pop_size": 20},
15+
"OriginalPSO": {"epoch": 10, "pop_size": 20},
1516
}
1617

1718
# Sample data

tests/test_MhaMlpRegressor.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,8 +6,8 @@
66

77
import pytest
88
from sklearn.datasets import make_regression
9-
from sklearn.model_selection import train_test_split
109
from sklearn.metrics import r2_score
10+
from sklearn.model_selection import train_test_split
1111
from metaperceptron import MhaMlpRegressor
1212

1313

tests/test_MlpRegressor.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,6 @@
99
import torch
1010
from metaperceptron import MlpRegressor
1111

12-
1312
# Test data
1413
X = np.random.rand(100, 5) # 100 samples, 5 features each
1514
y = np.random.rand(100, 1) # 100 target values

0 commit comments

Comments
 (0)