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.

  • Release v0.5.0 - Multi-step Workflows: Multi-step workflows and matrix nulls as wildcards.

See Git commit history for detailed implementation progress

🛣️ Upcoming Releases

  • Release v0.6.0 - Step Output Chaining: Step output references and cache restoration for resumable workflows.
  • Release v0.7.0 - Enhanced Workflow: Dry and comparison runs, runtime estimation and processing summary
  • Release v0.8.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.5.0.tar.gz (52.2 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.5.0-py3-none-any.whl (54.8 kB view details)

Uploaded Python 3

File details

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

File metadata

  • Download URL: causaliq_workflow-0.5.0.tar.gz
  • Upload date:
  • Size: 52.2 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.5.0.tar.gz
Algorithm Hash digest
SHA256 d51bda623e2b2715316b7762ee1c2fcf298e25306ca9c30f62bc5232e0cb2fff
MD5 b220cfdb18f2c27e7399167003c18c5a
BLAKE2b-256 444e77bb6d14c28397334838301c22f858715035f13512a1e78bdec8680f1d72

See more details on using hashes here.

File details

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

File metadata

File hashes

Hashes for causaliq_workflow-0.5.0-py3-none-any.whl
Algorithm Hash digest
SHA256 19406ea4fc2744601244e0af977959ed2560cfa7f72ec4e70437753a4d20174e
MD5 efc21ed5582f5e483201cd4fa41d54d4
BLAKE2b-256 3657afd6127a8dd2cd3847b8234c62a2eb0df79b401a6c9725808e0e0cdd10c5

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