Fix deepsource issues and failing tests in save_geotif

danforthcenter · Sep 17, 2024 · 93eaa69 · 93eaa69
1 parent 2dafa76
commit 93eaa69
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/plantcv/geospatial/save_geotif.py b/plantcv/geospatial/save_geotif.py
@@ -1,7 +1,6 @@
 # Save georeferenced TIF files from plantcv segmented masks
 
 import rasterio
-from rasterio.mask import mask
 
 
 def save_geotif(out_img, in_img, out_path, nodata=None):
@@ -34,7 +33,7 @@ def save_geotif(out_img, in_img, out_path, nodata=None):
     # For color outputs
     if len(out_img.shape) == 3:
         # Reorder dimensions
-        out_img = out_img.transpose(2,0,1)
+        out_img = out_img.transpose(2, 0, 1)
         # Reorder channels if needed
         chord = [waves[480], waves[560], waves[650]]
         # For multispec

diff --git a/tests/test_geospatial_save_geotif.py b/tests/test_geospatial_save_geotif.py
@@ -12,7 +12,7 @@ def test_save_geotif_color(test_data, tmpdir):
     # Create test image
     img = read_geotif(filename=test_data.cropped_tif, bands="B,G,R,RE,N")
     filename = os.path.join(cache_dir, 'test.tif')
-    save_geotif(out_img=img, in_img=img, out_path=filename)
+    save_geotif(out_img=img.array_data, in_img=img, out_path=filename)
     # Assert that the file was created
     assert os.path.exists(filename)