xclele/yolov10
1
0
Fork 0
mirror of https://github.com/THU-MIG/yolov10.git synced 2025-05-29 02:14:22 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix(loggers): DVCLive add package version check (#3112)

Browse Source
This commit is contained in:
Ivan Shcheklein 2023-06-09 02:19:34 -07:00 committed by GitHub
parent 29c954a138
commit 3892e38021
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
ultralytics/yolo/utils/callbacks/dvc.py
View File

@ -1,6 +1,8 @@
# Ultralytics YOLO 🚀, GPL-3.0 license
import os
import pkg_resources as pkg
from ultralytics.yolo.utils import LOGGER, TESTS_RUNNING
from ultralytics.yolo.utils.torch_utils import get_flops, get_num_params
@ -10,8 +12,12 @@ try:
import dvclive
assert not TESTS_RUNNING # do not log pytest
assert version('dvclive')
except (ImportError, AssertionError):
ver = version('dvclive')
if pkg.parse_version(ver) < pkg.parse_version('2.11.0'):
LOGGER.debug(f'DVCLive is detected but version {ver} is incompatible (>=2.11 required).')
dvclive = None # noqa: F811
except (ImportError, AssertionError, TypeError):
dvclive = None
# DVCLive logger instance