Attempt to fix the tileddataset plot

DKISTDC · Dec 18, 2024 · 35d2ca1 · 35d2ca1
1 parent 0960f09
commit 35d2ca1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dkist/dataset/tests/test_tiled_dataset.py b/dkist/dataset/tests/test_tiled_dataset.py
@@ -81,7 +81,7 @@ def test_tileddataset_plot(share_zscale):
     newtiles = []
     for tile in ds.flat:
         newtiles.append(tile.rebin((1, 8, 8), operation=np.sum))
-    ds = TiledDataset(np.array(newtiles).reshape(ds.shape), inventory=ds.inventory)
+    ds = TiledDataset(np.array(newtiles).reshape(ds.shape), inventory=newtiles[0].inventory)
     fig = plt.figure(figsize=(600, 800))
     ds.plot(0, share_zscale=share_zscale)
     return plt.gcf()