Something went wrong. Try again.
lol
Something went wrong. Try again.
2.7 kB · 134 lines
Nix
at master
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135{ lib, stdenv, buildPythonPackage, fetchFromGitHub, fetchpatch, pythonAtLeast,
# build-system setuptools,
# dependencies numpy, pandas, pydantic, tqdm, toolz,
# optional dependencies (torch) torch, lightning, scipy,
# tests pytestCheckHook, distutils, matplotlib, pyarrow, statsmodels, writableTmpDirAsHomeHook, which,}:
buildPythonPackage (finalAttrs: { pname = "gluonts"; version = "0.16.2"; pyproject = true;
src = fetchFromGitHub { owner = "awslabs"; repo = "gluonts"; tag = "v${finalAttrs.version}"; hash = "sha256-h0+RYgGMz0gPchiKGIu0/NGcWBky5AWNTJKzoupn/iQ="; };
# pydantic.v1.errors.ConfigError: unable to infer type for attribute "target" disabled = pythonAtLeast "3.14";
patches = [ # Fixes _pickle.UnpicklingError: Weights only load failed. # https://github.com/awslabs/gluonts/pull/3269 (fetchpatch { name = "fix-torch-load_from_checkpoint"; url = "https://github.com/awslabs/gluonts/pull/3269/commits/6420e75cfbeabcd94e2ff09dfed3b2eeb4881710.patch"; hash = "sha256-UeLjgKra+Y3uPoTBle+YCxD0a1ahu6d5anrMHn4HH2I="; }) ];
build-system = [ setuptools ];
dependencies = [ numpy pandas pydantic tqdm toolz ];
optional-dependencies = { torch = [ torch lightning scipy ]; };
pythonRelaxDeps = [ "numpy" "toolz" ];
pythonImportsCheck = [ "gluonts" "gluonts.core" "gluonts.dataset" "gluonts.ev" "gluonts.evaluation" "gluonts.ext" "gluonts.model" "gluonts.shell" "gluonts.time_feature" "gluonts.torch" "gluonts.transform" ];
nativeCheckInputs = [ pytestCheckHook distutils matplotlib pyarrow statsmodels writableTmpDirAsHomeHook which ] ++ finalAttrs.passthru.optional-dependencies.torch;
disabledTestPaths = [ # requires `cpflows`, not in Nixpkgs "test/torch/model" ] ++ lib.optionals stdenv.hostPlatform.isDarwin [ # Trace/BPT trap: 5 "test/torch/test_torch_item_id_info.py" ];
disabledTests = [ # tries to access network "test_against_former_evaluator" ] ++ lib.optionals stdenv.hostPlatform.isDarwin [ # RuntimeError: *** -[__NSPlaceholderArray initWithObjects:count:]: attempt to insert nil object from objects[1] "test_forecast" ];
meta = { description = "Probabilistic time series modeling in Python"; homepage = "https://ts.gluon.ai"; changelog = "https://github.com/awslabs/gluonts/releases/tag/${finalAttrs.src.tag}"; license = lib.licenses.asl20; maintainers = with lib.maintainers; [ bcdarwin ]; };})