Skip to content
This repository has been archived by the owner on Oct 7, 2024. It is now read-only.

Commit

Permalink
Pin TensorFlow version to 2.1.
Browse files Browse the repository at this point in the history 
TensorFlow 2.2 introduces a dependency on gast 0.3 which is incompatible with tensorflow_probability 0.8, which we need pinned for trfl.

PiperOrigin-RevId: 307572875
Change-Id: I4ed8e951651165a04a5144ae5574151c1f4bd313
  • Loading branch information
@aslanides @copybara-github
aslanides authored and copybara-github committed Apr 21, 2020
1 parent 5edfe03 commit 6c12227
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion setup.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,7 @@
baselines_require = [
'dm-sonnet',
'dm-tree',
'tensorflow',
'tensorflow == 2.1',
'tensorflow_probability >= 0.8, < 0.9',
'trfl @ git+git://github.com/deepmind/trfl.git#egg=trfl',
'tqdm',
Expand Down

0 comments on commit 6c12227

Please sign in to comment.