FilesExpand file tree

 main

/

python_environment_check.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

54 lines (45 loc) · 1.59 KB

 main

/

python_environment_check.py

Copy path

Top

File metadata and controls

• Code
• Blame

54 lines (45 loc) · 1.59 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

import sys

from distutils.version import LooseVersion

if LooseVersion(sys.version) < LooseVersion('3.8'):

print('[FAIL] We recommend Python 3.8 or newer but'

' found version %s' % (sys.version))

else:

print('[OK] Your Python version is %s' % (sys.version))

def get_packages(pkgs):

versions = []

for p in pkgs:

try:

imported = __import__(p)

try:

versions.append(imported.__version__)

except AttributeError:

try:

versions.append(imported.version)

except AttributeError:

try:

versions.append(imported.version_info)

except AttributeError:

versions.append('0.0')

except ImportError:

print(f'[FAIL]: {p} is not installed and/or cannot be imported.')

versions.append('N/A')

return versions

def check_packages(d):

versions = get_packages(d.keys())

for (pkg_name, suggested_ver), actual_ver in zip(d.items(), versions):

if actual_ver == 'N/A':

continue

actual_ver, suggested_ver = LooseVersion(actual_ver), LooseVersion(suggested_ver)

if actual_ver < suggested_ver:

print(f'[FAIL] {pkg_name} {actual_ver}, please upgrade to >= {suggested_ver}')

else:

print(f'[OK] {pkg_name} {actual_ver}')

if __name__ == '__main__':

d = {

'numpy': '1.21.2',

'scipy': '1.7.0',

'matplotlib': '3.4.3',

'sklearn': '1.0',

'pandas': '1.3.2'

}

check_packages(d)