initial public release commit

Author: DrRzGa

.dockerignore

@@ -0,0 +1 @@
+.venv

.gitignore

@@ -0,0 +1,49 @@
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+*$py.class
+
+# 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
+
+
+.pdm.toml
+.pdm-python
+.pdm-build/
+
+# PEP 582; used by e.g. github.com/David-OConnor/pyflow and github.com/pdm-project/pdm
+__pypackages__/
+
+# Environments
+.env
+.venv
+env/
+venv/
+ENV/
+env.bak/
+venv.bak/
+
+# mypy
+.mypy_cache/
+.dmypy.json
+dmypy.json
+
+artifacts/*
+.vscode/*

.pre-commit-config.yaml

@@ -0,0 +1,46 @@
+fail_fast: True
+
+repos:
+-   repo: https://github.com/pre-commit/pre-commit-hooks
+    rev: v5.0.0
+    hooks:
+    -   id: fix-encoding-pragma
+        exclude: 'docs/.*'
+    -   id: check-yaml
+        exclude: 'docs/.*'
+    -   id: trailing-whitespace
+        exclude: 'docs/.*'
+    -   id: name-tests-test
+        args: ["--pytest-test-first"]
+        exclude: 'docs/.*'
+    -   id: check-added-large-files
+        args: ["--maxkb=10"]
+    -   id: check-docstring-first
+        exclude: 'docs/.*'
+    -   id: check-case-conflict
+        exclude: 'docs/.*'
+    -   id: check-shebang-scripts-are-executable
+        exclude: 'docs/.*'
+
+-   repo: https://github.com/gitleaks/gitleaks
+    rev: v8.16.1
+    hooks:
+    -   id: gitleaks
+        exclude: '.*/tests/.*'
+
+- repo: https://github.com/astral-sh/ruff-pre-commit
+  rev: v0.9.3
+  hooks:
+    - id: ruff
+    - id: ruff-format
+
+-   repo: https://github.com/pre-commit/mirrors-mypy
+    rev: v1.14.1
+    hooks:
+    -   id: mypy
+        name: mypy
+        entry: mypy
+        language: python
+        types: [python]
+        additional_dependencies:
+            - "pydantic>=2.8.2"

.python-version

@@ -0,0 +1 @@
+3.10.12

docker/Dockerfile

@@ -0,0 +1,5 @@
+FROM sinapsis-nvidia:base
+RUN apt-get update && apt-get install -y ffmpeg python3.10-dev
+
+RUN uv pip install sinapsis-data-tools[all] --extra-index-url https://pypi.sinapsis.tech
+WORKDIR /app/.venv

docker/Dockerfile-dev

@@ -0,0 +1,12 @@
+FROM sinapsis-nvidia:base
+# install essential packages
+RUN apt-get update && apt-get install -y ffmpeg python3.10-dev
+
+COPY . /app/
+WORKDIR /app
+
+RUN uv sync --frozen  --all-extras --all-packages
+RUN uv build --all-packages
+
+# install the generated wheels for this app
+RUN uv pip install /app/dist/sinapsis_*none-any.whl

docker/README.md

@@ -0,0 +1,4 @@
+> [!IMPORTANT]
+> To build the sinapsis-data-tools image you need
+the sinapsis-nvidia:base image. To build it, please follow the instructions in
+the [sinapsis repo](https://github.com/Sinapsis-ai/sinapsis?tab=readme-ov-file#docker)

docker/compose.yaml

@@ -0,0 +1,20 @@
+services:
+  sinapsis-data-tools: &default
+    image: sinapsis-data-tools:base
+    build:
+      context: ../
+      dockerfile: docker/Dockerfile
+      ssh:
+        - default
+    environment:
+      PYTHONPATH: /lib/python3.10/:/app/.venv/lib/python3.10/site-packages
+      LD_LIBRARY_PATH: /app/.venv/lib/python3.10/site-packages/nvidia/cublas/lib
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: nvidia
+              count: 1
+              capabilities: [ gpu ]
+    volumes:
+      - "~/.cache/sinapsis/:/root/.cache/sinapsis"