@@ -3,10 +3,10 @@
33# Python deps
44, buildPythonPackage, pythonOlder, python
55# Python libraries
6−, numpy, tensorflow-tensorboard_2, absl-py
6+, numpy, tensorflow-tensorboard, absl-py
77, setuptools, wheel, Keras, keras-preprocessing, google-pasta
88, opt-einsum, astunparse, h5py
9−, termcolor, grpcio, six, wrapt, protobuf, tensorflow-estimator_2
9+, termcolor, grpcio, six, wrapt, protobuf, tensorflow-estimator
1010, dill, flatbuffers-python, tblib, typing-extensions
1111# Common deps
1212, git, pybind11, which, binutils, glibcLocales, cython, perl
@@ -94,8 +94,8 @@ let
9494 setuptools
9595 six
9696 tblib
97− tensorflow-estimator_2
98− tensorflow-tensorboard_2
97+ tensorflow-estimator
98+ tensorflow-tensorboard
9999 termcolor
100100 typing-extensions
101101 wheel
@@ -457,12 +457,12 @@ in buildPythonPackage {
457457 protobuf
458458 six
459459 tblib
460− tensorflow-estimator_2
460+ tensorflow-estimator
461461 termcolor
462462 typing-extensions
463463 wrapt
464464 ] ++ lib.optionals withTensorboard [
465− tensorflow-tensorboard_2
465+ tensorflow-tensorboard
466466 ];
467467
468468 nativeBuildInputs = lib.optional cudaSupport addOpenGLRunpath;