@@ -170,7 +170,7 @@ def cythonize(*args, **kwargs):
170170 'skl2onnx' ,
171171 'pyod>=0.7.6' , # 0.7.5 crashes setuptools
172172 'tensorflow' ,
173- # tensorflow transient dep, lower versions not compatible with Python3.10+
173+ # tensorflow transitive dep, lower versions not compatible with Python3.10+
174174 'absl-py>=0.12.0' ,
175175 'tensorflow-hub' ,
176176 'tf2onnx' ,
@@ -531,6 +531,15 @@ def get_portability_package_data():
531531 # urllib 2.x is a breaking change for the headless chrome tests
532532 'urllib3<2,>=1.21.1'
533533 ],
534+ 'tft' : [
535+ 'tensorflow_transform>=1.14.0,<1.15.0' ,
536+ # TFT->TFX-BSL require pandas 1.x, which is not compatible
537+ # with numpy 2.x
538+ 'numpy<2' ,
539+ # tensorflow-transform requires dill, but doesn't set dill as a
540+ # hard requirement in setup.py.
541+ 'dill'
542+ ],
534543 # Optional dependencies to unit-test ML functionality.
535544 # We don't expect users to install this extra. Users should install
536545 # necessary dependencies individually, or we should create targeted
@@ -605,15 +614,6 @@ def get_portability_package_data():
605614 'absl-py>=0.12.0'
606615 ],
607616 'redis' : ['redis>=5.0.0,<6' ],
608- 'tft' : [
609- 'tensorflow_transform>=1.14.0,<1.15.0' ,
610- # TFT->TFX-BSL require pandas 1.x, which is not compatible
611- # with numpy 2.x
612- 'numpy<2' ,
613- # tensorflow-transform requires dill, but doesn't set dill as a
614- # hard requirement in setup.py.
615- 'dill'
616- ],
617617 'tfrecord' : ['crcmod>=1.7,<2.0' ],
618618 'onnx' : [
619619 'onnxruntime==1.13.1' ,
0 commit comments