Fix tensorboard numpy 2.0 incompatibility breaking binding

allegroai · Jul 4, 2024 · aa227a0 · aa227a0
1 parent 9594e5d
commit aa227a0
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/clearml/binding/frameworks/tensorflow_bind.py b/clearml/binding/frameworks/tensorflow_bind.py
@@ -738,13 +738,14 @@ def get_data(value_dict, metric_search_order):
                     LoggerRoot.get_base_logger(TensorflowBinding).debug(
                         'No tag for \'value\' existing keys %s' % ', '.join(vdict.keys()))
                     continue
+                # noinspection PyBroadException
                 try:
                     from tensorboard.plugins.hparams.metadata import SESSION_START_INFO_TAG
 
                     if tag == SESSION_START_INFO_TAG:
                         self._add_hparams(vdict)
                         continue
-                except ImportError:
+                except Exception:
                     pass
                 metric, values = get_data(vdict, supported_metrics)
                 if metric == 'simpleValue':