From 0ed03c56122398dcf1b2616bdf87772526f99ac1 Mon Sep 17 00:00:00 2001 From: Sahil Batra Date: Tue, 16 Jun 2026 18:24:41 -0400 Subject: [PATCH] Add dataflow and feature_store packages to wheel Dataflow workers install from the built wheel, which only included ml_pipelines_kfp. The dataflow package was missing __init__.py files and was not listed in pyproject.toml, causing ModuleNotFoundError on workers for imports like dataflow.models.iris_schema. Co-Authored-By: Claude Opus 4.6 --- pyproject.toml | 2 +- src/dataflow/__init__.py | 0 src/dataflow/models/__init__.py | 0 src/dataflow/utils/__init__.py | 0 4 files changed, 1 insertion(+), 1 deletion(-) create mode 100644 src/dataflow/__init__.py create mode 100644 src/dataflow/models/__init__.py create mode 100644 src/dataflow/utils/__init__.py diff --git a/pyproject.toml b/pyproject.toml index 7aa6d79..d305432 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -65,4 +65,4 @@ requires = ["hatchling"] build-backend = "hatchling.build" [tool.hatch.build.targets.wheel] -packages = ["src/ml_pipelines_kfp"] \ No newline at end of file +packages = ["src/ml_pipelines_kfp", "src/dataflow", "src/feature_store"] \ No newline at end of file diff --git a/src/dataflow/__init__.py b/src/dataflow/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/src/dataflow/models/__init__.py b/src/dataflow/models/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/src/dataflow/utils/__init__.py b/src/dataflow/utils/__init__.py new file mode 100644 index 0000000..e69de29