ea7c3f3ecce96dc852329a405d802c4d6f0be564,reagent/prediction/predictor_wrapper.py,DiscreteDqnWithPreprocessor,input_prototype,#DiscreteDqnWithPreprocessor#,46

Before Change

        return q_values

    def input_prototype(self):
        return (self.state_preprocessor.input_prototype(),)


class DiscreteDqnWithPreprocessorWithIdList(ModelBase):
    

After Change

        return q_values

    def input_prototype(self):
        return sparse_input_prototype(
            model=self.model,
            state_preprocessor=self.state_preprocessor,
            state_feature_config=self.state_feature_config,
        )


class DiscreteDqnPredictorWrapper(torch.jit.ScriptModule):
    def __init__(

Instances