mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-19 05:16:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

py3 iterator compatibility

Browse Source
This commit is contained in:
Roy Frostig 2018-12-07 10:22:15 -08:00
parent db748a046f
commit 479f256abd
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/generated_fun_test.py
View File

@ -98,7 +98,7 @@ def eval_fun(fun, *args):
counter = it.count()
def fresh_var(ty):
return Var(counter.next(), ty)
return Var(next(counter), ty)
def gen_array_type(size):
# TODO(dougalm): randomize this