remove superfluous print statement

lanl · Sep 19, 2024 · e904108 · e904108
1 parent 0805ce2
commit e904108
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/hippynn/molecular_dynamics/md.py b/hippynn/molecular_dynamics/md.py
@@ -483,11 +483,7 @@ def _update_data(self, model_outputs: dict):
             try:
                 self._data[f"output_{key}"].append(value.cpu().detach()[0])
             except KeyError:
-                self._data[f"output_{key}"] = [value.cpu().detach()[0]]
-
-            for key, value in variable._data.items():
-                print(variable.name, key, value.shape)
-
+                self._data[f"output_{key}"] = [value.cpu().detach()[0]]     
 
     def run(self, dt: float, n_steps: int, record_every: int = None):
         """Run `n_steps` of MD algorithm.