diff --git a/pkgs/development/python-modules/pyldavis/default.nix b/pkgs/development/python-modules/pyldavis/default.nix new file mode 100644 index 000000000000..8b26af56615c --- /dev/null +++ b/pkgs/development/python-modules/pyldavis/default.nix @@ -0,0 +1,51 @@ +{ + lib, + buildPythonPackage, + fetchFromGitHub, + funcy, + ipython, + jinja2, + joblib, + numpy, + pandas, + scikit-learn, + scipy, +}: + +buildPythonPackage rec { + pname = "pyLDAvis"; + version = "3.4.1"; + format = "setuptools"; + + src = fetchFromGitHub { + owner = "bmabey"; + repo = "pyLDAvis"; + rev = version; + sha256 = "sha256-WIQytds3PeU85l6ix2UUIwypjpM5rMZvQxiHx9BY91Y="; + }; + + propagatedBuildInputs = [ + funcy + jinja2 + joblib + ipython + numpy + pandas + scikit-learn + scipy + ]; + + pythonImportsCheck = [ + "pyLDAvis" + "pyLDAvis.gensim_models" + ]; + + meta = with lib; { + homepage = "https://github.com/bmabey/pyLDAvis"; + description = "Python library for interactive topic model visualization"; + maintainers = with lib.maintainers; [ gm6k ]; + license = licenses.bsd3; + sourceProvenance = with sourceTypes; [ fromSource ]; + platforms = platforms.all; + }; +} diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix index 6dc85ed24bb2..4c1bba82bc10 100644 --- a/pkgs/top-level/python-packages.nix +++ b/pkgs/top-level/python-packages.nix @@ -10419,6 +10419,8 @@ self: super: with self; { inherit (pkgs) protobuf; }; + pyldavis = callPackage ../development/python-modules/pyldavis { }; + pylddwrap = callPackage ../development/python-modules/pylddwrap { }; pyloadapi = callPackage ../development/python-modules/pyloadapi { };