Skip to content

Commit

Permalink
[pre-commit.ci] auto fixes from pre-commit.com hooks
Browse files Browse the repository at this point in the history
for more information, see https://pre-commit.ci
  • Loading branch information
pre-commit-ci[bot] committed Mar 12, 2024
1 parent 38df06d commit eb6d11e
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions nobrainer/processing/segmentation.py
Original file line number Diff line number Diff line change
Expand Up @@ -39,7 +39,7 @@ def fit(
opt_args=None,
loss=losses.dice,
metrics=metrics.dice,
callbacks=None
callbacks=None,
):
"""Train a segmentation model"""
# TODO: check validity of datasets
Expand Down Expand Up @@ -84,11 +84,11 @@ def _compile():
self.model_.summary()

if callbacks is not None and not isinstance(callbacks, list):
raise AttributeError('Callbacks must be either of type list or None')
raise AttributeError("Callbacks must be either of type list or None")

if callbacks is None:
callbacks = []

if self.checkpoint_tracker:
callbacks.append(self.checkpoint_tracker)
self.model_.fit(
Expand Down

0 comments on commit eb6d11e

Please sign in to comment.