fix docstring for fractions

ChristopherMayes · Nov 20, 2024 · cbbf531 · cbbf531
1 parent 4834750
commit cbbf531
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/pmd_beamphysics/particles.py b/pmd_beamphysics/particles.py
@@ -1145,7 +1145,7 @@ def fractional_split(self, fractions: Union[float, int, list], key: str):
 
         Parameters
         ----------
-        fractions : float, int, or list of float/int
+        fractions : float or list of float
             A fraction or a list of fractions used for splitting the particles. All values must be between 0 and 1 (exclusive).
 
         key : str
@@ -1174,8 +1174,8 @@ def fractional_split(self, fractions: Union[float, int, list], key: str):
             raise ValueError("All fraction values must be between 0 and 1 (exclusive)")
 
         # Sort particles by the specified key
-        ixs = np.argsort(particles[key])
-        sorted_particles = particles[ixs]
+        ixs = np.argsort(self[key])
+        sorted_particles = self[ixs]
 
         # Sorted weights
         ws = sorted_particles.weight