Pre-commit failure fixes.

neuronets · Aug 25, 2023 · eb32be3 · eb32be3
1 parent 8e6839d
commit eb32be3
Show file tree

Hide file tree

Showing 3 changed files with 0 additions and 4 deletions.
diff --git a/nobrainer/processing/checkpoint.py b/nobrainer/processing/checkpoint.py
@@ -6,8 +6,6 @@
 
 import tensorflow as tf
 
-from .base import BaseEstimator
-
 
 class CheckpointTracker(tf.keras.callbacks.ModelCheckpoint):
     """Class for saving/loading estimators at/from checkpoints."""

diff --git a/nobrainer/processing/segmentation.py b/nobrainer/processing/segmentation.py
@@ -1,6 +1,5 @@
 import importlib
 import logging
-import os
 
 import tensorflow as tf
 

diff --git a/nobrainer/tests/checkpoint_test.py b/nobrainer/tests/checkpoint_test.py
@@ -4,7 +4,6 @@
 
 import numpy as np
 from numpy.testing import assert_allclose
-import pytest
 import tensorflow as tf
 
 from nobrainer.models import meshnet