Traceback (most recent call last):
  File "/workspace/shared/deep-learning-course/check_learner.py", line 55, in <module>
    check(args.module)
  File "/workspace/shared/deep-learning-course/check_learner.py", line 12, in check
    result=m.predict_batch(x,w,torch.tensor([.2]))
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/workspace/shared/deep-learning-course/exercises/stages.py", line 9, in predict_batch
    raise NotImplementedError('Implement the affine map.')
NotImplementedError: Implement the affine map.
