Skip to content

Commit e110ec4

Browse files
author
Han Wang
committed
correct way of getting site package path
1 parent 5061492 commit e110ec4

File tree

1 file changed

+5
-6
lines changed

1 file changed

+5
-6
lines changed

setup.py

Lines changed: 5 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -4,17 +4,16 @@
44
from setuptools_scm import get_version
55
from packaging.version import LegacyVersion
66
from os import path, makedirs
7-
import imp, sys, platform
7+
import os, imp, sys, platform, sysconfig
88

99
def get_dp_install_path() :
10-
site_packages_path = path.join(path.dirname(path.__file__), 'site-packages')
11-
dp_scm_version = get_version(root="./", relative_to=__file__)
10+
site_packages_path = sysconfig.get_paths()['purelib']
11+
dp_scm_version = get_version(root=".", relative_to=__file__)
1212
python_version = 'py' + str(sys.version_info.major + sys.version_info.minor * 0.1)
1313
os_info = sys.platform
1414
machine_info = platform.machine()
15-
dp_pip_install_path = site_packages_path + '/deepmd'
16-
dp_setup_install_path = site_packages_path + '/deepmd_kit-' + dp_scm_version + '-' + python_version + '-' + os_info + '-' + machine_info + '.egg/deepmd'
17-
15+
dp_pip_install_path = os.path.join(site_packages_path, 'deepmd')
16+
dp_setup_install_path = os.path.join(site_packages_path, 'deepmd_kit-' + dp_scm_version + '-' + python_version + '-' + os_info + '-' + machine_info + '.egg', 'deepmd')
1817
return dp_pip_install_path, dp_setup_install_path
1918

2019
readme_file = path.join(path.dirname(path.abspath(__file__)), 'README.md')

0 commit comments

Comments
 (0)