polars-talib 0.2.0__tar.gz

polars_talib-0.2.0/.agents/plugins/marketplace.json

@@ -0,0 +1,20 @@
+{
+  "name": "polars-ta-extension",
+  "interface": {
+    "displayName": "Polars Ta Extension"
+  },
+  "plugins": [
+    {
+      "name": "polars-talib",
+      "source": {
+        "source": "local",
+        "path": "./plugins/polars-talib"
+      },
+      "policy": {
+        "installation": "AVAILABLE",
+        "authentication": "ON_INSTALL"
+      },
+      "category": "Productivity"
+    }
+  ]
+}

polars_talib-0.2.0/.claude-plugin/marketplace.json

@@ -0,0 +1,16 @@
+{
+  "name": "polars-ta-extension",
+  "owner": {
+    "name": "Yvictor",
+    "url": "https://github.com/Yvictor"
+  },
+  "description": "Polars TA-Lib coding skills.",
+  "plugins": [
+    {
+      "name": "polars-talib",
+      "source": "./plugins/polars-talib",
+      "description": "Use all 201 TA-Lib indicators in native Polars expressions, grouped time series and lazy pipelines.",
+      "version": "0.2.0"
+    }
+  ]
+}

polars_talib-0.2.0/.cursor-plugin/marketplace.json

@@ -0,0 +1,16 @@
+{
+  "name": "polars-ta-extension",
+  "owner": {
+    "name": "Yvictor"
+  },
+  "metadata": {
+    "description": "Polars TA-Lib coding skills."
+  },
+  "plugins": [
+    {
+      "name": "polars-talib",
+      "source": "plugins/polars-talib",
+      "description": "Use all 201 TA-Lib indicators in native Polars expressions, grouped time series and lazy pipelines."
+    }
+  ]
+}

polars_talib-0.2.0/.gitattributes

