pkgs/development/python-modules/pandas-ta/default.nix


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53

{
  lib,
  buildPythonPackage,
  fetchurl,
  nix-update-script,
  numpy,
  pandas,
  python-dateutil,
  pytz,
  setuptools,
  six,
}:

buildPythonPackage rec {
  pname = "pandas-ta";
  version = "0.3.14b";
  pyproject = true;

  src = fetchurl {
    url = "https://www.pandas-ta.dev/assets/zip/pandas_ta-${version}.tar.gz";
    hash = "sha256-D6Na7IMdKBXqMLhxaIqNIKdrKIp74tJswAw1zYwJqZM=";
  };

  postPatch = ''
    substituteInPlace pandas_ta/momentum/squeeze_pro.py \
      --replace-fail "import NaN" "import nan"
  '';

  build-system = [ setuptools ];

  dependencies = [
    numpy
    pandas
    python-dateutil
    pytz
    setuptools
    six
  ];

  # PyTestCheckHook failing because of missing test dependency. Packages has been tested manually.
  doCheck = false;

  passthru.updateScript = nix-update-script { };

  pythonImportsCheck = [ "pandas_ta" ];

  meta = {
    description = "Technical Analysis Indicators";
    homepage = "https://www.pandas-ta.dev/";
    license = lib.licenses.unfree;
    maintainers = with lib.maintainers; [ derdennisop ];
  };
}