Fix tensorboard numpy 2.0 incompatibility breaking binding

This commit is contained in:
allegroai 2024-07-04 15:27:52 +03:00
parent 9594e5dddd
commit aa227a0cdb

View File

@ -738,13 +738,14 @@ class EventTrainsWriter(object):
LoggerRoot.get_base_logger(TensorflowBinding).debug( LoggerRoot.get_base_logger(TensorflowBinding).debug(
'No tag for \'value\' existing keys %s' % ', '.join(vdict.keys())) 'No tag for \'value\' existing keys %s' % ', '.join(vdict.keys()))
continue continue
# noinspection PyBroadException
try: try:
from tensorboard.plugins.hparams.metadata import SESSION_START_INFO_TAG from tensorboard.plugins.hparams.metadata import SESSION_START_INFO_TAG
if tag == SESSION_START_INFO_TAG: if tag == SESSION_START_INFO_TAG:
self._add_hparams(vdict) self._add_hparams(vdict)
continue continue
except ImportError: except Exception:
pass pass
metric, values = get_data(vdict, supported_metrics) metric, values = get_data(vdict, supported_metrics)
if metric == 'simpleValue': if metric == 'simpleValue':