Merge pull request #113 from juglab/fix_import

Fix import

Author: Tim-Oliver Buchholz

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

n2v/version.py

@@ -1 +1 @@
-__version__ = '0.3.0'
+__version__ = '0.3.1'