From a868630b86e05e263792b2d843de84fa9c3e110a Mon Sep 17 00:00:00 2001
From: ekindogus <ekindogus@users.noreply.github.com>
Date: Mon, 8 Apr 2024 15:01:22 -0700
Subject: [PATCH] Update nequip.py

improving string formatting.
---
 jax_md/_nn/nequip.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/jax_md/_nn/nequip.py b/jax_md/_nn/nequip.py
index d9dc3da8..e233b9a8 100644
--- a/jax_md/_nn/nequip.py
+++ b/jax_md/_nn/nequip.py
@@ -192,7 +192,7 @@ def __call__(
 
     # we gather the instructions for the tp as well as the tp output irreps
     mode = 'uvu'
-    trainable = 'True'
+    trainable = True
     irreps_after_tp = []
     instructions = []