Meng-Ling-Ori / Value_free

0 stars 0 forks source link

ToDo: `if devaluation` in `run agent()` #4

Open SaschaFroelich opened 3 years ago

SaschaFroelich commented 3 years ago

In run agent(), there is this piece of code:

    if devaluation: 
        if t < self.trials - self.trials_test:
            U = U[0]
        else:
            U = U[1]

Wouldn't it be better to keep U an array, and just change the corresponding entry? It would throw an error later anyway, when self.env.obtained_rewrads(t,U) is called, as that function expects an array of size > 2.

Meng-Ling-Ori commented 3 years ago

Thank you ! I will correct it.