Add check_python() (#3088)
* Add check_python() Checks python version against minimum version of 3.7.0. * remove packaging dependency * refactor import
This commit is contained in:
1 parent
91547edec1
commit
57b0d3a605
1 file changed
+11
-1
+11
-1
@@ -16,6 +16,7 @@ from pathlib import Path
|
|||||||
import cv2
|
import cv2
|
||||||
import numpy as np
|
import numpy as np
|
||||||
import pandas as pd
|
import pandas as pd
|
||||||
|
import pkg_resources as pkg
|
||||||
import torch
|
import torch
|
||||||
import torchvision
|
import torchvision
|
||||||
import yaml
|
import yaml
|
||||||
@@ -107,10 +108,19 @@ def check_git_status():
|
|||||||
print(e)
|
print(e)
|
||||||
|
|
||||||
|
|
||||||
|
def check_python(minimum='3.7.0', required=True):
|
||||||
|
# Check current python version vs. required python version
|
||||||
|
current = platform.python_version()
|
||||||
|
result = pkg.parse_version(current) >= pkg.parse_version(minimum)
|
||||||
|
if required:
|
||||||
|
assert result, f'Python {minimum} required by YOLOv5, but Python {current} is currently installed'
|
||||||
|
return result
|
||||||
|
|
||||||
|
|
||||||
def check_requirements(requirements='requirements.txt', exclude=()):
|
def check_requirements(requirements='requirements.txt', exclude=()):
|
||||||
# Check installed dependencies meet requirements (pass *.txt file or list of packages)
|
# Check installed dependencies meet requirements (pass *.txt file or list of packages)
|
||||||
import pkg_resources as pkg
|
|
||||||
prefix = colorstr('red', 'bold', 'requirements:')
|
prefix = colorstr('red', 'bold', 'requirements:')
|
||||||
|
check_python() # check python version
|
||||||
if isinstance(requirements, (str, Path)): # requirements.txt file
|
if isinstance(requirements, (str, Path)): # requirements.txt file
|
||||||
file = Path(requirements)
|
file = Path(requirements)
|
||||||
if not file.exists():
|
if not file.exists():
|
||||||
|
|||||||
Reference in new issue
Block a user