Traceback (most recent call last):
  File "c:\Users\User\Desktop\debugrem\clawd-one-and-only-one-shot\run_mnist_eval.py", line 70, in <module>
    evaluate_mnist()
    ~~~~~~~~~~~~~~^^
  File "c:\Users\User\Desktop\debugrem\clawd-one-and-only-one-shot\run_mnist_eval.py", line 55, in evaluate_mnist
    stats = agent.evaluate(test_images, test_labels)
  File "c:\Users\User\Desktop\debugrem\clawd-one-and-only-one-shot\hippocampaif\agent\mnist_agent.py", line 134, in evaluate
    result = self.classify(images[i])
  File "c:\Users\User\Desktop\debugrem\clawd-one-and-only-one-shot\hippocampaif\agent\mnist_agent.py", line 106, in classify
    result = self.classifier.classify(image, features=features)
  File "c:\Users\User\Desktop\debugrem\clawd-one-and-only-one-shot\hippocampaif\learning\one_shot_classifier.py", line 145, in classify
    opt_result = self.amgd.optimize(
        image, exemplar['image'], self.canvas
    )
  File "c:\Users\User\Desktop\debugrem\clawd-one-and-only-one-shot\hippocampaif\learning\amgd.py", line 101, in optimize
    dD_du, dD_dv = canvas.compute_color_gradient(
                   ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~^
        level_img1, level_img2, u, v
        ^^^^^^^^^^^^^^^^^^^^^^^^^^^^
    )
    ^
  File "c:\Users\User\Desktop\debugrem\clawd-one-and-only-one-shot\hippocampaif\learning\distortable_canvas.py", line 175, in compute_color_gradient
    grad_y, grad_x = np.gradient(warped)
                     ~~~~~~~~~~~^^^^^^^^
  File "C:\Users\User\Desktop\debugrem\clawd-one-and-only-one-shot\.venv\Lib\site-packages\numpy\lib\_function_base_impl.py", line 1313, in gradient
    out[tuple(slice1)] = (f[tuple(slice4)] - f[tuple(slice2)]) / (2. * ax_dx)
        ~~~~~^^^^^^^^
KeyboardInterrupt