Updated `parse_policy_info` function in augment.py

[LakshmiKalaKadali]

This PR is raised to add more flexible control over the randomness in the augmentation process by changing level += tf.random.normal([], dtype=tf.float32) to level += level_std * tf.random.normal([], dtype=tf.float32) to the function parse_policy_info instead of standard deviation always 1.