@@ -337,7 +337,7 @@ def op_record(name, nargs, dtypes, shapes, rng_factory, diff_modes,
337337 # TODO(phawkins): np.unwrap does not correctly promote its default period
338338 # argument under NumPy 1.21 for bfloat16 inputs. It works fine if we
339339 # explicitly pass a bfloat16 value that does not need promition. We should
340- # probably add a custom test harness for unwrap that tests_version2 the period
340+ # probably add a custom test harness for unwrap that tests the period
341341 # argument anyway.
342342 op_record ("unwrap" , 1 , [t for t in float_dtypes if t != dtypes .bfloat16 ],
343343 nonempty_nonscalar_array_shapes ,
@@ -539,7 +539,7 @@ def _promote_like_jnp(fun, inexact=False):
539539 """Decorator that promotes the arguments of `fun` to `jnp.result_type(*args)`.
540540
541541 jnp and np have different type promotion semantics; this decorator allows
542- tests_version2 make an np reference implementation act more like an jnp
542+ tests make an np reference implementation act more like an jnp
543543 implementation.
544544 """
545545 _promote = _promote_dtypes_inexact if inexact else _promote_dtypes
@@ -3056,7 +3056,7 @@ def testIdentity(self, n, dtype):
30563056 for left in [None , 0 ]
30573057 for right in [None , 1 ]
30583058 for dtype in default_dtypes
3059- # following types lack precision for meaningful tests_version2
3059+ # following types lack precision for meaningful tests
30603060 if dtype not in [np .int8 , np .int16 , np .float16 , jnp .bfloat16 ]
30613061 ))
30623062 def testInterp (self , shape , dtype , period , left , right ):
@@ -3562,7 +3562,7 @@ def testHistogramBinEdges(self, shape, dtype, bins, range, weights):
35623562 for shape in [(5 ,), (5 , 5 )]
35633563 for dtype in default_dtypes
35643564 # We only test explicit integer-valued bin edges because in other cases
3565- # rounding errors lead to flaky tests_version2 .
3565+ # rounding errors lead to flaky tests .
35663566 for bins in [np .arange (- 5 , 6 ), np .array ([- 5 , 0 , 3 ])]
35673567 for density in [True , False ]
35683568 for weights in [True , False ]
@@ -5108,7 +5108,7 @@ def testLongLong(self):
51085108 @jtu .ignore_warning (category = UserWarning ,
51095109 message = "Explicitly requested dtype.*" )
51105110 def testArange (self ):
5111- # test cases inspired by dask tests_version2 at
5111+ # test cases inspired by dask tests at
51125112 # https://github.com/dask/dask/blob/main/dask/array/tests/test_creation.py#L92
51135113 np_arange = jtu .with_jax_dtype_defaults (np .arange )
51145114 self .assertAllClose (bm .arange (77 ).value ,
@@ -5777,7 +5777,7 @@ def testDisableNumpyRankPromotionBroadcasting(self):
57775777 FLAGS .jax_numpy_rank_promotion = prev_flag
57785778
57795779 def testStackArrayArgument (self ):
5780- # tests_version2 https://github.com/google/jax/issues/1271
5780+ # tests https://github.com/google/jax/issues/1271
57815781 @jax .jit
57825782 def foo (x ):
57835783 return bm .stack (x )
@@ -6050,7 +6050,7 @@ def testFromString(self):
60506050 self .assertArraysEqual (expected , actual )
60516051
60526052
6053- # Most grad tests_version2 are at the lax level (see lax_test.py), but we add some here
6053+ # Most grad tests are at the lax level (see lax_test.py), but we add some here
60546054# as needed for e.g. particular compound ops of interest.
60556055
60566056GradTestSpec = collections .namedtuple (
@@ -6126,7 +6126,7 @@ def testOpGradSpecialValue(self, op, special_value, order):
61266126 atol = {np .float32 : 3e-3 })
61276127
61286128 def testSincGradArrayInput (self ):
6129- # tests_version2 for a bug almost introduced in #5077
6129+ # tests for a bug almost introduced in #5077
61306130 jax .grad (lambda x : bm .sinc (x ).sum ())(jnp .arange (10. )) # doesn't crash
61316131
61326132 def testTakeAlongAxisIssue1521 (self ):
0 commit comments