You need to sign in or sign up before continuing.
Unverified Commit f81d68aa authored by Nicolas Hug's avatar Nicolas Hug Committed by GitHub
Browse files

Fix test RNG (#7693)

parent f02caa53
......@@ -22,6 +22,7 @@ from common_utils import (
make_image,
make_images,
make_segmentation_mask,
set_rng_seed,
)
from torch import nn
from torchvision import datapoints, transforms as legacy_transforms
......@@ -35,6 +36,12 @@ from torchvision.transforms.v2.utils import query_spatial_size
DEFAULT_MAKE_IMAGES_KWARGS = dict(color_spaces=["RGB"], extra_dims=[(4,)])
@pytest.fixture(autouse=True)
def fix_rng_seed():
set_rng_seed(0)
yield
class NotScriptableArgsKwargs(ArgsKwargs):
"""
This class is used to mark parameters that render the transform non-scriptable. They still work in eager mode and
......
......@@ -21,6 +21,7 @@ from common_utils import (
make_image,
make_segmentation_mask,
make_video,
set_rng_seed,
)
from torch.testing import assert_close
from torchvision import datapoints
......@@ -28,6 +29,12 @@ from torchvision.transforms.functional import pil_modes_mapping
from torchvision.transforms.v2 import functional as F
@pytest.fixture(autouse=True)
def fix_rng_seed():
set_rng_seed(0)
yield
def _to_tolerances(maybe_tolerance_dict):
if not isinstance(maybe_tolerance_dict, dict):
return dict(rtol=None, atol=None)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment