Project Icon

hamilton

简化数据转换DAG的构建与管理

Hamilton是一个Python库,用于构建数据转换的有向无环图(DAG)。它通过Python函数定义DAG,自动构建执行流程,具有良好的可移植性和表达能力。Hamilton支持多种执行环境,提供数据验证、实验跟踪等功能,并有UI界面用于可视化和监控。适用于ETL、机器学习、LLM应用等数据处理场景,有助于提高团队协作效率和代码可维护性。

Hamilton — 便携式且富有表现力的
数据转换有向无环图(DAG)

文档状态 支持的Python版本 PyPi版本 总下载量 Hamilton Slack


Hamilton是一个轻量级Python库,用于创建数据转换的有向无环图(DAG)。你的DAG是可移植的;它可以在任何运行Python的地方运行,无论是脚本、笔记本、Airflow管道还是FastAPI服务器等。你的DAG是富有表现力的;Hamilton具有广泛的功能来定义和修改DAG的执行(例如,数据验证、实验跟踪、远程执行)。

要创建DAG,只需编写常规的Python函数,并通过参数指定它们的依赖关系。如下所示,这将产生可读性强且随时可视化的代码。Hamilton会加载该定义并自动为你构建DAG!

创建项目
函数B()C()通过它们的参数引用函数A

Hamilton为任何处理数据的Python应用程序带来模块化和结构化:ETL管道、ML工作流、LLM应用、RAG系统、BI仪表板,而Hamilton UI允许你自动可视化、编目和监控执行。

Hamilton非常适合DAG,但如果你需要循环或条件逻辑来创建LLM代理或模拟,请查看我们的姐妹库 Burr 🤖 。

安装

Hamilton支持Python 3.8+。我们包含了可选的visualization依赖项以显示Hamilton DAG。对于可视化,需要在系统上单独安装Graphviz

pip install "sf-hamilton[visualization]"

要使用Hamilton UI,请安装uisdk依赖项。

pip install "sf-hamilton[ui,sdk]"

要在浏览器中试用Hamilton,请访问 www.tryhamilton.dev

为什么使用Hamilton?

数据团队编写代码以提供业务价值,但很少有团队有资源来标准化实践并提供质量保证。从概念验证到生产以及跨职能协作(例如,数据科学、工程、运维)对于大小团队来说仍然具有挑战性。Hamilton旨在帮助项目的整个生命周期:

  • 关注点分离。Hamilton将DAG的"定义"和"执行"分开,让数据科学家专注于解决问题,工程师管理生产管道。

  • 有效协作Hamilton UI提供了一个共享接口,供团队在整个开发周期中检查结果和调试故障。

  • 从开发到生产的低摩擦。使用@config.when()在执行环境之间修改DAG,而不是容易出错的if/else功能标志。笔记本扩展避免了将代码从笔记本迁移到Python模块的痛苦。

  • 可移植的转换。你的DAG独立于基础设施或编排,这意味着你可以在本地开发和调试,并在不同环境中重用代码(本地、Airflow、FastAPI等)。

  • 可维护的DAG定义。Hamilton通过一行代码自动构建DAG,无论它有10个还是1000个节点。它还可以将多个Python模块组装成一个管道,鼓励模块化。

  • 富有表现力的DAG函数修饰符是一个独特的功能,可以保持代码DRY并减少维护大型DAG的复杂性。其他框架不可避免地会导致代码冗余或函数臃肿。

  • 内置编码风格。Hamilton DAG使用Python函数定义,鼓励模块化、易读、自文档化和可单元测试的代码。

  • 数据和模式验证。使用@check_output装饰函数以验证输出属性,并引发警告或异常。添加SchemaValidator()适配器以自动检查类似数据框的对象(pandas、polars、Ibis等),以跟踪和验证它们的模式。

  • 为插件而生。Hamilton 的设计可以与所有工具完美配合,并提供了合适的抽象来创建与您的技术栈的自定义集成。我们活跃的社区将帮助您构建所需的一切!

Hamilton 用户界面

