pkgs/development/python-modules/magika/default.nix

   1 { lib
   2 , buildPythonPackage
   3 , click
   4 , fetchPypi
   5 , magika
   6 , numpy
   7 , onnxruntime
   8 , poetry-core
   9 , python-dotenv
  10 , pythonOlder
  11 , stdenv
  12 , tabulate
  13 , testers
  14 , tqdm
  15 }:
  16
  17 buildPythonPackage rec {
  18   pname = "magika";
  19   version = "0.5.0";
  20   pyproject = true;
  21   disabled = pythonOlder "3.9";
  22
  23   src = fetchPypi {
  24     inherit pname version;
  25     hash = "sha256-r6C7iDCG/o3JEvweQGb4upr+LuHvmNtkwtduZGehCsc=";
  26   };
  27
  28   nativeBuildInputs = [
  29     poetry-core
  30   ];
  31
  32   propagatedBuildInputs = [
  33     click
  34     numpy
  35     onnxruntime
  36     python-dotenv
  37     tabulate
  38     tqdm
  39   ];
  40
  41   pythonImportsCheck = [ "magika" ];
  42
  43   passthru.tests.version = testers.testVersion { package = magika; };
  44
  45   meta = with lib; {
  46     description = "Magika: Detect file content types with deep learning";
  47     homepage = "https://github.com/google/magika";
  48     license = licenses.asl20;
  49     maintainers = with maintainers; [ mihaimaruseac ];
  50     mainProgram = "magika";
  51     # Currently, disabling on AArch64 as it onnx runtime crashes on ofborg
  52     broken = stdenv.isAarch64 && stdenv.isLinux;
  53   };
  54 }