Atanu
Databricks Employee
Databricks Employee

You’ll need to either move the training to the run or manually log the model yourself if you want to log the model as an artifact of the run and register it the way you’re doing it.

I am not sure if that really help here.