From d0fcf3ea9657a32414b5f6084977ae1fb07e6bfb Mon Sep 17 00:00:00 2001
From: Vinam Arora <vinam@posteo.de>
Date: Thu, 28 Nov 2024 02:34:47 -0500
Subject: [PATCH] Comment fix

---
 torch_brain/models/poyo.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/torch_brain/models/poyo.py b/torch_brain/models/poyo.py
index 4dae68d..efb2015 100644
--- a/torch_brain/models/poyo.py
+++ b/torch_brain/models/poyo.py
@@ -138,7 +138,7 @@ def forward(
         output_session_index: TensorType["batch", "n_out", int],
         output_timestamps: TensorType["batch", "n_out", float],
     ) -> TensorType["batch", "n_out", "dim_out", float]:
-        """Forward pass of the POYO+ model.
+        """Forward pass of the POYO model.
 
         The model processes input spike sequences through its encoder-processor-decoder
         architecture to generate task-specific predictions.