Skip to content

Commit 25263a7

Browse files
Fix seeding in torch backend
1 parent d5049de commit 25263a7

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

src/probnum/backend/random/_torch.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -115,6 +115,4 @@ def _make_rng(seed: np.random.SeedSequence) -> torch.Generator:
115115
# state = seed.generate_state(_RNG_STATE_SIZE // 4, dtype=np.uint32)
116116
# rng.set_state(torch.ByteTensor(state.view(np.uint8)))
117117

118-
rng.manual_seed(int(seed.generate_state(1, dtype=np.uint64)[0]))
119-
120-
return rng
118+
return rng.manual_seed(int(seed.generate_state(1, dtype=np.int64)[0]))

0 commit comments

Comments
 (0)