From d9bef24f246e03381893b75b3daafc3e086b5549 Mon Sep 17 00:00:00 2001
From: rongaoli <78596449+rongaoli@users.noreply.github.com>
Date: Sun, 12 Oct 2025 22:09:24 +0800
Subject: [PATCH] Fix docstring for target_to_epsilon method

---
 flow_matching/path/affine.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/flow_matching/path/affine.py b/flow_matching/path/affine.py
index 81cb7ed..5b2ad5b 100644
--- a/flow_matching/path/affine.py
+++ b/flow_matching/path/affine.py
@@ -220,7 +220,7 @@ def velocity_to_epsilon(self, velocity: Tensor, x_t: Tensor, t: Tensor) -> Tenso
         return a_t * x_t + b_t * velocity
 
     def target_to_epsilon(self, x_1: Tensor, x_t: Tensor, t: Tensor) -> Tensor:
-        r"""Convert from x_1 representation to velocity.
+        r"""Convert from x_1 representation to epsilon.
 
         | given :math:`X_1`.
         | return :math:`\epsilon`.