Fix a typo
Browse files- pipeline.py +2 -2
pipeline.py
CHANGED
@@ -160,7 +160,7 @@ class RepoEmbeddingPipeline(Pipeline):
|
|
160 |
for func in repo_info["funcs"]
|
161 |
]
|
162 |
entry["code_embeddings"] = code_embeddings
|
163 |
-
entry["
|
164 |
np.mean([x[1] for x in code_embeddings], axis=0).tolist()
|
165 |
if code_embeddings
|
166 |
else None
|
@@ -171,7 +171,7 @@ class RepoEmbeddingPipeline(Pipeline):
|
|
171 |
for doc in repo_info["docs"]
|
172 |
]
|
173 |
entry["doc_embeddings"] = doc_embeddings
|
174 |
-
entry["
|
175 |
np.mean([x[1] for x in doc_embeddings], axis=0).tolist()
|
176 |
if doc_embeddings
|
177 |
else None
|
|
|
160 |
for func in repo_info["funcs"]
|
161 |
]
|
162 |
entry["code_embeddings"] = code_embeddings
|
163 |
+
entry["mean_code_embedding"] = (
|
164 |
np.mean([x[1] for x in code_embeddings], axis=0).tolist()
|
165 |
if code_embeddings
|
166 |
else None
|
|
|
171 |
for doc in repo_info["docs"]
|
172 |
]
|
173 |
entry["doc_embeddings"] = doc_embeddings
|
174 |
+
entry["mean_doc_embedding"] = (
|
175 |
np.mean([x[1] for x in doc_embeddings], axis=0).tolist()
|
176 |
if doc_embeddings
|
177 |
else None
|