From 5c2f9ac883a9767f8ab0c5b4b3379047b39f2852 Mon Sep 17 00:00:00 2001
From: okozachenko1203 <okozachenko1203@users.noreply.github.com>
Date: Wed, 21 Aug 2024 01:16:14 +1000
Subject: [PATCH] fix: correct arg type of cluster object

---
 magnum_cluster_api/resources.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/magnum_cluster_api/resources.py b/magnum_cluster_api/resources.py
index df847727..1615bd38 100644
--- a/magnum_cluster_api/resources.py
+++ b/magnum_cluster_api/resources.py
@@ -2288,7 +2288,7 @@ def create_cluster_class(
 
 def mutate_machine_deployment(
     context: context.RequestContext,
-    cluster: objects.Cluster,
+    cluster: magnum_objects.Cluster,
     node_group: magnum_objects.NodeGroup,
     machine_deployment: dict = None,
 ):
@@ -2393,7 +2393,7 @@ def mutate_machine_deployment(
 
 
 def generate_machine_deployments_for_cluster(
-    context: context.RequestContext, cluster: objects.Cluster
+    context: context.RequestContext, cluster: magnum_objects.Cluster
 ) -> list:
     machine_deployments = []
     for ng in cluster.nodegroups: