create stub dist-info for onnxruntime in ONNX-Runtime v1.10.0 with CUDA

Author: lexming

easybuild/easyconfigs/o/ONNX-Runtime/ONNX-Runtime-1.10.0-foss-2021a-CUDA-11.3.1.eb

@@ -15,6 +15,7 @@ and transforms."""
 
 toolchain = {'name': 'foss', 'version': '2021a'}
 
+# ONNX Runtime with support for GPUs is distributed under the name 'onnxruntime-gpu' in PyPI
 source_urls = ['http://pypi.python.org/packages/source/o/onnxruntime-gpu']
 local_whl_tmpl = 'onnxruntime_gpu-%%(version)s-cp39-cp39-manylinux_2_17_%s.manylinux2014_%s.whl'
 sources = [local_whl_tmpl % ('%(arch)s', '%(arch)s')]
@@ -35,8 +36,23 @@ dependencies = [
 use_pip = True
 download_dep_fail = True
 
+# Due to its name 'onnxruntime-gpu', this package does not fullfil requirements on 'onnxruntime' although it provides
+# the 'onnxruntime' python module. Fix this dependency issue in pip by creating a stub 'onnxruntime' dist-info folder
+_py_sitepkgs = '%(installdir)s/lib/python%(pyshortver)s/site-packages'
+postinstallcmds = [
+    "cp -r %s/onnxruntime{_gpu,}-%%(version)s.dist-info" % _py_sitepkgs,
+    "sed -i 's/onnxruntime.gpu/onnxruntime/g' %s/onnxruntime-%%(version)s.dist-info/{METADATA,RECORD}" % _py_sitepkgs,
+]
+
 sanity_pip_check = True
 
+_py_sitepkgs_dirs = ['onnxruntime', 'onnxruntime-%(version)s.dist-info', 'onnxruntime_gpu-%(version)s.dist-info']
+
+sanity_check_paths = {
+    'files': [],
+    'dirs': ['lib/python%%(pyshortver)s/site-packages/%s' % x for x in _py_sitepkgs_dirs],
+}
+
 options = {'modulename': 'onnxruntime'}
 
 moduleclass = 'devel'