Hi @patrickrchao Great work on the repo. Just out of curiosity could this be adapted to test for distribution shift? If yes how would this be done? Cheers Andrew