Skip to content

Commit fdec6a3

Browse files
authored
Add dill dep where tensorflow-transform is used. (#36158)
1 parent 0c433e3 commit fdec6a3

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

sdks/python/setup.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -528,6 +528,9 @@ def get_portability_package_data():
528528
'pyod',
529529
'tensorflow',
530530
'tensorflow-hub',
531+
# tensorflow-transform requires dill, but doesn't set dill as a
532+
# hard requirement in setup.py.
533+
'dill',
531534
'tensorflow-transform',
532535
'tf2onnx',
533536
'torch',
@@ -588,7 +591,11 @@ def get_portability_package_data():
588591
'tensorflow>=2.12.0',
589592
'torch>=1.9.0'
590593
],
591-
'tft': ['tensorflow_transform>=1.14.0,<1.15.0'],
594+
'tft': [
595+
'tensorflow_transform>=1.14.0,<1.15.0'
596+
# tensorflow-transform requires dill, but doesn't set dill as a
597+
# hard requirement in setup.py.
598+
, 'dill'],
592599
'onnx': [
593600
'onnxruntime==1.13.1',
594601
'torch==1.13.1',

0 commit comments

Comments
 (0)