525f94ff0bfdb617175791a36f37bdd4ccb808d0,ml/rl/test/preprocessing/test_feature_extractor.py,TestTrainingFeatureExtractor,test_extract_sarsa_discrete_action,#TestTrainingFeatureExtractor#,232

Before Change

        )
        // Setup
        ws, net = self.create_ws_and_net(extractor)
        input_record = net.input_record() + schema.NewRecord(
            net, schema.Struct(("reward", schema.Scalar()))
        )
        self.setup_state_features(ws, input_record.state_features)
        self.setup_next_state_features(ws, input_record.next_state_features)
        action = self.setup_action(ws, input_record.action)
        next_action = self.setup_next_action(ws, input_record.next_action)

After Change

        )
        // Setup
        ws, net = self.create_ws_and_net(extractor)
        input_record = self.create_extra_input_record(net)
        self.setup_state_features(ws, input_record.state_features)
        self.setup_next_state_features(ws, input_record.next_state_features)
        action = self.setup_action(ws, input_record.action)
        next_action = self.setup_next_action(ws, input_record.next_action)
        reward = self.setup_reward(ws, input_record.reward)
        extra_data = self.setup_extra_data(ws, input_record)
        // Run
        ws.run(net)
        res = extractor.extract(ws, input_record, net.output_record())
        o = res.training_input

Instances