topic ModuleNotFoundError: No module named 'model_train' when using mlflow.sklearn.load_model in Machine Learning https://community.databricks.com/t5/machine-learning/modulenotfounderror-no-module-named-model-train-when-using/m-p/83965#M3600 <P>Hello,</P><P>I have multiple versions of a model registered in model registry. When I am trying to load any other version except model version 1 by&nbsp;</P><DIV><DIV><EM>mlflow.sklearn.load_model(f"models:/{model_name}/{model_version}")</EM></DIV><DIV>I am getting <STRONG>ModuleNotFoundError: No module named 'model_train'.&nbsp;</STRONG></DIV><DIV>&nbsp;</DIV><DIV>All the versions have been registered successfully, below is the log when the latest version was registered.</DIV><DIV><EM>Registered model &lt;model_name&gt; already exists. Creating a new version of this model... </EM></DIV><DIV><EM>2024/08/22 13:22:58 INFO mlflow.store.model_registry.abstract_store: Waiting up to 300 seconds for model version to finish creation. Model name: &lt;model_name&gt;, version 7 </EM></DIV><DIV><EM>Created version '7' of model &lt;model_name&gt;.</EM></DIV></DIV> Thu, 22 Aug 2024 18:05:42 GMT yorabhir 2024-08-22T18:05:42Z ModuleNotFoundError: No module named 'model_train' when using mlflow.sklearn.load_model https://community.databricks.com/t5/machine-learning/modulenotfounderror-no-module-named-model-train-when-using/m-p/83965#M3600 <P>Hello,</P><P>I have multiple versions of a model registered in model registry. When I am trying to load any other version except model version 1 by&nbsp;</P><DIV><DIV><EM>mlflow.sklearn.load_model(f"models:/{model_name}/{model_version}")</EM></DIV><DIV>I am getting <STRONG>ModuleNotFoundError: No module named 'model_train'.&nbsp;</STRONG></DIV><DIV>&nbsp;</DIV><DIV>All the versions have been registered successfully, below is the log when the latest version was registered.</DIV><DIV><EM>Registered model &lt;model_name&gt; already exists. Creating a new version of this model... </EM></DIV><DIV><EM>2024/08/22 13:22:58 INFO mlflow.store.model_registry.abstract_store: Waiting up to 300 seconds for model version to finish creation. Model name: &lt;model_name&gt;, version 7 </EM></DIV><DIV><EM>Created version '7' of model &lt;model_name&gt;.</EM></DIV></DIV> Thu, 22 Aug 2024 18:05:42 GMT https://community.databricks.com/t5/machine-learning/modulenotfounderror-no-module-named-model-train-when-using/m-p/83965#M3600 yorabhir 2024-08-22T18:05:42Z