From 40e19905c2e7e45cb8454b2061a906446346bb86 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Bar=C4=B1=C5=9F=20Can=20Durak?=
 <36421093+bcdurak@users.noreply.github.com>
Date: Tue, 24 Sep 2024 16:16:32 +0200
Subject: [PATCH] Bugfixing mlflow registry linting issue (#3043)

* linting mlflow registry

* dummy comment

* getattr and setattr
---
 src/zenml/integrations/mlflow/steps/mlflow_registry.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/zenml/integrations/mlflow/steps/mlflow_registry.py b/src/zenml/integrations/mlflow/steps/mlflow_registry.py
index ced4018368..093341f39a 100644
--- a/src/zenml/integrations/mlflow/steps/mlflow_registry.py
+++ b/src/zenml/integrations/mlflow/steps/mlflow_registry.py
@@ -146,8 +146,8 @@ def mlflow_register_model_step(
         metadata.zenml_pipeline_run_uuid = pipeline_run_uuid
     if metadata.zenml_workspace is None:
         metadata.zenml_workspace = zenml_workspace
-    if metadata.model_extra.get("mlflow_run_id", None) is None:
-        metadata.model_extra["mlflow_run_id"] = mlflow_run_id
+    if getattr(metadata, "mlflow_run_id", None) is None:
+        setattr(metadata, "mlflow_run_id", mlflow_run_id)
 
     # Register model version
     model_version = model_registry.register_model_version(