Merge pull request #113 from juglab/fix_import

Fix import
juglab · Sep 6, 2021 · c059e9e · c059e9e
2 parents ed31748 + b0822c4
commit c059e9e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/n2v/models/n2v_standard.py b/n2v/models/n2v_standard.py
@@ -304,7 +304,7 @@ def prepare_for_training(self, optimizer=None, **kwargs):
                     TensorBoard(log_dir=str(self.logdir / 'logs'), write_graph=False, profile_batch=0))
 
         if self.config.train_reduce_lr is not None:
-            from keras.callbacks import ReduceLROnPlateau
+            from tensorflow.keras.callbacks import ReduceLROnPlateau
             rlrop_params = self.config.train_reduce_lr
             if 'verbose' not in rlrop_params:
                 rlrop_params['verbose'] = True

diff --git a/n2v/version.py b/n2v/version.py
@@ -1 +1 @@
-__version__ = '0.3.0'
+__version__ = '0.3.1'