From 79b2b3f4ebcb43c21f1dcca57b6ef0427efcfa75 Mon Sep 17 00:00:00 2001 From: Claude Date: Mon, 15 Sep 2025 11:04:45 -0400 Subject: [PATCH] Add dill dep where tensorflow-transform is used. --- sdks/python/setup.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/sdks/python/setup.py b/sdks/python/setup.py index 4d7ba0d5a506..102eb3ac2d17 100644 --- a/sdks/python/setup.py +++ b/sdks/python/setup.py @@ -528,6 +528,9 @@ def get_portability_package_data(): 'pyod', 'tensorflow', 'tensorflow-hub', + # tensorflow-transform requires dill, but doesn't set dill as a + # hard requirement in setup.py. + 'dill', 'tensorflow-transform', 'tf2onnx', 'torch', @@ -588,7 +591,11 @@ def get_portability_package_data(): 'tensorflow>=2.12.0', 'torch>=1.9.0' ], - 'tft': ['tensorflow_transform>=1.14.0,<1.15.0'], + 'tft': [ + 'tensorflow_transform>=1.14.0,<1.15.0' + # tensorflow-transform requires dill, but doesn't set dill as a + # hard requirement in setup.py. + , 'dill'], 'onnx': [ 'onnxruntime==1.13.1', 'torch==1.13.1',