removed print

BAMeScience · Feb 8, 2025 · 3da0bb5 · 3da0bb5
1 parent 30fdc88
commit 3da0bb5
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/equitrain/model_wrappers.py b/equitrain/model_wrappers.py
@@ -20,8 +20,6 @@ def forward(self, *args):
         if not isinstance(y_pred, dict):
             y_pred = {'energy': y_pred[0], 'forces': y_pred[1], 'stress': y_pred[2]}
 
-        print(y_pred)
-
         return y_pred