Merge pull request #8 from PhysicsLuis666/fix_hist

Fix histogram output filename error.

Author: austin-hoover

orbit_tools/diag.py

@@ -92,7 +92,8 @@ def __init__(
         **kwargs
     ) -> None:
         super().__init__(**kwargs)
-
+        
+        self.node = None
         self.index =0
         self.axis = axis
         self.ndim = len(axis)
@@ -166,7 +167,9 @@ def compute_histogram(self, bunch: Bunch) -> np.ndarray:
     def track(self, params_dict: dict) -> None:
         bunch_copy = Bunch()
         bunch = params_dict["bunch"]
+        node = params_dict["node"]
         bunch.copyBunchTo(bunch_copy)
+        self.node = node
 
         if self.transform is not None:
             bunch_copy = self.transform(bunch_copy)