Skip to main content

Workflow engine for causal discovery and inference

Project description

causaliq-workflow

Python Versions License: MIT Coverage

GitHub Actions-inspired workflow orchestration for causal discovery experiments within the CausalIQ ecosystem. Execute causal discovery workflows using familiar CI/CD patterns with conservative execution and comprehensive action framework.

Status

🚧 Active Development - This repository is currently in active development, which involves:

  • migrating functionality from the legacy monolithic discovery repo to support legacy experiments and analysis
  • ensure CausalIQ development standards are met
  • adding new features to provide a comprehensive, open, causal discovery workflow.

Features

Implemented Releases

  • Release v0.1.0 - Workflow Foundations: Plug-in actions, basic workflow and CLI support, 100% test coverage

  • Release v0.2.0 - Knowledge Workflows: Integrate with causaliq-knowledge generate_graph action and write results to workflow caches.

  • Release v0.3.0 - Aggregation Workflows: Matrix-driven aggregation processing with filter expressions for multi-source workflows.

  • Release v0.4.0 - Conservative Execution: Formalise action patterns and implement conservative execution for creation, update and aggregation steps.

See Git commit history for detailed implementation progress

🛣️ Upcoming Releases

  • Release v0.5.0 - Step Output Chaining: Step output references and cache restoration for resumable workflows.
  • Release v0.6.0 - Enhanced Workflow: Dry and comparison runs, runtime estimation and processing summary
  • Release v0.7.0 - Discovery Integration: Structure learning algorithms integrated

causaliq-core Integration

causaliq-workflow builds on causaliq-core for its action framework and caching infrastructure:

  • CausalIQActionProvider - Base class for all action providers
  • ActionInput/ActionResult - Type-safe action interfaces
  • ActionValidationError/ActionExecutionError - Exception handling
  • TokenCache/JsonCompressor - SQLite-based caching with JSON tokenisation

Brief Example Usage

Example Workflow Definition, experiment.yml:

description: "Causal Discovery Experiment"
id: "experiment-001"
workflow_cache: "results/{{id}}_cache.db"  # All results stored here

matrix:
  network: ["asia", "cancer"]
  algorithm: ["pc", "ges"]
  sample_size: ["100", "1K"]

steps:
  - name: "Structure Learning"
    uses: "causaliq-discovery"
    with:
      algorithm: "{{algorithm}}"
      sample_size: "{{sample_size}}"
      dataset: "data/{{network}}"
      # Results cached with key: {network, algorithm, sample_size}

Execute with modes:

cqflow run experiment.yml --mode=dry-run  # Validate and preview (default)
cqflow run experiment.yml --mode=run      # Execute (skip if outputs exist)
cqflow run experiment.yml --mode=force    # Re-execute all without skip

Note that cqflow is a short synonym for causaliq-workflow which can also be used.

Upcoming Key Innovations

🔄 Workflow Orchestration

  • Continuous Integration (CI) testing: Workflow specification syntax
  • Dask distributed computing: Scalable parallel processing
  • Dependency management: Automatic handling of data and processing dependencies
  • Error recovery: Robust handling of failures and restarts

📊 Experiment Management

  • Configuration management: YAML-based experiment specifications
  • Parameter sweeps: Systematic exploration of algorithm parameters
  • Version control: Git-based tracking of experiments and results
  • Reproducibility: Deterministic execution with seed management

Integration with CausalIQ Ecosystem

  • 🔍 CausalIQ Discovery is called by this package to perform structure learning.
  • 📊 CausalIQ Analysis is called by this package to perform results analysis and generate assets for research papers.
  • 🔮 CausalIQ Predict is called by this package to perform causal prediction.
  • 🔄 Zenodo Synchronisation is used by this package to download datasets and upload results.
  • 🧪 CausalIQ Papers are defined in terms of CausalIQ Workflows allowing the reproduction of experiments, results and published paper assets created by the CausalIQ ecosystem.

LLM Support

The following provides project-specific context for this repo which should be provided after the personal and ecosystem context:

tbc

Prerequisites

  • Python 3.9-3.13
  • Git

Installation

git clone https://github.com/causaliq/causaliq-workflow.git
cd causaliq-workflow

# Set up development environment
scripts/setup-env.ps1 -Install
scripts/activate.ps1

Example workflows: docs/userguide/examples.md

Research Context

Supporting research for May 2026 paper on LLM integration for intelligent model averaging. The CI workflow architecture enables sophisticated experimental designs while maintaining familiar syntax for the research community.

Migration target: Existing workflows from monolithic discovery repo by end 2026.

Quick Start

# Clone and set up
git clone https://github.com/causaliq/causaliq-workflow.git
cd causaliq-workflow
.\scripts\setup-env.ps1 -Install
.\scripts\activate.ps1

# Verify installation
cqflow --help
.\scripts\check_ci.ps1

Documentation

Full API documentation is available at: http://127.0.0.1:8000/ (when running mkdocs serve)

Contributing

This repository is part of the CausalIQ ecosystem. For development setup:

  1. Clone the repository
  2. Run scripts/setup-env -Install to set up environments
  3. Run scripts/check_ci to verify all tests pass
  4. Start documentation server with mkdocs serve

Supported Python Versions: 3.9, 3.10, 3.11, 3.12, 3.13 Default Python Version: 3.11
License: MIT

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

causaliq_workflow-0.4.0.tar.gz (49.8 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

causaliq_workflow-0.4.0-py3-none-any.whl (52.4 kB view details)

Uploaded Python 3

File details

Details for the file causaliq_workflow-0.4.0.tar.gz.

File metadata

  • Download URL: causaliq_workflow-0.4.0.tar.gz
  • Upload date:
  • Size: 49.8 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.11.9

File hashes

Hashes for causaliq_workflow-0.4.0.tar.gz
Algorithm Hash digest
SHA256 736d3d0741940f2888d956a5978d85647b276fbd0fdb60c394a65523c6630905
MD5 47931796a957c10dd2447c01fdbfa4cf
BLAKE2b-256 18f6827f13665622b3762459978c6666676962c7197d74e24f49cbce12bc6e21

See more details on using hashes here.

File details

Details for the file causaliq_workflow-0.4.0-py3-none-any.whl.

File metadata

File hashes

Hashes for causaliq_workflow-0.4.0-py3-none-any.whl
Algorithm Hash digest
SHA256 ad7605e8db114008b675147922f8b61f00e7ae03a4867c3a8670bca865d0210a
MD5 b6609bc24cf4140dc3b5e966ef258cb1
BLAKE2b-256 5a4f6843d610cd4ddcd02480a08ab6ebe8f13f0b7ed85f8620625f1e32e21925

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page