Error in w4_unittest.test_classify

All unit tests pass up to this point. I do get a mismatch for the test below classify():

Accuracy 0.3796875

Expected Result
Accuracy ~0.69

The unit test result:

→ 31 v1, v2 = model([q1, q2])
32
33 for j in range(batch_size): TypeError: cannot unpack non-iterable NoneType object

I’ve tried restarting the kernel. Same results.

After a little sleuthing, this happens the very first time “model” is called.

Any ideas?

Thanks!

v1, v2 = model((q1, q2))

I’m seeing this, too. On further investigation, I see that in the unittest, model is of type <class 'method'> rather than <class 'trax.layers.combinators.Parallel'>. :thinking: