Missing imports

Signed-off-by: João Lucas de Sousa Almeida <[email protected]>

Author: Joao-L-S-Almeida

terratorch/datasets/generic_pixel_wise_dataset.py

@@ -8,7 +8,8 @@
 from functools import partial
 from pathlib import Path
 from typing import Any, List, Union
-
+from functools import reduce
+import operator
 import albumentations as A
 import matplotlib as mpl
 import numpy as np