Henry65
/

RepoSim4Py

Feature Extraction

code-understanding

text-embeddings-inference

Model card Files Files and versions

Henry65 commited on Jul 24, 2023

Commit

066b297

·

1 Parent(s): ac219c3

Update RepoPipeline.py

Files changed (1) hide show

RepoPipeline.py +1 -1

RepoPipeline.py CHANGED Viewed

@@ -201,7 +201,7 @@ class RepoPipeline(Pipeline):
                 info["mean_readme_embedding"] = torch.mean(readme_embeddings, dim=0).cpu().numpy()
                 info["code_embeddings_shape"] = info["code_embeddings"].shape
-                info["doc_embeddings_shape"] = info["doc_embeddings_shape"].shape
                 progress_bar.update(1)
                 model_outputs.append(info)

                 info["mean_readme_embedding"] = torch.mean(readme_embeddings, dim=0).cpu().numpy()
                 info["code_embeddings_shape"] = info["code_embeddings"].shape
+                info["doc_embeddings_shape"] = info["doc_embeddings"].shape
                 progress_bar.update(1)
                 model_outputs.append(info)