Unverified Commit 3e5137fe authored by Hongzhi (Steve), Chen's avatar Hongzhi (Steve), Chen Committed by GitHub
Browse files

[Misc] Autoformat python dgl (Part 2). (#5332)


Co-authored-by: default avatarUbuntu <ubuntu@ip-172-31-28-63.ap-northeast-1.compute.internal>
parent 6e58f5f1
...@@ -200,7 +200,7 @@ if wheel_include_libs: ...@@ -200,7 +200,7 @@ if wheel_include_libs:
fo.write( fo.write(
"include dgl/tensoradapter/%s/%s\n" % (backend, ta_name) "include dgl/tensoradapter/%s/%s\n" % (backend, ta_name)
) )
if backend == 'pytorch': if backend == "pytorch":
for sparse_path in glob.glob( for sparse_path in glob.glob(
os.path.join(dir_, "dgl_sparse", SPARSE_LIB_PATTERN) os.path.join(dir_, "dgl_sparse", SPARSE_LIB_PATTERN)
): ):
...@@ -213,10 +213,7 @@ if wheel_include_libs: ...@@ -213,10 +213,7 @@ if wheel_include_libs:
os.path.join(dir_, "dgl_sparse", sparse_name), os.path.join(dir_, "dgl_sparse", sparse_name),
os.path.join(CURRENT_DIR, "dgl", "dgl_sparse"), os.path.join(CURRENT_DIR, "dgl", "dgl_sparse"),
) )
fo.write( fo.write("include dgl/dgl_sparse/%s\n" % sparse_name)
"include dgl/dgl_sparse/%s\n" % sparse_name
)
setup_kwargs = {"include_package_data": True} setup_kwargs = {"include_package_data": True}
...@@ -240,13 +237,15 @@ if include_libs: ...@@ -240,13 +237,15 @@ if include_libs:
), ),
) )
) )
if backend == 'pytorch': if backend == "pytorch":
data_files.append( data_files.append(
( (
"dgl/dgl_sparse", "dgl/dgl_sparse",
glob.glob( glob.glob(
os.path.join( os.path.join(
os.path.dirname(os.path.relpath(path, CURRENT_DIR)), os.path.dirname(
os.path.relpath(path, CURRENT_DIR)
),
"dgl_sparse", "dgl_sparse",
SPARSE_LIB_PATTERN, SPARSE_LIB_PATTERN,
) )
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment