From 45d2b6529eff3dacb0bec62c6ebee67ea807db33 Mon Sep 17 00:00:00 2001
From: N_T <nils.thuerey @ tum.de>
Date: Tue, 12 Aug 2025 10:58:56 +0200
Subject: [PATCH] more phiflow 3.4 updates for HEAT SIP

---
 physgrad-code.ipynb | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/physgrad-code.ipynb b/physgrad-code.ipynb
index 02716e3..219463a 100644
--- a/physgrad-code.ipynb
+++ b/physgrad-code.ipynb
@@ -347,7 +347,7 @@
         "def loss_function(net, x_gt: CenteredGrid, sip: bool):\n",
         "    y_target = diffuse.fourier(x_gt, 8., 1)\n",
         "    with math.precision(32):\n",
-        "        prediction = field.native_call(net, field.to_float(y_target)).vector[0]\n",
+        "        prediction = field.native_call(net, field.to_float(y_target))\n",
         "        prediction += field.mean(x_gt) - field.mean(prediction)\n",
         "    x = field.stop_gradient(prediction)\n",
         "    if sip:\n",