From 4f0558e6ef1fd822852969dbb068efa1cf363cb4 Mon Sep 17 00:00:00 2001
From: Ryan Grout <ryan@ryangrout.org>
Date: Wed, 14 Feb 2024 12:48:54 -0700
Subject: [PATCH] Rework fill_depressions to use _output_handler.

---
 pysheds/sgrid.py | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/pysheds/sgrid.py b/pysheds/sgrid.py
index f0b3229..71625e2 100644
--- a/pysheds/sgrid.py
+++ b/pysheds/sgrid.py
@@ -2148,11 +2148,13 @@ def fill_depressions(self, dem, nodata_out=np.nan, **kwargs):
                       Raster representing digital elevation data with multi-celled
                       depressions removed.
         """
-        # Implementation detail of priority flood algorithm.
-        tuple_type = Tuple([from_dtype(dem.dtype), int64, int64])
         dem_mask = self._get_nodata_cells(dem)
-        return _self._priority_flood(dem, dem_mask, tuple_type)
-
+        result = _self._priority_flood(dem, dem_mask)
+        dem_filled = self._output_handler(data=result,
+                                        viewfinder=dem.viewfinder,
+                                        metadata=dem.metadata,
+                                        nodata=dem.nodata)
+        return dem_filled
 
     def detect_flats(self, dem, **kwargs):
         """