pair_schnetpack.cpp: streamline variable names

atomistic-machine-learning · Nov 5, 2024 · bf4e1a5 · bf4e1a5
1 parent f78ed37
commit bf4e1a5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/interfaces/lammps/pair_schnetpack.cpp b/interfaces/lammps/pair_schnetpack.cpp
@@ -324,8 +324,8 @@ void PairSCHNETPACK::compute(int eflag, int vflag){
   if(debug_mode){
     std::cout << "SchNetPack model input:\n";
     std::cout << "_positions:\n" << positions_tensor << "\n";
-    std::cout << "_idx_i:\n" << edges_tensor[0] << "\n";
-    std::cout << "_idx_j:\n" << edges_tensor[1] << "\n";
+    std::cout << "_idx_i:\n" << idx_i_tensor << "\n";
+    std::cout << "_idx_j:\n" << idx_j_tensor << "\n";
     std::cout << "_idx_m:\n" << idx_m_tensor << "\n";
     std::cout << "_offsets:\n" << offsets_tensor << "\n";
     std::cout << "_cell:\n" << cell_tensor << "\n";