From 77b0602b03c2044d3455f8cf3dd4e3e3e66b34d4 Mon Sep 17 00:00:00 2001
From: CLU Authors <no-reply@google.com>
Date: Tue, 28 Jan 2025 06:13:56 -0800
Subject: [PATCH] Creating a NNX model with no bias results in a None being
 passed to the overview.

PiperOrigin-RevId: 720549126
---
 clu/parameter_overview.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/clu/parameter_overview.py b/clu/parameter_overview.py
index a47823a..3dc9337 100644
--- a/clu/parameter_overview.py
+++ b/clu/parameter_overview.py
@@ -98,6 +98,13 @@ def count_parameters(params: _ParamsContainer) -> int:
 
 
 def _make_row(name, value) -> _ParamRow:
+  if value is None:
+    return _ParamRow(
+        name=name,
+        shape=(),
+        dtype="",
+        size=0,
+    )
   return _ParamRow(
       name=name,
       shape=value.shape,