Merge branch 'develop' into check_log_regex

Azure · Sep 4, 2024 · 44f2d27 · 44f2d27
2 parents bffd26b + b43e143
commit 44f2d27
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests_e2e/orchestrator/lib/agent_test_suite_combinator.py b/tests_e2e/orchestrator/lib/agent_test_suite_combinator.py
@@ -451,7 +451,7 @@ def _get_runbook_images(self, loader: AgentTestLoader) -> List[VmImageInfo]:
             return images
 
         # If it is not image or image set, it must be a URN, VHD, or an image from a gallery
-        if not self._is_urn(self.runbook.image) and not self._is_vhd(self.runbook.image) and not self._is_image_from_gallery(self.runbook.image):
+        if not self._is_urn(self.runbook.image) and not self._is_vhd(self.runbook.image) and not CustomImage._is_image_from_gallery(self.runbook.image):
             raise Exception(f"The 'image' parameter must be an image, image set name, urn, vhd, or an image from a shared gallery: {self.runbook.image}")
 
         i = VmImageInfo()