Initial commit

skills/nix-packaging/python/darts/package.nix

@@ -0,0 +1,61 @@
+{ lib, buildPythonPackage, fetchFromGitHub, pythonRelaxDepsHook, setuptools
+, holidays, joblib, matplotlib, nfoursid, numpy, pandas, pmdarima, pyod
+, requests, scikit-learn, scipy, shap
+# , statsforecast
+, statsmodels, tbats, tqdm, typing-extensions, xarray, xgboost
+, pytorch-lightning }:
+
+buildPythonPackage rec {
+  pname = "darts";
+  version = "0.31.0";
+  format = "pyproject";
+
+  src = fetchFromGitHub {
+    owner = "unit8co";
+    repo = pname;
+    rev = version;
+    hash = "sha256-piSYRJIFr3RQTt/idfTRrqx/dD794He4d2F9flBJv7Q=";
+  };
+
+  nativeBuildInputs = [ pythonRelaxDepsHook ];
+
+  buildInputs = [ setuptools ];
+
+  propagatedBuildInputs = [
+    holidays
+    joblib
+    matplotlib
+    nfoursid
+    numpy
+    pandas
+    pmdarima
+    pyod
+    requests
+    scikit-learn
+    scipy
+    shap
+    # statsforecast
+    statsmodels
+    tbats
+    tqdm
+    typing-extensions
+    xarray
+    xgboost
+    pytorch-lightning
+  ];
+
+  pythonRelaxDeps = [ "pmdarima" ];
+
+  pythonRemoveDeps = [ "statsforecast" ];
+
+  pythonImportsCheck = [ "darts" ];
+
+  meta = with lib; {
+    description = ''
+      A python library for user-friendly forecasting and anomaly detection on time series
+    '';
+    homepage = "https://unit8co.github.io/darts/";
+    license = licenses.asl20;
+    maintainers = with maintainers; [ breakds ];
+  };
+}