[LINT] Corrected linting in tf helper (#56)

benchopt · Sep 2, 2022 · 77c6ac6 · 77c6ac6
1 parent a6ed9d5
commit 77c6ac6
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/utils/tf_helper.py b/utils/tf_helper.py
@@ -18,7 +18,8 @@ def on_epoch_end(self, epoch, logs=None):
             self.model.stop_training = not self.cb_(self.model)
 
     class LRWDSchedulerCallback(tf.keras.callbacks.LearningRateScheduler):
-        """Callback that schedules jointly the learning rate and the weight decay
+        """Callback that schedules jointly the learning rate and the weight
+        decay
 
         This is necessary as in TensorFlow, the decoupled weight decay is not
         multiplied by the learning rate.