diff options
Diffstat (limited to 'pkgs/development/python-modules/mask-rcnn/default.nix')
-rw-r--r-- | pkgs/development/python-modules/mask-rcnn/default.nix | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/pkgs/development/python-modules/mask-rcnn/default.nix b/pkgs/development/python-modules/mask-rcnn/default.nix index 4c500a18515..ae07e7928a6 100644 --- a/pkgs/development/python-modules/mask-rcnn/default.nix +++ b/pkgs/development/python-modules/mask-rcnn/default.nix @@ -12,7 +12,7 @@ , pillow , scikitimage , scipy -, tensorflow_2 +, tensorflow }: buildPythonPackage rec { @@ -39,7 +39,7 @@ buildPythonPackage rec { pillow scikitimage scipy - tensorflow_2 # Keras only supports tensorflow 2 now + tensorflow ]; meta = with lib; { |