From 06089edfe65a4455e27179d51eefe3d8fa1c0e89 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Fran=C3=A7ois=20Rozet?= <francois.rozet@outlook.com>
Date: Thu, 9 Feb 2023 15:58:03 +0100
Subject: [PATCH] =?UTF-8?q?=F0=9F=A9=B9=20Fix=20FFJTransform=20without=20c?=
 =?UTF-8?q?ontext?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 zuko/flows.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/zuko/flows.py b/zuko/flows.py
index ff78042..1566adf 100644
--- a/zuko/flows.py
+++ b/zuko/flows.py
@@ -869,7 +869,7 @@ def forward(self, y: Tensor = None) -> Transform:
         return FreeFormJacobianTransform(
             f=partial(self.f, y=y),
             time=self.time,
-            phi=(y, *self.ode.parameters()),
+            phi=self.ode.parameters() if y is None else (y, *self.ode.parameters()),
             exact=self.exact,
         )