a07d174b57d3be7914fc1ce172c87ea7d487d241,tensorforce/core/objectives/objective.py,Objective,input_signature,#Objective#Any#,54

Before Change

            ]

        elif function == "reference":
            return [
                self.states_spec.signature(batched=True),
                TensorSpec(type="int", shape=(2,)).signature(batched=True),
                self.internals_spec.signature(batched=True),
                self.auxiliaries_spec.signature(batched=True),
                self.actions_spec.signature(batched=True),
                self.reward_spec.signature(batched=True)
            ]

        elif function == "comparative_loss":
            return [
                self.states_spec.signature(batched=True),

After Change

            )

        elif function == "reference":
            return SignatureDict(
                states=self.states_spec.signature(batched=True),
                horizons=TensorSpec(type="int", shape=(2,)).signature(batched=True),
                internals=self.internals_spec.signature(batched=True),
                auxiliaries=self.auxiliaries_spec.signature(batched=True),
                actions=self.actions_spec.signature(batched=True),
                reward=self.reward_spec.signature(batched=True)
            )

        elif function == "comparative_loss":
            return SignatureDict(
                states=self.states_spec.signature(batched=True),

Instances