MilesCranmer commited on
Commit
3dafb8f
1 Parent(s): ebac605

Tweak deprecation test for UserWarning

Browse files
Files changed (1) hide show
  1. test/test.py +1 -1
test/test.py CHANGED
@@ -271,7 +271,7 @@ class TestMiscellaneous(unittest.TestCase):
271
  def test_deprecation(self):
272
  # Ensure that deprecation works as expected, with a warning,
273
  # and sets the correct value.
274
- with self.assertWarns(DeprecationWarning):
275
  model = PySRRegressor(fractionReplaced=0.2)
276
  # This is a deprecated parameter, so we should get a warning.
277
 
 
271
  def test_deprecation(self):
272
  # Ensure that deprecation works as expected, with a warning,
273
  # and sets the correct value.
274
+ with self.assertWarns(UserWarning):
275
  model = PySRRegressor(fractionReplaced=0.2)
276
  # This is a deprecated parameter, so we should get a warning.
277