chore: fix retrieval logic for squad dataset generation to handle golden sample logic properly#3

daved01 merged 1 commit intomaindeconvolute-labs/benchmarks:mainfrom

chore/squadfixdeconvolute-labs/benchmarks:chore/squadfixCopy head branch name to clipboard