Skip to content

Commit b1fc728

Browse files
authored
Fix the path of samples in count graphnet. (#157)
1 parent 822b839 commit b1fc728

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

tools/count_sample.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,13 @@
11
# 统计 samples 目录下, graph_net.json 文件的数量
22
import os
33

4+
5+
filename = os.path.abspath(__file__)
6+
root_dir = os.path.dirname(os.path.dirname(filename))
7+
samples_dir = os.path.join(root_dir, "samples")
8+
49
graph_net_count = 0
5-
for root, dirs, files in os.walk("../samples"):
10+
for root, dirs, files in os.walk(samples_dir):
611
for file in files:
712
if file == "graph_net.json":
813
graph_net_count += 1

0 commit comments

Comments
 (0)