Skip to content

Commit

Permalink
Merge branch 'mc/feat/UNet_channel_ind' of https://github.com/CAREami…
Browse files Browse the repository at this point in the history
…cs/careamics into mc/feat/UNet_channel_ind
  • Loading branch information
melisande-c committed May 13, 2024
2 parents 253c31b + c5a3f65 commit 0aa43e6
Showing 1 changed file with 2 additions and 3 deletions.
5 changes: 2 additions & 3 deletions src/careamics/models/ind_channel_model.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,5 @@
"""For creating networks that train independent models for each channel."""

from operator import getitem, setitem
from typing import Any, Dict, Type

Expand Down Expand Up @@ -46,9 +47,7 @@ def __init__(
in_channels = getitem(model_kwargs, in_channels_keyword)
except AttributeError as e:
raise ValueError(
"Input channels keyword, '{}', not found in model_kwargs".format(
in_channels_keyword
)
f"Input channels keyword, '{in_channels_keyword}', not found in model_kwargs"
) from e

# Create a copy of the model config but for a single channel
Expand Down

0 comments on commit 0aa43e6

Please sign in to comment.