From 562d3c7d030bded34726bfb15c04b6e8a97d0d06 Mon Sep 17 00:00:00 2001
From: Steven Pawley <dr.stevenpawley@gmail.com>
Date: Mon, 20 May 2024 10:54:12 -0600
Subject: [PATCH] allow creating a Raster from a list of rasterio.band objects

---
 pyspatialml/raster.py | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/pyspatialml/raster.py b/pyspatialml/raster.py
index 2b3a2c9..d5f099e 100644
--- a/pyspatialml/raster.py
+++ b/pyspatialml/raster.py
@@ -501,10 +501,17 @@ def __init__(
                         src_layers.append(RasterLayer(band))
                 self._layers = src_layers
                 return
+        
+            # from a list of rasterio.band objects
+            elif all(isinstance(x, rasterio.Band) for x in src):
+                src_layers = []
+                for band in src:
+                    src_layers.append(RasterLayer(band))
+                self._layers = src_layers
+                return
         else:
             raise ValueError("Cannot create a Raster object from a mixture of inputs")
         
-                # normal
         # try:
         #     src_layers = []
         #     for band in src: