get hyperv_generation for shared gallery image

microsoft · Jun 17, 2024 · 0e37ed0 · 0e37ed0
1 parent 4f91209
commit 0e37ed0
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/lisa/sut_orchestrator/azure/platform_.py b/lisa/sut_orchestrator/azure/platform_.py
@@ -1308,6 +1308,9 @@ def _create_node_runbook(
             azure_node_runbook.community_gallery_image = None
             azure_node_runbook.shared_gallery.resolve_version(self)
             azure_node_runbook.update_raw()
+            azure_node_runbook.hyperv_generation = _get_gallery_image_generation(
+                azure_node_runbook.shared_gallery.query_platform(self)
+            )
         elif azure_node_runbook.community_gallery_image:
             azure_node_runbook.marketplace = None
             azure_node_runbook.community_gallery_image.resolve_version(self)