include Image in runhouse init

run-house · Nov 27, 2024 · afefbf0 · afefbf0
1 parent 7312a89
commit afefbf0
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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