A method includes creating one or more first policy shims to be applied to a ML/AI module, applying the one or more first policy shims to an input or an output of the ML/AI module and executing the ML/AI module on a data set in response to the applying step. The one or more first policy shims includes an input policy shim and an output policy shim and the applying step includes applying the input policy shim to the data set prior to the executing step and applying the output policy shim to an output of the executing step.