nur-packages

My NUR packages
git clone git://git.sikmir.ru/nur-packages
Log | Files | Refs | README | LICENSE

package.nix (1009B)

      1 {
      2   lib,
      3   fetchFromGitHub,
      4   python3Packages,
      5   writableTmpDirAsHomeHook,
      6 }:
      7 
      8 python3Packages.buildPythonPackage (finalAttrs: {
      9   pname = "segmentation-models-pytorch";
     10   version = "0.5.0";
     11   pyproject = true;
     12 
     13   __structuredAttrs = true;
     14 
     15   src = fetchFromGitHub {
     16     owner = "qubvel-org";
     17     repo = "segmentation_models.pytorch";
     18     tag = "v${finalAttrs.version}";
     19     hash = "sha256-QtrmMbVcFHftV69stJHk0+3n1o0inlO22xHs/smLlGg=";
     20   };
     21 
     22   build-system = with python3Packages; [ setuptools ];
     23 
     24   dependencies = with python3Packages; [
     25     numpy
     26     pillow
     27     safetensors
     28     timm
     29     torch
     30     torchvision
     31     tqdm
     32   ];
     33 
     34   nativeCheckInputs = with python3Packages; [
     35     pytestCheckHook
     36     gitpython
     37     writableTmpDirAsHomeHook
     38   ];
     39 
     40   doCheck = false;
     41 
     42   meta = {
     43     description = "Image segmentation models with pre-trained backbones";
     44     homepage = "https://github.com/qubvel-org/segmentation_models.pytorch";
     45     license = lib.licenses.mit;
     46     maintainers = [ lib.maintainers.sikmir ];
     47   };
     48 })