Traceback (most recent call last):


 "/Users/crwilcox/workspace/google-cloud-python/pubsub/tests/unit/pubsub_v1/subscriber/test_scheduler.py", line 51, in test_schedule

scheduler_.shutdown()


 "/Users/crwilcox/workspace/google-cloud-python/pubsub/google/cloud/pubsub_v1/subscriber/scheduler.py", line 121, in shutdown

self._executor._work_queue. 
AttributeError: '_queue.SimpleQueue' object has no attribute 'queue'

Traceback (most recent call last):


 "/usr/local/google/home/schsam/.local/lib/python2.7/site-packages/absl/third_party/unittest3_backport/case.py", line 37, in testPartExecutor

yield


 "/usr/local/google/home/schsam/.local/lib/python2.7/site-packages/absl/third_party/unittest3_backport/case.py", line 162, in run

testMethod()


 "/usr/local/google/home/schsam/.local/lib/python2.7/site-packages/absl/testing/parameterized.py", line 262, in bound_param_test

test_method(self, **testcase_params)


 "simulate_test.py", line 164, in test_grad_through_nvt

assert grad(do_sim)(1.0) > 0.0


 "/usr/local/google/home/schsam/Source/jax/jax/api.py", line 235, in grad_f

_, g = value_and_grad_f(*args, **kwargs)


 "/usr/local/google/home/schsam/Source/jax/jax/api.py", line 289, in value_and_grad_f

g = vjp_py(onp.ones((), dtype=dtype))


 "/usr/local/google/home/schsam/Source/jax/jax/api_util.py", line 62, in apply_jaxtree_fun

ans = fun(*args)


 "/usr/local/google/home/schsam/Source/jax/jax/api.py", line 822, in out_vjp_packed

return out_vjp(cotangent_in)


 "/usr/local/google/home/schsam/Source/jax/jax/interpreters/ad.py", line 112, in vjp_

_, arg_cts = backward_pass(jaxpr, consts, (), dummy_args, dummy_primal_and_ct)


 "/usr/local/google/home/schsam/Source/jax/jax/interpreters/ad.py", line 180, in backward_pass

eqn.params, subjaxprs, sub_consts, sub_freevar_vals, invals, ct_in)


 "/usr/local/google/home/schsam/Source/jax/jax/interpreters/ad.py", line 536, in call_transpose

ans = primitive.bind(fun, all_args, **params)


 "/usr/local/google/home/schsam/Source/jax/jax/core.py", line 636, in call_bind

ans = primitive.impl(f, *args, **params)


 "/usr/local/google/home/schsam/Source/jax/jax/interpreters/xla.py", line 591, in xla_call_impl

compiled_fun = xla_callable(fun, device_values, *map(abstractify, args))


 "/usr/local/google/home/schsam/Source/jax/jax/linear_util.py", line 208, in memoized_fun

ans = call(f, *args)


 "/usr/local/google/home/schsam/Source/jax/jax/interpreters/xla.py", line 604, in xla_callable

jaxpr, (pval, consts, env) = pe.trace_to_subjaxpr(fun, master, False).call_wrapped(pvals)


 "/usr/local/google/home/schsam/Source/jax/jax/linear_util.py", line 147, in call_wrapped

ans = self.f(*args, **dict(self.params, **kwargs))


 "/usr/local/google/home/schsam/Source/jax/jax/interpreters/ad.py", line 186, in backward_pass

cts_out = get_primitive_transpose(eqn.primitive)(ct_in, *invals, **eqn.params)


 "/usr/local/google/home/schsam/Source/jax/jax/lax/lax.py", line 2818, in _scatter_add_transpose_rule

for i in xrange(len(t.shape)):
AttributeError: 'Zero' object has no attribute 'shape'
