From 09737c95cf8ac7bf83ee826aa0313f00d9d7b40a Mon Sep 17 00:00:00 2001 From: Alexandra Belousov Date: Wed, 27 Nov 2024 13:02:50 +0200 Subject: [PATCH] include Image in runhouse init --- runhouse/__init__.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/runhouse/__init__.py b/runhouse/__init__.py index c4efd5fd31..01161b0bc1 100644 --- a/runhouse/__init__.py +++ b/runhouse/__init__.py @@ -13,6 +13,8 @@ OnDemandCluster, ) +from runhouse.resources.images import Image, ImageSetupStepType + # WARNING: Any built-in module that is imported here must be capitalized followed by all lowercase, or we will # will not find the module class when attempting to reconstruct it from a config. from runhouse.resources.module import Module, module