fix: missing var def

tests/test_optimizer.py

@@ -61,11 +61,11 @@ def test_reset(optimizer):
 @pytest.mark.parametrize("accum_steps", [1, 2, 3])
 @pytest.mark.parametrize("use_agc", [True, False])
 def test_parse_grad(optimizer, use_agc, accum_steps):
+    var = tf.Variable([1.0, 2.0], dtype=tf.float32)
     if accum_steps == 1:
         expected_grad = tf.zeros_like(var)  # gradients should not be applied yet
     else:
         expected_grad = tf.constant([3.0, 4.0])
-    var = tf.Variable([1.0, 2.0], dtype=tf.float32)
     optimizer.add_slot(var, "ga", initializer=expected_grad)
     accum_gradient = optimizer.get_slot(var, "ga")