python310Packages.invisible-watermark: pytorch -> torch

2022-12-04 21:49:16 -08:00
parent bc85ef8158
commit 7833b95d8e
1 changed files with 3 additions and 2 deletions
@@ -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