@@ -0,0 +1 @@
+examples/* linguist-documentation

polars_talib-0.2.0/.github/workflows/CI.yml

@@ -0,0 +1,219 @@
+name: CI
+
+on:
+  push:
+    branches: [master, main]
+    tags: ['v*', '0.*']
+  pull_request:
+  workflow_dispatch:
+
+permissions:
+  contents: read
+
+concurrency:
+  group: ci-${{ github.ref }}
+  cancel-in-progress: ${{ !startsWith(github.ref, 'refs/tags/') }}
+
+env:
+  PYO3_USE_ABI3_FORWARD_COMPATIBILITY: '1'
+  CARGO_BUILD_JOBS: '4'
+  POLARS_MAX_THREADS: '4'
+
+jobs:
+  wheels:
+    name: Wheel / ${{ matrix.name }}
+    runs-on: ${{ matrix.runner }}
+    strategy:
+      fail-fast: false
+      matrix:
+        include:
+          - {name: linux-x64, runner: ubuntu-24.04, target: x86_64-unknown-linux-gnu, manylinux: '2_28', python-arch: x64}
+          - {name: linux-arm64, runner: ubuntu-24.04-arm, target: aarch64-unknown-linux-gnu, manylinux: '2_28', python-arch: arm64}
+          - {name: musl-x64, runner: ubuntu-24.04, target: x86_64-unknown-linux-musl, manylinux: musllinux_1_2, python-arch: x64}
+          - {name: musl-arm64, runner: ubuntu-24.04-arm, target: aarch64-unknown-linux-musl, manylinux: musllinux_1_2, python-arch: arm64}
+          - {name: macos-x64, runner: macos-15-intel, target: x86_64-apple-darwin, manylinux: 'off', python-arch: x64}
+          - {name: macos-arm64, runner: macos-15, target: aarch64-apple-darwin, manylinux: 'off', python-arch: arm64}
+          - {name: windows-x64, runner: windows-2025, target: x86_64-pc-windows-msvc, manylinux: 'off', python-arch: x64}
+          - {name: windows-arm64, runner: windows-11-arm, target: aarch64-pc-windows-msvc, manylinux: 'off', python-arch: arm64}
+    steps:
+      - uses: actions/checkout@v6
+      - uses: actions/setup-python@v6
+        with:
+          python-version: '3.12'
+          architecture: ${{ matrix.python-arch }}
+      - uses: dtolnay/rust-toolchain@1.90.0
+        with:
+          components: rustfmt
+      - uses: astral-sh/setup-uv@v8.2.0
+      - name: CMake (native hosts)
+        if: runner.os != 'Linux'
+        run: uv pip install --system 'cmake>=4.2,<5'
+      - name: Build self-contained release wheel
+        uses: PyO3/maturin-action@v1.46.0
+        with:
+          target: ${{ matrix.target }}
+          manylinux: ${{ matrix.manylinux }}
+          args: --release --locked --out dist
+          sccache: 'true'
+          rust-toolchain: '1.90.0'
+          before-script-linux: |
+            if test -x /opt/python/cp312-cp312/bin/python; then
+              export PATH=/opt/python/cp312-cp312/bin:$PATH
+            fi
+            if command -v apk >/dev/null; then
+              apk add --no-cache cmake make
+            else
+              python3 -m pip install 'cmake>=4.2,<5'
+            fi
+      - name: Test installed wheel (native)
+        if: ${{ !startsWith(matrix.name, 'musl-') }}
+        shell: bash
+        run: |
+          uv venv .wheel-test
+          uv pip install --python .wheel-test dist/*.whl 'polars==1.44.2' 'TA-Lib==0.8.1' numpy pytest pytest-benchmark
+          uv run --no-project --python .wheel-test python -m pytest tests --ignore=tests/test_bench.py -q
+          uv run --no-project --python .wheel-test python scripts/install_skill.py --agent codex --project .skill-test
+      - name: Test installed wheel (Alpine, native CPU)
+        if: startsWith(matrix.name, 'musl-')
+        run: |
+          docker run --rm -v "$PWD:/work" -w /work python:3.12-alpine sh -ec '
+            pip install --only-binary=:all: dist/*.whl "polars==1.44.2" "TA-Lib==0.8.1" numpy pytest pytest-benchmark
+            python -m pytest tests --ignore=tests/test_bench.py -q
+          '
+      - uses: actions/upload-artifact@v7
+        with:
+          name: wheels-${{ matrix.name }}
+          path: dist/*.whl
+          if-no-files-found: error
+
+  native-plugins:
+    name: Native Claude Code and Codex plugins
+    runs-on: ubuntu-24.04
+    steps:
+      - uses: actions/checkout@v6
+      - uses: actions/setup-node@v6
+        with:
+          node-version: '24'
+      - uses: actions/setup-python@v6
+        with:
+          python-version: '3.12'
+      - name: Install tested native CLIs
+        run: npm install --global @openai/codex@0.155.1 @anthropic-ai/claude-code@2.1.278
+      - name: Install and discover plugin skills with native CLIs
+        run: python scripts/check_native_plugins.py
+
+  compatibility:
+    name: Python ${{ matrix.python }} / Polars ${{ matrix.polars }}
+    needs: wheels
+    runs-on: ubuntu-24.04
+    strategy:
+      fail-fast: false
+      matrix:
+        python: ['3.10', '3.11', '3.13', '3.14']
+        polars: ['1.20.0', '1.44.2']
+    steps:
+      - uses: actions/checkout@v6
+      - uses: actions/setup-python@v6
+        with:
+          python-version: ${{ matrix.python }}
+      - uses: actions/download-artifact@v7
+        with:
+          name: wheels-linux-x64
+          path: dist
+      - run: python -m pip install dist/*.whl 'polars==${{ matrix.polars }}' 'TA-Lib==0.8.1' numpy pytest pytest-benchmark
+      - run: python -m pytest tests --ignore=tests/test_bench.py -q
+
+  source:
+    name: Source distribution and generators
+    runs-on: ubuntu-24.04
+    steps:
+      - uses: actions/checkout@v6
+      - uses: dtolnay/rust-toolchain@1.90.0
+        with:
+          components: rustfmt
+      - uses: actions/setup-python@v6
+        with:
+          python-version: '3.12'
+      - uses: astral-sh/setup-uv@v8.2.0
+      - run: uv pip install --system maturin 'cmake>=4.2,<5' pytest numpy 'TA-Lib==0.8.1' 'polars==1.44.2' pytest-benchmark
+      - name: Check Python dependency lock
+        run: uv lock --check
+      - name: Check generated wrappers are current
+        run: |
+          python scripts/generate_indicators.py
+          python scripts/generate_skill_reference.py
+          git diff --exit-code -- src/generated.rs talib/src/generated.rs python/polars_talib/_generated.py python/polars_talib/_param_types.py docs/indicators.md plugins/polars-talib/skills/polars-talib/reference/functions.md
+      - name: Rust crate tests
+        run: cargo test --manifest-path talib/Cargo.toml
+      - name: Validate system-library overrides
+        run: python scripts/check_build_overrides.py
+      - name: Build source distribution and install it without system TA-Lib
+        run: |
+          maturin sdist --out dist
+          uv pip install --system dist/*.tar.gz
+          python -m pytest tests --ignore=tests/test_bench.py -q
+      - uses: actions/upload-artifact@v7
+        with:
+          name: sdist
+          path: dist/*.tar.gz
+
+  benchmark:
+    name: Release benchmark
+    needs: wheels
+    runs-on: ubuntu-24.04
+    steps:
+      - uses: actions/checkout@v6
+      - uses: actions/setup-python@v6
+        with:
+          python-version: '3.12'
+      - uses: actions/download-artifact@v7
+        with:
+          name: wheels-linux-x64
+          path: dist
+      - uses: astral-sh/setup-uv@v8.2.0
+      # Informational: shared runners are too noisy to gate a release on a
+      # fixed regression threshold. The comparison and raw samples are kept as
+      # a job summary and artifact for humans to judge.
+      - name: Compare release wheels on the same runner
+        run: |
+          uv venv .benchmark-base
+          uv venv .benchmark-candidate
+          uv pip install --python .benchmark-base 'polars-talib==0.1.6' 'polars==1.44.2' 'numpy==2.5.3'
+          uv pip install --python .benchmark-candidate dist/*.whl 'polars==1.44.2' 'numpy==2.5.3'
+          .benchmark-base/bin/python scripts/benchmark.py --output baseline.json --repeats 51 --pin-cpus 4
+          .benchmark-candidate/bin/python scripts/benchmark.py --output candidate.json --repeats 51 --pin-cpus 4
+          python scripts/compare_benchmarks.py baseline.json candidate.json --output comparison.md --report-only
+          cat comparison.md >> "$GITHUB_STEP_SUMMARY"
+      - uses: actions/upload-artifact@v7
+        if: always()
+        with:
+          name: benchmark
+          path: |
+            baseline.json
+            candidate.json
+            comparison.md
+
+  release:
+    name: Publish tested wheels
+    if: startsWith(github.ref, 'refs/tags/')
+    # native-plugins validates repository files with third-party CLIs fetched
+    # from npm at run time; it is a required PR check but must not be able to
+    # block wheel publication when npm or those CLI versions are unavailable.
+    needs: [wheels, compatibility, source]
+    runs-on: ubuntu-24.04
+    steps:
+      - uses: actions/download-artifact@v7
+        with:
+          pattern: wheels-*
+          merge-multiple: true
+          path: dist
+      - uses: actions/download-artifact@v7
+        with:
+          name: sdist
+          path: dist
+      - uses: PyO3/maturin-action@v1.46.0
+        env:
+          MATURIN_PYPI_TOKEN: ${{ secrets.PYPI_API_TOKEN }}
+        with:
+          command: upload
+          args: --non-interactive dist/*

polars_talib-0.2.0/.gitignore

@@ -0,0 +1,176 @@
+# Generated by Cargo
+# will have compiled files and executables
+debug/
+target/
+
+# Remove Cargo.lock from gitignore if creating an executable, leave it for libraries
+# More information here https://doc.rust-lang.org/cargo/guide/cargo-toml-vs-cargo-lock.html
+/talib/Cargo.lock
+/talib-sys/Cargo.lock
+
+# These are backup files generated by rustfmt
+**/*.rs.bk
+
+# MSVC Windows builds of rustc generate these, which store debugging information
+*.pdb
+dependencies
+
+
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+*$py.class
+
+# C extensions
+*.so
+
+# Distribution / packaging
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+share/python-wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+MANIFEST
+
+# PyInstaller
+#  Usually these files are written by a python script from a template
+#  before PyInstaller builds the exe, so as to inject date/other infos into it.
+*.manifest
+*.spec
+
+# Installer logs
+pip-log.txt
+pip-delete-this-directory.txt
+
+# Unit test / coverage reports
+htmlcov/
+.tox/
+.nox/
+.coverage
+.coverage.*
+.cache
+nosetests.xml
+coverage.xml
+*.cover
+*.py,cover
+.hypothesis/
+.pytest_cache/
+cover/
+
+# Translations
+*.mo
+*.pot
+
+# Django stuff:
+*.log
+local_settings.py
+db.sqlite3
+db.sqlite3-journal
+
+# Flask stuff:
+instance/
+.webassets-cache
+
+# Scrapy stuff:
+.scrapy
+
+# Sphinx documentation
+docs/_build/
+
+# PyBuilder
+.pybuilder/
+target/
+
+# Jupyter Notebook
+.ipynb_checkpoints
+
+# IPython
+profile_default/
+ipython_config.py
+
+# pyenv
+#   For a library or package, you might want to ignore these files since the code is
+#   intended to run in multiple environments; otherwise, check them in:
+# .python-version
+
+# pipenv
+#   According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control.
+#   However, in case of collaboration, if having platform-specific dependencies or dependencies
+#   having no cross-platform support, pipenv may install dependencies that don't work, or not
+#   install all needed dependencies.
+#Pipfile.lock
+
+# poetry
+#   Similar to Pipfile.lock, it is generally recommended to include poetry.lock in version control.
+#   This is especially recommended for binary packages to ensure reproducibility, and is more
+#   commonly ignored for libraries.
+#   https://python-poetry.org/docs/basic-usage/#commit-your-poetrylock-file-to-version-control
+#poetry.lock
+
+# pdm
+#   Similar to Pipfile.lock, it is generally recommended to include pdm.lock in version control.
+#pdm.lock
+#   pdm stores project-wide configurations in .pdm.toml, but it is recommended to not include it
+#   in version control.
+#   https://pdm.fming.dev/#use-with-ide
+.pdm.toml
+
+# PEP 582; used by e.g. github.com/David-OConnor/pyflow and github.com/pdm-project/pdm
+__pypackages__/
+
+# Celery stuff
+celerybeat-schedule
+celerybeat.pid
+
+# SageMath parsed files
+*.sage.py
+
+# Environments
+.env
+.venv
+env/
+venv/
+ENV/
+env.bak/
+venv.bak/
+
+# Spyder project settings
+.spyderproject
+.spyproject
+
+# Rope project settings
+.ropeproject
+
+# mkdocs documentation
+/site
+
+# mypy
+.mypy_cache/
+.dmypy.json
+dmypy.json
+
+# Pyre type checker
+.pyre/
+
+# pytype static type analyzer
+.pytype/
+
+# Cython debug symbols
+cython_debug/
+
+# pytest benchmark
+.benchmarks
+.wheel-test/
+.skill-test/

