Skip to content

Commit 08a101c

Browse files
committed
fixed unit tests
1 parent cc840c4 commit 08a101c

1 file changed

Lines changed: 5 additions & 2 deletions

File tree

tests/pipeline/test_bankchurn_pipeline.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,15 @@
11
import numpy as np
22

33
from moddata.pipeline.bankchurn_pipeline import BankchurnPipeline
4+
from moddata.src.config import BankchurnPipelineConfig
45

56

67
def test_bankchurn_pipeline_run():
78
X_train, X_test, y_train, y_test = BankchurnPipeline(
8-
random_state=12345,
9-
train_size=0.8
9+
config=BankchurnPipelineConfig(
10+
random_state=12345,
11+
train_size=0.8
12+
)
1013
).run()
1114

1215
assert X_train.shape == (8_000, 10)

0 commit comments

Comments
 (0)