Update RepoPipeline.py

RepoPipeline.py

@@ -309,12 +309,12 @@ class RepoPipeline(Pipeline):
                 info["mean_readme_embedding"] = torch.mean(readme_embeddings, dim=0).cpu().numpy()
 
                 # Repo-level mean embedding
-                info["mean_repo_embedding"] = torch.cat([
+                info["mean_repo_embedding"] = np.concatenate([
                     info["mean_code_embedding"],
                     info["mean_doc_embedding"],
                     info["mean_requirement_embedding"],
                     info["mean_readme_embedding"]
-                ], dim=1)
+                ], axis=1)
 
                 # TODO Remove test
                 info["code_embeddings_shape"] = info["code_embeddings"].shape