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 18, 2024
1 parent ab341ee commit 1171229
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 2 deletions.
3 changes: 2 additions & 1 deletion nobrainer/models/__init__.py
Original file line number Diff line number Diff line change
@@ -1,14 +1,15 @@
from .attention_unet import attention_unet
from .attention_unet_with_inception import attention_unet_with_inception
from .autoencoder import autoencoder
from .bayesian_meshnet import variational_meshnet
from .dcgan import dcgan
from .highresnet import highresnet
from .meshnet import meshnet
from .progressiveae import progressiveae
from .progressivegan import progressivegan
from .unet import unet
from .unetr import unetr
from .bayesian_meshnet import variational_meshnet


def get(name):
"""Return callable that creates a particular `tf.keras.Model`.
Expand Down
2 changes: 1 addition & 1 deletion nobrainer/models/tests/models_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,7 @@
from ..attention_unet import attention_unet
from ..attention_unet_with_inception import attention_unet_with_inception
from ..autoencoder import autoencoder
from ..bayesian_meshnet import variational_meshnet
from ..bayesian_vnet import bayesian_vnet
from ..bayesian_vnet_semi import bayesian_vnet_semi
from ..brainsiam import brainsiam
Expand All @@ -19,7 +20,6 @@
from ..unetr import unetr
from ..vnet import vnet
from ..vox2vox import Vox_ensembler, vox_gan
from ..bayesian_meshnet import variational_meshnet


def model_test(model_cls, n_classes, input_shape, kwds={}):
Expand Down

0 comments on commit 1171229

Please sign in to comment.