Bug Report for handwritten-digit-classifier

Bug Report for https://neetcode.io/problems/handwritten-digit-classifier

Please describe the bug below and include any steps to reproduce the bug or screenshots if possible.

The reference solution does not pass unit tests. Likely the initialization or RNG changed between versions.