From 00c99f30d92b18ffd92e030d7b914ac1d4b8b59f Mon Sep 17 00:00:00 2001 From: Mohit Kumar Date: Sun, 17 Nov 2024 11:00:40 +0530 Subject: [PATCH] Add test for 'verify_grad' without explicit RNG --- pytensor/gradient.py | 9 ++------- 1 file changed, 2 insertions(+), 7 deletions(-) diff --git a/pytensor/gradient.py b/pytensor/gradient.py index 5946a20dd4..ae2aac043c 100644 --- a/pytensor/gradient.py +++ b/pytensor/gradient.py @@ -1770,14 +1770,9 @@ def verify_grad( if rel_tol is None: rel_tol = max(_type_tol[str(p.dtype)] for p in pt) + # Initialize RNG if not provided if rng is None: - raise TypeError( - "rng should be a valid instance of " - "numpy.random.RandomState. You may " - "want to use tests.unittest" - "_tools.verify_grad instead of " - "pytensor.gradient.verify_grad." - ) + rng = np.random.default_rng() # We allow input downcast in `function`, because `numeric_grad` works in # the most precise dtype used among the inputs, so we may need to cast