CI: Replace depricated distutils lib

openucx · Nov 18, 2024 · 06e3f08 · 06e3f08
1 parent 409480d
commit 06e3f08
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/apps/test_ucx_tls.py b/test/apps/test_ucx_tls.py
@@ -11,8 +11,8 @@
 import re
 import itertools
 import contextlib
-from distutils.version import LooseVersion
 from optparse import OptionParser
+from pkg_resources import parse_version
 
 
 #expected AM transport selections per given number of eps
@@ -276,7 +276,7 @@ def test_tls_allow_list(ucx_info):
         override = 1
     else:
         fw_ver = open(f"/sys/class/infiniband/{dev}/fw_ver").read()
-        if LooseVersion(fw_ver) >= LooseVersion("16.23.0"):
+        if parse_version(fw_ver) >= parse_version("16.23.0"):
             dev_tl_map = am_tls[dev_name+"_roce_dc"]
         else:
             dev_tl_map = am_tls[dev_name+"_roce_no_dc"]