2222import distutils .ccompiler
2323
2424__version__ = os .environ .get ('NGRAPH_VERSION' , '0.0.0-dev' )
25- PYNGRAPH_SOURCE_DIR = os .path .abspath (os .path .dirname (__file__ ))
25+ PYNGRAPH_ROOT_DIR = os .path .abspath (os .path .dirname (__file__ ))
2626NGRAPH_DEFAULT_INSTALL_DIR = os .environ .get ('HOME' )
2727NGRAPH_ONNX_IMPORT_ENABLE = os .environ .get ('NGRAPH_ONNX_IMPORT_ENABLE' )
2828
@@ -50,7 +50,7 @@ def find_pybind_headers_dir():
5050 if os .environ .get ('PYBIND_HEADERS_PATH' ):
5151 pybind_headers_dir = os .environ .get ('PYBIND_HEADERS_PATH' )
5252 else :
53- pybind_headers_dir = os .path .join (PYNGRAPH_SOURCE_DIR , 'pybind11' )
53+ pybind_headers_dir = os .path .join (PYNGRAPH_ROOT_DIR , 'pybind11' )
5454
5555 found = os .path .exists (os .path .join (pybind_headers_dir , 'include/pybind11' ))
5656 if not found :
@@ -233,13 +233,13 @@ def cpp_flag(compiler):
233233]
234234
235235package_dir = {
236- 'ngraph' : PYNGRAPH_SOURCE_DIR + "/ngraph" ,
237- 'ngraph.utils' : PYNGRAPH_SOURCE_DIR + "/ngraph/utils" ,
238- 'ngraph.impl' : PYNGRAPH_SOURCE_DIR + "/ngraph/impl" ,
239- 'ngraph.impl.op' : PYNGRAPH_SOURCE_DIR + "/ngraph/impl/op" ,
240- 'ngraph.impl.op.util' : PYNGRAPH_SOURCE_DIR + "/ngraph/impl/op/util" ,
241- 'ngraph.impl.passes' : PYNGRAPH_SOURCE_DIR + "/ngraph/impl/passes" ,
242- 'ngraph.impl.runtime' : PYNGRAPH_SOURCE_DIR + "/ngraph/impl/runtime" ,
236+ 'ngraph' : PYNGRAPH_ROOT_DIR + "/ngraph" ,
237+ 'ngraph.utils' : PYNGRAPH_ROOT_DIR + "/ngraph/utils" ,
238+ 'ngraph.impl' : PYNGRAPH_ROOT_DIR + "/ngraph/impl" ,
239+ 'ngraph.impl.op' : PYNGRAPH_ROOT_DIR + "/ngraph/impl/op" ,
240+ 'ngraph.impl.op.util' : PYNGRAPH_ROOT_DIR + "/ngraph/impl/op/util" ,
241+ 'ngraph.impl.passes' : PYNGRAPH_ROOT_DIR + "/ngraph/impl/passes" ,
242+ 'ngraph.impl.runtime' : PYNGRAPH_ROOT_DIR + "/ngraph/impl/runtime" ,
243243}
244244packages = [
245245 'ngraph' ,
@@ -251,9 +251,9 @@ def cpp_flag(compiler):
251251 'ngraph.impl.runtime' ,
252252]
253253
254- sources = [PYNGRAPH_SOURCE_DIR + "/" + source for source in sources ]
254+ sources = [PYNGRAPH_ROOT_DIR + "/" + source for source in sources ]
255255
256- include_dirs = [PYNGRAPH_SOURCE_DIR , NGRAPH_CPP_INCLUDE_DIR , PYBIND11_INCLUDE_DIR ]
256+ include_dirs = [PYNGRAPH_ROOT_DIR , NGRAPH_CPP_INCLUDE_DIR , PYBIND11_INCLUDE_DIR ]
257257
258258library_dirs = [NGRAPH_CPP_LIBRARY_DIR ]
259259
@@ -274,13 +274,13 @@ def cpp_flag(compiler):
274274 (
275275 'licenses' ,
276276 [
277- PYNGRAPH_SOURCE_DIR + "/../licenses/" + license
278- for license in os .listdir (PYNGRAPH_SOURCE_DIR + "/../licenses" )
277+ PYNGRAPH_ROOT_DIR + "/../licenses/" + license
278+ for license in os .listdir (PYNGRAPH_ROOT_DIR + "/../licenses" )
279279 ],
280280 ),
281281 (
282282 '' ,
283- [PYNGRAPH_SOURCE_DIR + "/../LICENSE" ],
283+ [PYNGRAPH_ROOT_DIR + "/../LICENSE" ],
284284 )
285285]
286286
@@ -302,10 +302,10 @@ def cpp_flag(compiler):
302302 'pyngraph/pyngraph_onnx_import.cpp' ,
303303 'pyngraph/onnx_import/onnx_import.cpp' ,
304304 ]
305- onnx_sources = [PYNGRAPH_SOURCE_DIR + "/" + source for source in onnx_sources ]
305+ onnx_sources = [PYNGRAPH_ROOT_DIR + "/" + source for source in onnx_sources ]
306306
307307 package_dir ['ngraph.impl.onnx_import' ] = (
308- PYNGRAPH_SOURCE_DIR + "/ngraph/impl/onnx_import"
308+ PYNGRAPH_ROOT_DIR + "/ngraph/impl/onnx_import"
309309 )
310310 packages .append ('ngraph.impl.onnx_import' )
311311
@@ -360,17 +360,17 @@ def build_extensions(self):
360360 build_ext .build_extensions (self )
361361
362362
363- with open (os .path .join (PYNGRAPH_SOURCE_DIR , 'requirements.txt' )) as req :
363+ with open (os .path .join (PYNGRAPH_ROOT_DIR , 'requirements.txt' )) as req :
364364 requirements = req .read ().splitlines ()
365365
366366setup (
367367 name = 'ngraph-core' ,
368+ description = open (os .path .join (PYNGRAPH_ROOT_DIR , 'README.md' )).read (),
368369 version = __version__ ,
369370 author = 'Intel' ,
370371371372 url = 'https://ai.intel.com/' ,
372373 license = 'License :: OSI Approved :: Apache Software License' ,
373- description = 'Python API for nGraph' ,
374374 long_description = '' ,
375375 ext_modules = ext_modules ,
376376 package_dir = package_dir ,
0 commit comments