diff --git a/experimental/python/Makefile b/experimental/python/Makefile
index 5270be3103..be0fe664c0 100644
--- a/experimental/python/Makefile
+++ b/experimental/python/Makefile
@@ -17,7 +17,10 @@ lint:
 	uv run ruff format --diff
 
 codegen:
-	find databricks -name _models | xargs rm -rf
+	find databricks/bundles -type d -mindepth 1 -maxdepth 1 \
+	  ! -path databricks/bundles/core \
+	  ! -path databricks/bundles/resources \
+	  -exec rm -rf {} \;
 
 	cd codegen; uv run -m pytest codegen_tests
 	cd codegen; uv run -m codegen.main --output ..
diff --git a/experimental/python/databricks/bundles/compute/__init__.py b/experimental/python/databricks/bundles/compute/__init__.py
deleted file mode 100644
index e69de29bb2..0000000000