Skip to content

Commit 0ec471c

Browse files
committed
Remove seed setting in test_aggregator_output
Seed is already set to 0 because of the autoused fix_randomness fixture declared in conftest.py
1 parent e34c413 commit 0ec471c

1 file changed

Lines changed: 0 additions & 3 deletions

File tree

tests/unit/aggregation/test_values.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,3 @@
1-
import torch
21
from pytest import mark, param
32
from torch import Tensor, tensor
43
from torch.testing import assert_close
@@ -118,8 +117,6 @@
118117
def test_aggregator_output(A: Aggregator, J: Tensor, expected_output: Tensor) -> None:
119118
"""Test that the output values of an aggregator are fixed (on cpu)."""
120119

121-
if isinstance(A, GradVac):
122-
torch.manual_seed(0)
123120
assert_close(A(J), expected_output, rtol=0, atol=1e-4)
124121

125122

0 commit comments

Comments
 (0)