diff --git a/src/babyyoda/Histo2D_v2.py b/src/babyyoda/Histo2D_v2.py index e5cba3c..ff89066 100644 --- a/src/babyyoda/Histo2D_v2.py +++ b/src/babyyoda/Histo2D_v2.py @@ -141,7 +141,8 @@ def __getitem__(self, slices): if isinstance(slices, tuple): if len(slices) == 2: ix, iy = self.__get_indices(slices) - return self.__get_by_indices(ix, iy) + if isinstance(ix, int) and isinstance(iy, int): + return self.__get_by_indices(ix, iy) # TODO implement slice raise TypeError("Invalid argument type")