您可以在 Hamilton 用户界面 中跟踪 Hamilton DAG 的执行情况。它会自动填充包含血缘关系/追踪的数据目录,并提供执行可观察性以检查结果和调试错误。您可以将其作为本地服务器运行,或作为使用 Docker 的自托管应用程序运行。

描述1 描述2 描述3

DAG 目录、自动数据集分析和执行跟踪

开始使用 Hamilton 用户界面

  1. 要使用 Hamilton 用户界面,请安装依赖项(参见"安装"部分)并使用以下命令启动服务器:

    hamilton ui
    
  2. 首次连接时,创建一个"用户名"和一个新项目("project_id"应为"1")。

创建项目

  1. 通过创建一个包含您的"用户名"和"project_id"的"HamiltonTracker"对象并将其添加到您的"Builder"中来跟踪您的 Hamilton DAG。现在,您的 DAG 将出现在用户界面的目录中,所有执行都将被跟踪!

    from hamilton import driver
    from hamilton_sdk.adapters import HamiltonTracker
    import my_dag
    
    # 使用您的"用户名"和"project_id"
    tracker = HamiltonTracker(
       username="my_username",
       project_id=1,
       dag_name="hello_world",
    )
    
    # 将跟踪器添加到"Builder"中,这将把 DAG 添加到目录中
    dr = (
       driver.Builder()
       .with_modules(my_dag)
       .with_adapters(tracker)  # 在这里添加您的跟踪器
       .build()
    )
    
    # 执行"Driver"将跟踪结果
    dr.execute(["C"])
    

文档和学习资源

Hamilton 与 X 的比较

Hamilton 不是一个编排工具(您可能不需要一个),也不是一个特征存储(但您可以用它来构建一个!)。它的目的是帮助您构建和管理数据转换。如果您了解 dbt,Hamilton 对 Python 的作用就像 dbt 对 SQL 的作用一样。

另一种理解方式是考虑数据栈的不同层。Hamilton 位于资产层。它帮助您组织数据转换代码(表达层),管理变更,以及验证和测试数据。

目的示例工具
编排资产创建的操作系统Airflow, Metaflow, Prefect, Dagster
资产将表达式组织成有意义的单元
(如数据集、ML 模型、表格)
Hamilton, dbt, dlt, SQLMesh, Burr
表达编写数据转换的语言pandas, SQL, polars, Ibis, LangChain
执行执行数据转换Spark, Snowflake, DuckDB, RAPIDS
数据数据的物理表示,输入和输出S3, Postgres, 文件系统, Snowflake

有关更多信息,请参阅我们的为什么使用 Hamilton?页面和框架代码比较

📑 许可证

Hamilton 在 BSD 3-Clause Clear 许可下发布。详情请参阅 LICENSE

🌎 社区

👨‍💻 贡献

我们非常支持新贡献者的改变,无论大小!在打开拉取请求之前,请务必通过创建问题或在现有问题上发表评论来讨论潜在的更改。好的首次贡献包括创建示例或与您喜欢的 Python 库的集成! 要贡献代码,请查看我们的贡献指南开发者设置指南行为准则

😎 使用者

Hamilton最初由Stitch Fix开发,后来原始创建者成立了DAGWorks Inc!该库经过实战检验,自2019年以来一直支持生产用例。

阅读更多关于起源故事

🤝 代码贡献者

贡献者

🙌 特别鸣谢 & 🦟 Bug猎人

感谢我们出色的社区及其对Hamilton库的积极参与。

Nils Olsson, Michał Siedlaczek, Alaa Abedrabbo, Shreya Datar, Baldo Faieta, Anwar Brini, Gourav Kumar, Amos Aikman, Ankush Kundaliya, David Weselowski, Peter Robinson, Seth Stokes, Louis Maddox, Stephen Bias, Anup Joseph, Jan Hurst, Flavia Santos, Nicolas Huray, Manabu Niseki, Kyle Pounder, Alex Bustos, Andy Day, Alexander Cai, Nils Müller-Wendt, Paul Larsen, Kemal Eren, Jernej Frank, Noah Ridge

🎓 引用

我们感谢通过引用以下内容之一来引用Hamilton:

