Lazyhope
/

RepoSim

Feature Extraction

code-understanding

text-embeddings-inference

Model card Files Files and versions

Lazyhope commited on May 12, 2023

Commit

e063f35

·

1 Parent(s): e922192

Fix a typo

Files changed (1) hide show

pipeline.py +2 -2

pipeline.py CHANGED Viewed

@@ -160,7 +160,7 @@ class RepoEmbeddingPipeline(Pipeline):
                     for func in repo_info["funcs"]
                 ]
                 entry["code_embeddings"] = code_embeddings
-                entry["mean_code_embeddings"] = (
                     np.mean([x[1] for x in code_embeddings], axis=0).tolist()
                     if code_embeddings
                     else None
@@ -171,7 +171,7 @@ class RepoEmbeddingPipeline(Pipeline):
                     for doc in repo_info["docs"]
                 ]
                 entry["doc_embeddings"] = doc_embeddings
-                entry["mean_doc_embeddings"] = (
                     np.mean([x[1] for x in doc_embeddings], axis=0).tolist()
                     if doc_embeddings
                     else None

                     for func in repo_info["funcs"]
                 ]
                 entry["code_embeddings"] = code_embeddings
+                entry["mean_code_embedding"] = (
                     np.mean([x[1] for x in code_embeddings], axis=0).tolist()
                     if code_embeddings
                     else None
                     for doc in repo_info["docs"]
                 ]
                 entry["doc_embeddings"] = doc_embeddings
+                entry["mean_doc_embedding"] = (
                     np.mean([x[1] for x in doc_embeddings], axis=0).tolist()
                     if doc_embeddings
                     else None