fix: resolve test failures by adding ML dependency checks

fix: resolve test failures by adding ML dependency checks #185