polars_talib-0.2.0/.vscode/settings.json

@@ -0,0 +1,7 @@
+{
+    "rust-analyzer.linkedProjects": [
+        "./talib-sys/Cargo.toml",
+        "./talib/Cargo.toml",
+        "./Cargo.toml"
+    ]
+}

polars_talib-0.2.0/.vscode/snippets.code-snippets

@@ -0,0 +1,48 @@
+{
+	"CDL": {
+		"prefix": "CDL",
+		"body": [
+			"#[polars_expr(output_type=Int32)]",
+			"fn $1(inputs: &[Series]) -> PolarsResult<Series> {",
+			"let open = &mut inputs[0].to_float()?.rechunk();",
+			"let high = &mut inputs[1].to_float()?.rechunk();",
+			"let low = &mut inputs[2].to_float()?.rechunk();",
+			"let close = &mut inputs[3].to_float()?.rechunk();",
+			"let (open_ptr, _open) = get_series_f64_ptr(open)?;",
+			"let (high_ptr, _high) = get_series_f64_ptr(high)?;",
+			"let (low_ptr, _low) = get_series_f64_ptr(low)?;",
+			"let (close_ptr, _close) = get_series_f64_ptr(close)?;",
+			"let len = close.len();"
+			""
+			"let res = ta_$1(open_ptr, high_ptr, low_ptr, close_ptr, len);"
+			"match res {",
+				"Ok(out) => Ok(Int32Chunked::new("", out).into_series()),",
+				"Err(ret_code) => ta_code2err(ret_code),",
+			"}",
+			"}",
+		],
+	},
+	"CDLK": {
+		"prefix": "CDLK",
+		"body": [
+			"#[polars_expr(output_type=Int32)]",
+			"fn $1(inputs: &[Series], kwargs: CDLKwargs) -> PolarsResult<Series> {",
+			"let open = &mut inputs[0].to_float()?.rechunk();",
+			"let high = &mut inputs[1].to_float()?.rechunk();",
+			"let low = &mut inputs[2].to_float()?.rechunk();",
+			"let close = &mut inputs[3].to_float()?.rechunk();",
+			"let (open_ptr, _open) = get_series_f64_ptr(open)?;",
+			"let (high_ptr, _high) = get_series_f64_ptr(high)?;",
+			"let (low_ptr, _low) = get_series_f64_ptr(low)?;",
+			"let (close_ptr, _close) = get_series_f64_ptr(close)?;",
+			"let len = close.len();"
+			""
+			"let res = ta_$1(open_ptr, high_ptr, low_ptr, close_ptr, len, , &kwargs);"
+			"match res {",
+				"Ok(out) => Ok(Int32Chunked::new("", out).into_series()),",
+				"Err(ret_code) => ta_code2err(ret_code),",
+			"}",
+			"}",
+		],
+	}
+}