Ax
Ax copied to clipboard
handle ObservationFeatures without trial_index in Relativize
Summary: This adds support for untransforming observation features in Relativize that do not have a trial index. This can occur when the Modelbridge does not use TrialAsTask and there are multiple SQ observations (e.g. for different trials) and we try to untransform the predictions for arms in a GR.
Differential Revision: D57126153