Update TFX python dependency constraint to allow TensorFlow 2.5.0

Please update TFX python dependency constraint to allow TensorFlow 2.5.0. As of right now only TensorFlow 2.4.1 and below are supported but TensorFlow 2.4.1 has many critical/high vulnerabilities which have been patched in TensorFlow 2.5.0

[jovyan@2b08f11f4755 venv]$ poetry add tfx --python ">=3.8,<3.9"
Using version ^0.30.0 for tfx

Updating dependencies
Resolving dependencies... (0.1s)

  SolverProblemError

  Because tfx (0.30.0) depends on tensorflow (>=1.15.2,<2.0.0 || >=2.4.0,<2.5.0 || >=2.6.0,<3)
   and no versions of tfx match >0.30.0,<0.31.0, tfx (>=0.30.0,<0.31.0) requires tensorflow (>=1.15.2,<2.0.0 || >=2.4.0,<2.5.0 || >=2.6.0,<3).
  So, because venv depends on both tensorflow (2.5.0) and tfx (^0.30.0), version solving failed.

  at /opt/jupyter/venv/lib/python3.8/site-packages/poetry/puzzle/solver.py:241 in _solve
      237│             packages = result.packages
      238│         except OverrideNeeded as e:
      239│             return self.solve_in_compatibility_mode(e.overrides, use_latest=use_latest)
      240│         except SolveFailure as e:
    → 241│             raise SolverProblemError(e)
      242│ 
      243│         results = dict(
      244│             depth_first_search(
      245│                 PackageNode(self._package, packages), aggregate_package_nodes

[dhruvesh09]

Can you try to pip install -i https://pypi-nightly.tensorflow.org/simple/ tfx==20210519

This worked for me

[google-ml-butler[bot]]

Are you satisfied with the resolution of your issue? Yes No