pkgs/by-name/rc/rclip/package.nix

   1 { lib
   2 , python3Packages
   3 , fetchFromGitHub
   4 }:
   5 python3Packages.buildPythonApplication rec {
   6   pname = "rclip";
   7   version = "1.10.3";
   8   pyproject = true;
   9
  10   src = fetchFromGitHub {
  11     owner = "yurijmikhalevich";
  12     repo = "rclip";
  13     rev = "refs/tags/v${version}";
  14     hash = "sha256-MdqO6X1dbIzmDuDSjrbmKbQfPIPcyhVIbodERYu9PZU=";
  15   };
  16
  17   nativeBuildInputs = with python3Packages; [
  18     poetry-core
  19   ];
  20
  21   propagatedBuildInputs = with python3Packages; [
  22     open-clip-torch
  23     pillow
  24     requests
  25     torch
  26     torchvision
  27     tqdm
  28   ];
  29
  30   nativeCheckInputs = with python3Packages; [ pytestCheckHook ];
  31
  32   pythonRelaxDeps = [ "torch" "torchvision" ];
  33
  34   pythonImportsCheck = [ "rclip" ];
  35
  36   disabledTestPaths = [
  37     # requires network
  38     "tests/e2e/test_rclip.py"
  39   ];
  40
  41   disabledTests = [
  42     # requires network
  43     "test_text_model_produces_the_same_vector_as_the_main_model"
  44     "test_loads_text_model_when_text_processing_only_requested_and_checkpoint_exists"
  45     "test_loads_full_model_when_text_processing_only_requested_and_checkpoint_doesnt_exist"
  46   ];
  47
  48   meta = with lib; {
  49     description = "AI-Powered Command-Line Photo Search Tool";
  50     homepage = "https://github.com/yurijmikhalevich/rclip";
  51     license = licenses.mit;
  52     maintainers = with maintainers; [ iynaix ];
  53     mainProgram = "rclip";
  54   };
  55 }