keras-cv
keras-cv copied to clipboard
Fix the SAM end-to-end model predict test
Fixes some of the failures seen in https://github.com/keras-team/keras-cv/actions/runs/8259807616/job/22594330565?pr=2383
One of the tests in SAM changed the global dtype policy. Until now, if the test failed, the policy was never reset due to which other unrelated tests failed. This PR should fix this issue and reduce the failure count on this ci run.