Ecgrl

.github/workflows/ci.yml

@@ -0,0 +1,37 @@
+name: CI
+
+on:
+  push:
+    branches: [ main, master ]
+  pull_request:
+
+permissions:
+  contents: read
+
+jobs:
+  all-tests:
+    runs-on: ubuntu-latest
+    steps:
+      - name: Checkout repository
+        uses: actions/checkout@v4
+
+      - name: Set up Python 3.10
+        uses: actions/setup-python@v5
+        with:
+          python-version: "3.10"
+
+      - name: Install uv
+        run: pip install --upgrade pip && pip install uv
+
+      - name: Install package (editable) with judge extras
+        run: uv pip install --system -e ".[judge]"
+
+      - name: Smoke test import
+        run: |
+          python - << 'PY'
+          import importlib
+          import torch
+          # Ensure core package imports
+          importlib.import_module("ecg_bench")
+          print("ecg_bench import OK; torch=", torch.__version__)
+          PY

.gitignore

@@ -1,13 +1,12 @@
 # Byte-compiled / optimized / DLL files
 __pycache__/
-*.py[cod]
+*.py[codz]
 *$py.class
 
 # C extensions
 *.so
 
 # Distribution / packaging
-.vscode/
 .Python
 build/
 develop-eggs/
@@ -47,7 +46,7 @@ htmlcov/
 nosetests.xml
 coverage.xml
 *.cover
-*.py,cover
+*.py.cover
 .hypothesis/
 .pytest_cache/
 cover/
@@ -107,17 +106,24 @@ ipython_config.py
 #   commonly ignored for libraries.
 #   https://python-poetry.org/docs/basic-usage/#commit-your-poetrylock-file-to-version-control
 #poetry.lock
+#poetry.toml
 
 # pdm
 #   Similar to Pipfile.lock, it is generally recommended to include pdm.lock in version control.
+#   pdm recommends including project-wide configuration in pdm.toml, but excluding .pdm-python.
+#   https://pdm-project.org/en/latest/usage/project/#working-with-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/latest/usage/project/#working-with-version-control
-.pdm.toml
+#pdm.toml
 .pdm-python
 .pdm-build/
 
+# pixi
+#   Similar to Pipfile.lock, it is generally recommended to include pixi.lock in version control.
+#pixi.lock
+#   Pixi creates a virtual environment in the .pixi directory, just like venv module creates one
+#   in the .venv directory. It is recommended not to include this directory in version control.
+.pixi
+
 # PEP 582; used by e.g. github.com/David-OConnor/pyflow and github.com/pdm-project/pdm
 __pypackages__/
 
@@ -130,6 +136,7 @@ celerybeat.pid
 
 # Environments
 .env
+.envrc
 .venv
 env/
 venv/
@@ -168,21 +175,46 @@ cython_debug/
 #  option (not recommended) you can uncomment the following to ignore the entire idea folder.
 #.idea/
 
+# Abstra
+# Abstra is an AI-powered process automation framework.
+# Ignore directories containing user credentials, local state, and settings.
+# Learn more at https://abstra.io/docs
+.abstra/
+
+# Visual Studio Code
+#  Visual Studio Code specific template is maintained in a separate VisualStudioCode.gitignore 
+#  that can be found at https://github.com/github/gitignore/blob/main/Global/VisualStudioCode.gitignore
+#  and can be added to the global gitignore or merged into this file. However, if you prefer, 
+#  you could uncomment the following to ignore the entire vscode folder
+# .vscode/
+
+# Ruff stuff:
+.ruff_cache/
+
 # PyPI configuration file
 .pypirc
 
+# Cursor
+#  Cursor is an AI-powered code editor. `.cursorignore` specifies files/directories to
+#  exclude from AI features like autocomplete and code analysis. Recommended for sensitive data
+#  refer to https://docs.cursor.com/context/ignore-files
+.cursorignore
+.cursorindexingignore
 
-# Extra
-ecg_bench/data
-ecg_bench/runs
-ecg_bench/wandb
-ecg_bench/pngs
-.huggingface/
-ecg_bench/atharva_demo/data
-ecg_bench/atharva_demo/dump/testchat
-
-.gradio/
+# Marimo
+marimo/_static/
+marimo/_lsp/
+__marimo__/
 
-*.sh
+# Hugging Face
+.huggingface/
 
-ecg_bench/scripts_dev
+### ecg_bench ignores
+ecg_bench/runs/
+ecg_bench/data/
+ecg_bench/logs/
+wandb/
+scripts/
+*.pt
+tests/OpenTSLM
+pngs/

.pre-commit-config.yaml

@@ -0,0 +1,15 @@
+repos:
+  - repo: local
+    hooks:
+      - id: ruff-format
+        name: ruff (format)
+        entry: python3 -m ruff format
+        language: system
+        types_or: [python]        # only run on changed .py files
+
+      - id: ruff
+        name: ruff (lint)
+        entry: python3 -m ruff
+        language: system
+        types_or: [python]
+        args: ["check", "--fix", "--show-fixes"]

.vscode/settings.json

@@ -0,0 +1,8 @@
+{
+    "files.watcherExclude": {
+        "**/data/**": true,
+        "**/target/**": true,
+        "**/transformers/**": true,
+        "**/wandb/**": true
+    }
+}