Add Continuous Integration to Public Branch

.github/workflows/lint.yml

@@ -0,0 +1,32 @@
+name: Lint
+
+on:
+  push:
+    branches: [public]
+  pull_request:
+    branches: 
+        - public
+        - '*'
+    workflow_dispatch:
+
+jobs:
+  ruff:
+    runs-on: ubuntu-latest
+    steps:
+      - name: Checkout
+        uses: actions/checkout@v4
+
+      - name: Set up Python
+        uses: actions/setup-python@v5
+        with:
+          python-version: "3.11"
+          cache: "pip"
+          cache-dependency-path: pyproject.toml
+
+      - name: Install dependencies
+        run: |
+          python -m pip install --upgrade pip
+          python -m pip install .[dev]
+
+      - name: Ruff check
+        run: make lint

.github/workflows/tests.yml

@@ -0,0 +1,32 @@
+name: Tests
+
+on:
+  push:
+    branches: [public]
+  pull_request:
+    branches: 
+        - public
+        - '*'
+    workflow_dispatch:
+
+jobs:
+  pytest:
+    runs-on: ubuntu-latest
+    steps:
+      - name: Checkout
+        uses: actions/checkout@v4
+
+      - name: Set up Python
+        uses: actions/setup-python@v5
+        with:
+          python-version: "3.11"
+          cache: "pip"
+          cache-dependency-path: pyproject.toml
+
+      - name: Install dependencies
+        run: |
+          python -m pip install --upgrade pip
+          python -m pip install .[dev]
+
+      - name: Run pytest
+        run: pytest --ignore cd_dynamax/dynamax

Makefile

@@ -0,0 +1,11 @@
+.PHONY: lint test
+
+clean:
+	ruff format --exclude cd_dynamax/dynamax --exclude "**/*.ipynb" .
+	ruff check --exclude cd_dynamax/dynamax --exclude "**/*.ipynb" . --fix
+
+lint:
+	ruff check . --exclude cd_dynamax/dynamax --exclude "**/*.ipynb"
+
+test:
+	pytest --ignore cd_dynamax/dynamax

cd_dynamax/src/__init__.py

@@ -28,31 +28,27 @@
     ### SSM classes ###
     "ContDiscreteNonlinearGaussianSSM",
     "ContDiscreteLinearGaussianSSM",
-
+    "SSM",
+    "Prior",
     ### Param classes ###
     "ParamsCDNLGSSM",
     "ParamsCDLGSSM",
-
     ### Non-linear filters/smoothers/forecasters ###
     "cdnlgssm_filter",
     "cdnlgssm_smoother",
     "cdnlgssm_forecast",
     "cdnlgssm_emissions",
-
     # Non-linear filtering Hyperparams
     "EKFHyperParams",
     "UKFHyperParams",
     "EnKFHyperParams",
-
     ### Linear filters/smoothers/forecasters/samplers ###
     "cdlgssm_filter",
     "cdlgssm_smoother",
     "cdlgssm_forecast",
     "cdlgssm_emissions",
     "cdlgssm_posterior_sample",
     "cdlgssm_joint_sample",
-
     # Linear filtering Hyperparams (typically don't need to use these directly, can rely on default KFHyperParams)
     "KFHyperParams",
-
 ]

cd_dynamax/src/continuous_discrete_linear_gaussian_ssm/__init__.py

@@ -9,4 +9,17 @@
 from .inference import cdlgssm_emissions
 from .inference import cdlgssm_posterior_sample
 from .inference import cdlgssm_joint_sample
-from .inference import KFHyperParams
+from .inference import KFHyperParams
+
+__all__ = [
+    "ParamsCDLGSSM",
+    "ParamsCDLGSSMDynamics",
+    "ContDiscreteLinearGaussianSSM",
+    "cdlgssm_filter",
+    "cdlgssm_smoother",
+    "cdlgssm_forecast",
+    "cdlgssm_emissions",
+    "cdlgssm_posterior_sample",
+    "cdlgssm_joint_sample",
+    "KFHyperParams",
+]