Merge branch 'v2' into data-loader-upgrade

rvankoert · Mar 15, 2024 · aef7be9 · aef7be9
2 parents e2baf48 + 2fcce72
commit aef7be9
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/data/augment_layers.py b/src/data/augment_layers.py
@@ -291,7 +291,8 @@ def estimate_background_color(self, image):
         Parameters
         ----------
         image : tf.Tensor
-            Input image tensor in the format [batch_size, height, width, channels].
+            Input image tensor in the format [batch_size, height, width,
+            channels].
 
         Returns
         -------