Problem regarding Ecole with PyTorch

[CenturionMKIII]

Hi,

I was trying to extract features with the Khalil extraction to train my PyTorch model, but then I found out that the extraction was not deterministic even by fixing the seeds(env.seed and ecole.seed). I checked my code and noticed that if I remove the line

model.to(device)

the extraction became non-deterministic. Does anyone ever have the same problem before? Am I doing anything wrong?

Thanks in advance.

My code is something like this:

import ecole
ecole.seed(0)
model=MODEL()
model.to(device)
env=ecole.environment.Branching(PARAMS)
env.seed(0)
_,_,_,_,_=env.reset(FILE_NAME)