fix name

Author: priyakasimbeg

reference_algorithms/qualification_baselines/external_tuning/jax_nadamw_target_setting.py

@@ -274,8 +274,8 @@ def update_params(
   else:
     grad_clip = None
 
-    # Create shardings for each argument
-  replicated = jax_sharding_utils.get_replicated_sharding()  # No partitioning
+  # Create shardings for each argument
+  replicated = jax_sharding_utils.get_replicate_sharding()  # No partitioning
   sharded = jax_sharding_utils.get_batch_dim_sharding()  # Partition along batch dimension
 
   # Create the sharding rules for each argument