Fixing doc string

DiogenesAnalytics · Jan 20, 2024 · e8434ff · e8434ff
1 parent 459c16d
commit e8434ff
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/autoencoder/training.py b/src/autoencoder/training.py
@@ -14,7 +14,7 @@ def build_anomaly_loss_function(
     model: Union[keras.Model, BaseAutoencoder],
     axis: Tuple[int, ...] = (1, 2, 3),
 ) -> Callable[[tf.Tensor, tf.Tensor], tf.Tensor]:
-    """Wraps the custom anomaly detection loss function."""
+    """Closure that sets up the custom anomaly detection loss function."""
     # check model type
     if isinstance(model, BaseAutoencoder):
         # get keras.Model object