small fix restart

run_mlm_flax.py

@@ -508,7 +508,7 @@ if __name__ == "__main__":
         init_value=0.0, end_value=training_args.learning_rate, transition_steps=training_args.warmup_steps
     )
 
-    if data_argsdata_argtatic_learning_rate:
+    if data_args.static_learning_rate:
         end_lr_value = training_args.learning_rate
     else:
         end_lr_value = 0