Unverified Commit f5c6c9fa authored by Nick Cao's avatar Nick Cao Committed by GitHub
Browse files

python3Packages.torchcrepe: cleanup, add missing torchcodec dependency (#458936)

parents b49eb81e 310cc0bc
Loading
Loading
Loading
Loading
+12 −6
Original line number Diff line number Diff line
@@ -2,24 +2,27 @@
  lib,
  buildPythonPackage,
  fetchFromGitHub,

  # build-system
  setuptools,

  # dependencies
  librosa,
  pytestCheckHook,
  pythonOlder,
  resampy,
  scipy,
  setuptools,
  torch,
  torchaudio,
  torchcodec,
  tqdm,

  # tests
  pytestCheckHook,
}:

buildPythonPackage {
  pname = "torchcrepe";
  version = "0.0.24";
  pyproject = true;
  build-system = [ setuptools ];

  disabled = pythonOlder "3.7";

  src = fetchFromGitHub {
    owner = "maxrmorrison";
@@ -29,12 +32,15 @@ buildPythonPackage {
    hash = "sha256-w2T8D3ATCHVBCBhMdLSYdV0yb9vUYwZLz+B2X2gteEU=";
  };

  build-system = [ setuptools ];

  dependencies = [
    librosa
    resampy
    scipy
    torch
    torchaudio
    torchcodec
    tqdm
  ];