Henry65
/

RepoSim4Py

Feature Extraction

code-understanding

text-embeddings-inference

Model card Files Files and versions

Henry65 commited on Jul 24, 2023

Commit

10cd0e3

·

1 Parent(s): a93c513

Update RepoPipeline.py

Files changed (1) hide show

RepoPipeline.py +1 -1

RepoPipeline.py CHANGED Viewed

@@ -315,7 +315,7 @@ class RepoPipeline(Pipeline):
                     info["mean_doc_embedding"],
                     info["mean_requirement_embedding"],
                     info["mean_readme_embedding"]
-                ], axis=1)
                 # TODO Remove test
                 info["code_embeddings_shape"] = info["code_embeddings"].shape

                     info["mean_doc_embedding"],
                     info["mean_requirement_embedding"],
                     info["mean_readme_embedding"]
+                ], axis=0)
                 # TODO Remove test
                 info["code_embeddings_shape"] = info["code_embeddings"].shape