Breast Cancer Assignment - Uncaught (in promise) TypeError: e is not a function

Hello,

I am stuck in the week 1 assignment of the course - Breast Cancer Classification. I get the following obscure error when I run the page:

tfjs@latest:17 Uncaught (in promise) TypeError: e is not a function
    at tfjs@latest:17:808714
    at tfjs@latest:17:159861
    at t.scopedRun (tfjs@latest:17:159990)
    at t.tidy (tfjs@latest:17:159758)
    at hN (tfjs@latest:17:220673)
    at t.transform (tfjs@latest:17:808692)
    at t.<anonymous> (tfjs@latest:17:798009)
    at c (tfjs@latest:17:105817)
    at Generator._invoke (tfjs@latest:17:105605)
    at Generator.next (tfjs@latest:17:106242)

This error is triggered by tfjs but comes from running the fitDataset function. I suspect there is something wrong with the model structure I made but I cannot seem to be able to identify the problem. Do you have any recommendations to breakdown the error message above, or ideas what could have been wrong? I can also share my model structure if that helps.