ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-06-26 18:16:07 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Fix attrs.__version__ deprecation warning (#1145)

Browse Source
This commit is contained in:
allegroai 2023-11-07 16:35:04 +02:00
parent ecdeb2d0a2
commit 68aa2a9e9e
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
clearml/utilities/attrs.py
View File

@ -2,12 +2,19 @@ import attr
from .version import Version from .version import Version
try:
# noinspection PyUnresolvedReferences
import importlib.metadata
attr_version = importlib.metadata.version("attrs")
except ImportError:
attr_version = attr.__version__
class attrs(object): class attrs(object):
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
if any(x in kwargs for x in ("eq", "order")): if any(x in kwargs for x in ("eq", "order")):
raise RuntimeError("Only `cmp` is supported for attr.attrs, not `eq` or `order`") raise RuntimeError("Only `cmp` is supported for attr.attrs, not `eq` or `order`")
if Version(attr.__version__) >= Version("19.2"): if Version(attr_version) >= Version("19.2"):
cmp = kwargs.pop("cmp", None) cmp = kwargs.pop("cmp", None)
if cmp is not None: if cmp is not None:
kwargs["eq"] = kwargs["order"] = cmp kwargs["eq"] = kwargs["order"] = cmp