about summary refs log tree commit diff
path: root/pkgs/development
diff options
context:
space:
mode:
authorFabian Affolter <fabian@affolter-engineering.ch>2022-12-05 15:32:20 +0100
committerGitHub <noreply@github.com>2022-12-05 15:32:20 +0100
commitab417860ceba9ece9e5b04e4e0b036c35869e430 (patch)
tree4b82b6100c6f41ad8ea0bbdb3475a08df1fcc755 /pkgs/development
parent0bda862edc8b106779faafa4b5467f5fb85f3362 (diff)
parent7833b95d8e30660533fe1ef0ec62715f18fcc65a (diff)
Merge pull request #204594 from tjni/invisible-watermark
python310Packages.invisible-watermark: pytorch -> torch
Diffstat (limited to 'pkgs/development')
-rw-r--r--pkgs/development/python-modules/invisible-watermark/default.nix5
1 files changed, 3 insertions, 2 deletions
diff --git a/pkgs/development/python-modules/invisible-watermark/default.nix b/pkgs/development/python-modules/invisible-watermark/default.nix
index 82357741bab50..ce77b9e3115c2 100644
--- a/pkgs/development/python-modules/invisible-watermark/default.nix
+++ b/pkgs/development/python-modules/invisible-watermark/default.nix
@@ -3,7 +3,7 @@
 , pythonOlder
 , fetchFromGitHub
 , opencv4
-, pytorch
+, torch
 , onnx
 , onnxruntime
 , pillow
@@ -14,6 +14,7 @@
 buildPythonPackage rec {
   pname = "invisible-watermark";
   version = "0.1.5";
+  format = "setuptools";
   disabled = pythonOlder "3.6";
 
   src = fetchFromGitHub {
@@ -25,7 +26,7 @@ buildPythonPackage rec {
 
   propagatedBuildInputs = [
     opencv4
-    pytorch
+    torch
     onnx
     onnxruntime
     pillow