@@ -20,7 +20,6 @@ py_library(
20
20
name = "aux_readout" ,
21
21
srcs = ["aux_readout.py" ],
22
22
deps = [
23
- "//third_party/py/torch" ,
24
23
],
25
24
)
26
25
@@ -42,9 +41,6 @@ py_library(
42
41
"//programl/models:model" ,
43
42
"//programl/proto:programl_py" ,
44
43
"//programl/util/py:progress" ,
45
- "//third_party/py/absl" ,
46
- "//third_party/py/numpy" ,
47
- "//third_party/py/torch" ,
48
44
],
49
45
)
50
46
@@ -55,15 +51,13 @@ py_library(
55
51
deps = [
56
52
"//programl/graph/format/py:graph_tuple" ,
57
53
"//programl/proto:programl_py" ,
58
- "//third_party/py/numpy" ,
59
54
],
60
55
)
61
56
62
57
py_library (
63
58
name = "ggnn_layer" ,
64
59
srcs = ["ggnn_layer.py" ],
65
60
deps = [
66
- "//third_party/py/torch" ,
67
61
],
68
62
)
69
63
@@ -76,8 +70,6 @@ py_library(
76
70
":loss" ,
77
71
":metrics" ,
78
72
":node_embeddings" ,
79
- "//third_party/py/absl" ,
80
- "//third_party/py/torch" ,
81
73
],
82
74
)
83
75
@@ -88,23 +80,20 @@ py_library(
88
80
":ggnn_layer" ,
89
81
":messaging_layer" ,
90
82
":readout" ,
91
- "//third_party/py/torch" ,
92
83
],
93
84
)
94
85
95
86
py_library (
96
87
name = "linear_net" ,
97
88
srcs = ["linear_net.py" ],
98
89
deps = [
99
- "//third_party/py/torch" ,
100
90
],
101
91
)
102
92
103
93
py_library (
104
94
name = "loss" ,
105
95
srcs = ["loss.py" ],
106
96
deps = [
107
- "//third_party/py/torch" ,
108
97
],
109
98
)
110
99
@@ -114,32 +103,27 @@ py_library(
114
103
deps = [
115
104
":linear_net" ,
116
105
":position_embeddings" ,
117
- "//third_party/py/torch" ,
118
106
],
119
107
)
120
108
121
109
py_library (
122
110
name = "metrics" ,
123
111
srcs = ["metrics.py" ],
124
112
deps = [
125
- "//third_party/py/torch" ,
126
113
],
127
114
)
128
115
129
116
py_library (
130
117
name = "node_embeddings" ,
131
118
srcs = ["node_embeddings.py" ],
132
119
deps = [
133
- "//third_party/py/numpy" ,
134
- "//third_party/py/torch" ,
135
120
],
136
121
)
137
122
138
123
py_library (
139
124
name = "position_embeddings" ,
140
125
srcs = ["position_embeddings.py" ],
141
126
deps = [
142
- "//third_party/py/torch" ,
143
127
],
144
128
)
145
129
@@ -148,6 +132,5 @@ py_library(
148
132
srcs = ["readout.py" ],
149
133
deps = [
150
134
":linear_net" ,
151
- "//third_party/py/torch" ,
152
135
],
153
136
)
0 commit comments