@@ -564,7 +564,7 @@ def version_leq(lhs: str, rhs: str) -> bool:
564
564
"""
565
565
566
566
lhs , rhs = str (lhs ), str (rhs )
567
- pkging , has_ver = optional_import ("pkg_resources" , name = " packaging" )
567
+ pkging , has_ver = optional_import ("packaging.Version " )
568
568
if has_ver :
569
569
try :
570
570
return cast (bool , pkging .version .Version (lhs ) <= pkging .version .Version (rhs ))
@@ -591,7 +591,8 @@ def version_geq(lhs: str, rhs: str) -> bool:
591
591
592
592
"""
593
593
lhs , rhs = str (lhs ), str (rhs )
594
- pkging , has_ver = optional_import ("pkg_resources" , name = "packaging" )
594
+ pkging , has_ver = optional_import ("packaging.Version" )
595
+
595
596
if has_ver :
596
597
try :
597
598
return cast (bool , pkging .version .Version (lhs ) >= pkging .version .Version (rhs ))
@@ -629,7 +630,7 @@ def pytorch_after(major: int, minor: int, patch: int = 0, current_ver_string: st
629
630
if current_ver_string is None :
630
631
_env_var = os .environ .get ("PYTORCH_VER" , "" )
631
632
current_ver_string = _env_var if _env_var else torch .__version__
632
- ver , has_ver = optional_import ("pkg_resources " , name = "parse_version " )
633
+ ver , has_ver = optional_import ("packaging.version " , name = "parse " )
633
634
if has_ver :
634
635
return ver ("." .join ((f"{ major } " , f"{ minor } " , f"{ patch } " ))) <= ver (f"{ current_ver_string } " ) # type: ignore
635
636
parts = f"{ current_ver_string } " .split ("+" , 1 )[0 ].split ("." , 3 )
0 commit comments