@inproceedings{DBLP:conf/vldb/KrawczykI22,
  title     = {Hamilton: a modular open source declarative paradigm for high level
               modeling of dataflows},
  author    = {Stefan Krawczyk and Elijah ben Izzy},
  editor    = {Satyanarayana R. Valluri and Mohamed Za{\"{\i}}t},
  booktitle = {1st International Workshop on Composable Data Management Systems,
               CDMS@VLDB 2022, Sydney, Australia, September 9, 2022},
  year      = {2022},
  url       = {https://cdmsworkshop.github.io/2022/Proceedings/ShortPapers/Paper6\_StefanKrawczyk.pdf},
  timestamp = {Wed, 19 Oct 2022 16:20:48 +0200},
  biburl    = {https://dblp.org/rec/conf/vldb/KrawczykI22.bib},
  bibsource = {dblp computer science bibliography, https://dblp.org}
}
@inproceedings{CEURWS:conf/vldb/KrawczykIQ22,
  title     = {Hamilton: enabling software engineering best practices for data transformations via generalized dataflow graphs},
  author    = {Stefan Krawczyk and Elijah ben Izzy and Danielle Quinn},
  editor    = {Cinzia Cappiello and Sandra Geisler and Maria-Esther Vidal},
  booktitle = {1st International Workshop on Data Ecosystems co-located with 48th International Conference on Very Large Databases (VLDB 2022)},
  pages     = {41--50},
  url       = {https://ceur-ws.org/Vol-3306/paper5.pdf},
  year      = {2022}
}
项目侧边栏1项目侧边栏2
推荐项目
Project Cover

豆包MarsCode

豆包 MarsCode 是一款革命性的编程助手,通过AI技术提供代码补全、单测生成、代码解释和智能问答等功能,支持100+编程语言,与主流编辑器无缝集成,显著提升开发效率和代码质量。

Project Cover

AI写歌

Suno AI是一个革命性的AI音乐创作平台,能在短短30秒内帮助用户创作出一首完整的歌曲。无论是寻找创作灵感还是需要快速制作音乐,Suno AI都是音乐爱好者和专业人士的理想选择。

Project Cover

有言AI

有言平台提供一站式AIGC视频创作解决方案,通过智能技术简化视频制作流程。无论是企业宣传还是个人分享,有言都能帮助用户快速、轻松地制作出专业级别的视频内容。

Project Cover

Kimi

Kimi AI助手提供多语言对话支持,能够阅读和理解用户上传的文件内容,解析网页信息,并结合搜索结果为用户提供详尽的答案。无论是日常咨询还是专业问题,Kimi都能以友好、专业的方式提供帮助。

Project Cover

阿里绘蛙

绘蛙是阿里巴巴集团推出的革命性AI电商营销平台。利用尖端人工智能技术,为商家提供一键生成商品图和营销文案的服务,显著提升内容创作效率和营销效果。适用于淘宝、天猫等电商平台,让商品第一时间被种草。

Project Cover

吐司

探索Tensor.Art平台的独特AI模型,免费访问各种图像生成与AI训练工具,从Stable Diffusion等基础模型开始,轻松实现创新图像生成。体验前沿的AI技术,推动个人和企业的创新发展。

Project Cover

SubCat字幕猫

SubCat字幕猫APP是一款创新的视频播放器,它将改变您观看视频的方式!SubCat结合了先进的人工智能技术,为您提供即时视频字幕翻译,无论是本地视频还是网络流媒体,让您轻松享受各种语言的内容。

Project Cover

美间AI

美间AI创意设计平台,利用前沿AI技术,为设计师和营销人员提供一站式设计解决方案。从智能海报到3D效果图,再到文案生成,美间让创意设计更简单、更高效。

Project Cover

AIWritePaper论文写作

AIWritePaper论文写作是一站式AI论文写作辅助工具,简化了选题、文献检索至论文撰写的整个过程。通过简单设定,平台可快速生成高质量论文大纲和全文,配合图表、参考文献等一应俱全,同时提供开题报告和答辩PPT等增值服务,保障数据安全,有效提升写作效率和论文质量。

投诉举报邮箱: service@vectorlightyear.com
@2024 懂AI·鲁ICP备2024100362号-6·鲁公网安备37021002001498号