From 0bfbbfdc801de3ea339f1a187f42c9625037ee5f Mon Sep 17 00:00:00 2001 From: Robert Sachunsky Date: Fri, 19 Jun 2026 22:01:02 +0200 Subject: [PATCH] training.metrics: allow module init without TFA --- src/eynollah/training/metrics.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/src/eynollah/training/metrics.py b/src/eynollah/training/metrics.py index caa0e65..2b4fc4f 100644 --- a/src/eynollah/training/metrics.py +++ b/src/eynollah/training/metrics.py @@ -5,7 +5,10 @@ import tensorflow as tf from tensorflow.keras import backend as K from tensorflow.keras.metrics import Metric, MeanMetricWrapper, get from tensorflow.keras.initializers import Zeros -from tensorflow_addons.image import connected_components +try: + from tensorflow_addons.image import connected_components +except ModuleNotFoundError: + pass # n/a beyond TF 2.15 (and only needed for training) ... import numpy as np