Discard the temporary weight field

yt-project · Dec 6, 2024 · 1ab751b · 1ab751b
1 parent 6abd8fc
commit 1ab751b
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/yt/visualization/volume_rendering/off_axis_projection.py b/yt/visualization/volume_rendering/off_axis_projection.py
@@ -498,6 +498,7 @@ def temp_weightfield(field, data):
             registry=data_source.ds.unit_registry,
             info={"imtype": "rendering"},
         )
+
     else:
         for grid, mask in data_source.blocks:
             data = []
@@ -520,12 +521,14 @@ def temp_weightfield(field, data):
             vol.sampler(pg, num_threads=num_threads)
 
         image = vol.finalize_image(camera, vol.sampler.aimage)
+
         image = ImageArray(
             image, funits, registry=data_source.ds.unit_registry, info=image.info
         )
 
-        if weight is not None:
-            data_source.ds.field_info.pop(("index", "temp_weightfield"))
+    # Remove the temporary weight field
+    if weight is not None:
+        data_source.ds.field_info.pop(("index", "temp_weightfield"))
 
     if method == "integrate":
         if weight is None: