Week 1 Assignment Exercise 8 Issue

Hello everyone, I am running into a problem with the following exercise:
When I shift+enter the cell, there is no error, and I also created a sample instance of the class with e linear regression model and it seemed to work fine. However, it doesnt pass the auto grader.

I was confused because this exercise is literally about 4 lines of code and should not be a problem, but it is:

image removed as it contained grade cell codes.posting codes, link to graded assignment is considered violation of code of conduct and against community guidelines

This should help you debug/correct your codes :point_down:t2:

remember your data for your code assignment cell has been recalled as X and not data_treatment or data_control. Also you are suppose to use predict_proba when you used your previous self recalled functions.

Thank you very much for the assistance! I will implement those suggestions right now. Have a great day!