Remove hardcoded class count assertions to support multiple datasets - #4

Remove hardcoded class count assertions to support multiple datasets#4

zhulinchng merged 1 commit into

code-simplificationzhulinchng/generative-data-augmentation:code-simplificationfrom

copilot/sub-pr-2-againzhulinchng/generative-data-augmentation:copilot/sub-pr-2-againCopy head branch name to clipboard