From 80cf5900d3918b9bd6762724b701937ac038fa3e Mon Sep 17 00:00:00 2001
From: Nathan Simpson <nsimpson@turing.ac.uk>
Date: Tue, 17 Sep 2024 12:12:04 +0100
Subject: [PATCH] type parameter check

---
 src/cloudcasting/utils.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/cloudcasting/utils.py b/src/cloudcasting/utils.py
index a495f7b..19ce71f 100644
--- a/src/cloudcasting/utils.py
+++ b/src/cloudcasting/utils.py
@@ -26,7 +26,7 @@
 def lon_lat_to_geostationary_area_coords(
     x: Sequence[float],
     y: Sequence[float],
-    xr_data: xr.Dataset,
+    xr_data: xr.Dataset | xr.DataArray,
 ) -> tuple[Sequence[float], Sequence[float]]:
     """Loads geostationary area and change from lon-lat to geostationaery coords
     Args: