From 0ca7d90240113d53855e4fabdfd19481c473da09 Mon Sep 17 00:00:00 2001
From: Ameya Daigavane <ameya.d.98@gmail.com>
Date: Fri, 29 Nov 2024 13:10:22 -0500
Subject: [PATCH] Fix.

---
 e3nn_jax/_src/so3grid.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/e3nn_jax/_src/so3grid.py b/e3nn_jax/_src/so3grid.py
index d4d5c3c..6a551a0 100644
--- a/e3nn_jax/_src/so3grid.py
+++ b/e3nn_jax/_src/so3grid.py
@@ -20,7 +20,7 @@ def __init__(
         if _perform_checks:
             if len(s2_signals.shape) < 3:
                 raise ValueError(
-                    f"Grid values should have atleast 3 axes. Got grid_values of shape {grid_values.shape}."
+                    f"s2_signals should have atleast 3 axes. Got {s2_signals.shape}."
                 )
 
         self.s2_signals = s2_signals