Comment fix

neuro-galaxy · Nov 28, 2024 · d0fcf3e · d0fcf3e
1 parent 69230b9
commit d0fcf3e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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.