Fix tests

lanl · Feb 1, 2024 · 9f9506f · 9f9506f
1 parent 11dfc6b
commit 9f9506f
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/scico/test/flax/test_clu.py b/scico/test/flax/test_clu.py
@@ -60,31 +60,31 @@ def test_count_parameters_empty():
 | Name | Shape | Size | Mean | Std |
 +------+-------+------+------+-----+
 +------+-------+------+------+-----+
-Total: 0"""
+Total weights: 0"""
 
 FLAX_CONV2D_PARAMETER_OVERVIEW = """+-------------+--------------+------+
 | Name        | Shape        | Size |
 +-------------+--------------+------+
 | conv/bias   | (2,)         | 2    |
 | conv/kernel | (3, 3, 3, 2) | 54   |
 +-------------+--------------+------+
-Total: 56"""
+Total weights: 56"""
 
 FLAX_CONV2D_PARAMETER_OVERVIEW_WITH_STATS = """+-------------+--------------+------+------+-----+
 | Name        | Shape        | Size | Mean | Std |
 +-------------+--------------+------+------+-----+
 | conv/bias   | (2,)         | 2    | 1.0  | 0.0 |
 | conv/kernel | (3, 3, 3, 2) | 54   | 1.0  | 0.0 |
 +-------------+--------------+------+------+-----+
-Total: 56"""
+Total weights: 56"""
 
 FLAX_CONV2D_MAPPING_PARAMETER_OVERVIEW_WITH_STATS = """+--------------------+--------------+------+------+-----+
 | Name               | Shape        | Size | Mean | Std |
 +--------------------+--------------+------+------+-----+
 | params/conv/bias   | (2,)         | 2    | 1.0  | 0.0 |
 | params/conv/kernel | (3, 3, 3, 2) | 54   | 1.0  | 0.0 |
 +--------------------+--------------+------+------+-----+
-Total: 56"""
+Total weights: 56"""
 
 # From https://github.com/google/CommonLoopUtils/blob/main/clu/parameter_overview_test.py
 def test_get_parameter_overview_empty():