Skip to content

Commit

Permalink
typo and docstring
Browse files Browse the repository at this point in the history
  • Loading branch information
c-w-feldmann committed Feb 11, 2025
1 parent 272f91a commit 7499cf0
Showing 1 changed file with 2 additions and 7 deletions.
9 changes: 2 additions & 7 deletions tests/test_pipeline.py
Original file line number Diff line number Diff line change
Expand Up @@ -380,12 +380,7 @@ class PipleCompatibillityTest(unittest.TestCase):
"""Test if the pipeline is compatible with other sklearn functionalities."""

def test_calibrated_classifier(self) -> None:
"""Test if the pipeline can be used in a sklearn pipeline.
Returns
-------
None
"""
"""Test if the pipeline can be used with a CalibratedClassifierCV."""
smi2mol = SmilesToMol()
mol2morgan = MolToMorganFP(radius=FP_RADIUS, n_bits=FP_SIZE)
d_tree = DecisionTreeClassifier()
Expand All @@ -396,7 +391,7 @@ def test_calibrated_classifier(self) -> None:
("decision_tree", d_tree),
]
)
calibrated_pipeline = CalibtatedClassifierCV(s_pipeline)
calibrated_pipeline = CalibratedClassifierCV(s_pipeline)
calibrated_pipeline.fit(TEST_SMILES, CONTAINS_OX)
predicted_value_array = s_pipeline.predict(TEST_SMILES)
for pred_val, true_val in zip(predicted_value_array, CONTAINS_OX):
Expand Down

0 comments on commit 7499cf0

Please sign in to comment.