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

Remove hardcoded class count assertions to support multiple datasets#3

zhulinchng merged 3 commits into

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

copilot/sub-pr-2zhulinchng/generative-data-augmentation:copilot/sub-pr-2Copy head branch name to clipboard