utils.py 641 B

1234567891011121314151617181920
  1. import re
  2. def to_snake_case(class_name: str) -> str:
  3. """Converts class name to snake case.
  4. This is used to unify metrics names when using class names within.
  5. Args:
  6. class_name: A string defining a class name usually in camel
  7. case.
  8. Returns:
  9. The class name in snake case.
  10. """
  11. # Insert _ between a lower- or digit-char and an upper-char
  12. name = re.sub(r"(?<=[a-z0-9])(?=[A-Z])", "_", class_name)
  13. # Insert _ between an upper-char followed by upper+lower (to split
  14. # "ABCReward" into "abc_reward")
  15. name = re.sub(r"(?<=[A-Z])(?=[A-Z][a-z])", "_", name)
  16. return name.lower()