From db9290ba05c7ca8895668b17c4c25a21edbd40f9 Mon Sep 17 00:00:00 2001
From: Chaithya G R <chaithyagr@gmail.com>
Date: Fri, 6 Dec 2024 15:10:50 +0100
Subject: [PATCH] Add support for density comp

---
 examples/example_learn_samples_multires.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/examples/example_learn_samples_multires.py b/examples/example_learn_samples_multires.py
index f84737dd..375fcfc2 100644
--- a/examples/example_learn_samples_multires.py
+++ b/examples/example_learn_samples_multires.py
@@ -83,6 +83,7 @@ def __init__(
         self.operator = get_operator("finufft", wrt_data=True, wrt_traj=True)(
             sample_points,
             shape=img_size,
+            density=True,
             squeeze_dims=False,
         )
         self.img_size = img_size