From c172abfe4bb06407f1d0a778cae83eef79a011b2 Mon Sep 17 00:00:00 2001
From: Sopot Cela <93710482+sopotc@users.noreply.github.com>
Date: Sun, 13 Oct 2024 05:50:47 -0700
Subject: [PATCH] Minor fix for error message in base genome

---
 src/tensorneat/genome/base.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/tensorneat/genome/base.py b/src/tensorneat/genome/base.py
index bad52d4..d5b0be2 100644
--- a/src/tensorneat/genome/base.py
+++ b/src/tensorneat/genome/base.py
@@ -37,7 +37,7 @@ class BaseGenome(StatefulBaseClass):
             try:
                 _ = input_transform(jnp.zeros(num_inputs))
             except Exception as e:
-                raise ValueError(f"Output transform function failed: {e}")
+                raise ValueError(f"Input transform function failed: {e}")
 
         if output_transform is not None:
             try: