Merge pull request #263552 from GaetanLepage/compressai (e83579a1) · Commits · nix / nixpkgs

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

0 → 100644

+89 −0

Original line number	Diff line number	Diff line
		{ lib
		, buildPythonPackage
		, pythonOlder
		, fetchFromGitHub
		, pybind11
		, setuptools
		, wheel
		, numpy
		, matplotlib
		, pytorch-msssim
		, scipy
		, torch
		, torchvision
		, ipywidgets
		, jupyter
		, plotly
		, pytestCheckHook
		}:

		buildPythonPackage rec {
		pname = "compressai";
		version = "1.2.4";
		pyproject = true;

		disabled = pythonOlder "3.6";

		src = fetchFromGitHub {
		owner = "InterDigitalInc";
		repo = "CompressAI";
		rev = "refs/tags/v${version}";
		hash = "sha256-nT2vd7t67agIWobJalORbRuns0UJGRGGbTX2/8vbTiY=";
		fetchSubmodules = true;
		};

		nativeBuildInputs = [
		pybind11
		setuptools
		wheel
		];

		propagatedBuildInputs = [
		numpy
		matplotlib
		pytorch-msssim
		scipy
		torch
		torchvision
		];

		passthru.optional-dependencies = {
		tutorials = [
		ipywidgets
		jupyter
		];
		};

		pythonImportsCheck = [
		"compressai"
		"compressai._CXX"
		];

		preCheck = ''
		# We have to delete the source because otherwise it is used intead the installed package.
		rm -rf compressai

		export HOME=$(mktemp -d)
		'';

		nativeCheckInputs = [
		plotly
		pytestCheckHook
		];

		disabledTests = [
		# Those tests require internet access to download some weights
		"test_image_codec"
		"test_update"
		"test_eval_model_pretrained"
		"test_cheng2020_anchor"
		"test_pretrained"
		];

		meta = with lib; {
		description = "A PyTorch library and evaluation platform for end-to-end compression research";
		homepage = "https://github.com/InterDigitalInc/CompressAI";
		license = licenses.bsd3Clear;
		maintainers = with maintainers; [ GaetanLepage ];
		};
		}

0 → 100644

+41 −0

Original line number	Diff line number	Diff line
		{ lib
		, buildPythonPackage
		, fetchFromGitHub
		, setuptools
		, wheel
		, torch
		}:

		buildPythonPackage rec {
		pname = "pytorch-msssim";
		version = "1.0.0";
		pyproject = true;

		src = fetchFromGitHub {
		owner = "VainF";
		repo = "pytorch-msssim";
		rev = "refs/tags/v${version}";
		hash = "sha256-bghglwQhgByC7BqbDvImSvt6edKF55NLYEPjqmmSFH8=";
		};

		nativeBuildInputs = [
		setuptools
		wheel
		];

		propagatedBuildInputs = [
		torch
		];

		pythonImportsCheck = [ "pytorch_msssim" ];

		# This test doesn't have (automatic) tests
		doCheck = false;

		meta = with lib; {
		description = "Fast and differentiable MS-SSIM and SSIM for pytorch";
		homepage = "https://github.com/VainF/pytorch-msssim";
		license = licenses.mit;
		maintainers = with maintainers; [ GaetanLepage ];
		};
		}

+4 −0

Original line number	Diff line number	Diff line
		@@ -2242,6 +2242,8 @@ self: super: with self; {

		compreffor = callPackage ../development/python-modules/compreffor { };

		compressai = callPackage ../development/python-modules/compressai { };

		concurrent-log-handler = callPackage ../development/python-modules/concurrent-log-handler { };

		conda = callPackage ../development/python-modules/conda { };
		@@ -11642,6 +11644,8 @@ self: super: with self; {

		pytorch-metric-learning = callPackage ../development/python-modules/pytorch-metric-learning { };

		pytorch-msssim = callPackage ../development/python-modules/pytorch-msssim { };

		pytorch-pfn-extras = callPackage ../development/python-modules/pytorch-pfn-extras { };

		pytraccar = callPackage ../development/python-modules/pytraccar { };