Skip to content

Commit

Permalink
isort + black
Browse files Browse the repository at this point in the history
  • Loading branch information
c-w-feldmann committed Feb 11, 2025
1 parent a8248b4 commit 84c266c
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions tests/test_pipeline.py
Original file line number Diff line number Diff line change
Expand Up @@ -399,7 +399,9 @@ def test_calibrated_classifier(self) -> None:
),
]
)
calibrated_pipeline = CalibratedClassifierCV(s_pipeline, cv=2, ensemble=True, method="isotonic")
calibrated_pipeline = CalibratedClassifierCV(
s_pipeline, cv=2, ensemble=True, method="isotonic"
)
calibrated_pipeline.fit(TEST_SMILES, CONTAINS_OX)
predicted_value_array = calibrated_pipeline.predict(TEST_SMILES)
predicted_proba_array = calibrated_pipeline.predict_proba(TEST_SMILES)
Expand All @@ -409,6 +411,5 @@ def test_calibrated_classifier(self) -> None:
self.assertEqual(predicted_proba_array.shape, (len(TEST_SMILES),))



if __name__ == "__main__":
unittest.main()

0 comments on commit 84c266c

Please sign in to comment.