Skip to content
forked from ml6team/fondant

Large-scale data processing made easy and reusable

License

Notifications You must be signed in to change notification settings

andres-vv/fondant

 
 

Repository files navigation

Production-ready data processing made easy and shareable
Explore the docs »

Discord PyPI version License GitHub Workflow Status Coveralls


🚀 Production-ready 👶 Easy 👫 Shareable
Benefit from built-in features such as autoscaling, data lineage, and pipeline caching, and deploy to (managed) platforms such as Vertex AI, Sagemaker, and Kubeflow Pipelines. Implement your custom data processing code using datastructures you know such as Pandas dataframes. Move from local development to remote deployment without any code changes. Fondant components are defined by a clear interface, which makes them reusable and shareable.
Compose your own pipeline using components available on our hub.

🪤 Why Fondant?

With the advent of transfer learning and now foundation models, everyone has started sharing and reusing machine learning models. Most of the work now goes into building data processing pipelines, which everyone still does from scratch. This doesn't need to be the case, though, if processing components would be shareable and pipelines composable. Realizing this is the main vision behind Fondant.

Towards that end, Fondant offers:

  • 🔧 Plug ‘n’ play composable data processing pipelines
  • 🧩 Library containing off-the-shelf reusable components
  • 🐼 A simple Pandas based interface for creating custom components
  • 📊 Built-in lineage, caching, and data explorer
  • 🚀 Production-ready, scalable deployment
  • ☁️ Integration with runners across different clouds (Vertex, Sagemaker, Kubeflow)

(back to top)

💨 Getting Started

Eager to get started? Follow our step by step guide to get your first pipeline up and running.

(back to top)

🧩 Reusable components

Fondant comes with a library of reusable components that you can leverage to compose your own pipeline:

  • Data ingestion: S3, GCS, ABS, Hugging Face, local file system, ...
  • Data Filtering: Duplicates, language, visual style, topic, format, aesthetics, NSFW, license, ...
  • Data Enrichment: Captions, segmentations, embeddings, ...
  • Data Transformation: Image cropping, image resizing, text chunking, ....
  • Data retrieval: Common Crawl, LAION, ...

👉 Check our Component Hub for an overview of all available components

(back to top)

🪄 Example pipelines

We have created several ready-made example pipelines for you to use as a starting point for exploring Fondant. They are hosted as separate repositories containing a notebook tutorial so you can easily clone them and get started:

📖 RAG tuning pipeline
End-to-end Fondant pipelines to index and evaluate RAG (Retrieval-Augmented Generation) systems.

🛋️ ControlNet Interior Design Pipeline
An end-to-end Fondant pipeline to collect and process data for the fine-tuning of a ControlNet model, focusing on images related to interior design.

🖼️ Filter creative common license images
An end-to-end Fondant pipeline that starts from our Fondant-CC-25M creative commons image dataset and filters and downloads the desired images.

🔢 Datacomp pipeline
An end-to-end Fondant pipeline filtering image-text data to train a CLIP model for the DataComp competition.

⚒️ Installation

Fondant can be installed using pip:

pip install fondant

For more detailed installation options, check the installation page on our documentation.

👨‍💻 Usage

Pipeline

Fondant allows you to easily define data pipelines comprised of both reusable and custom components. The following pipeline for instance uses the reusable load_from_hf_hub component to load a dataset from the Hugging Face Hub and process it using a custom component:

pipeline.py

from fondant.pipeline import Pipeline


pipeline = Pipeline(pipeline_name="example pipeline", base_path="fs://bucket")

dataset = pipeline.read(
    "load_from_hf_hub",
    arguments={
        "dataset_name": "lambdalabs/pokemon-blip-captions"
    },
)

dataset = dataset.apply(
    "components/custom_component",
    arguments={
        "min_width": 600,
        "min_height": 600,
    },
)

Component

To create a custom component, you first need to describe its contract as a yaml specification. It defines the data consumed and produced by the component and any arguments it takes.

fondant_component.yaml

name: Custom component
description: This is a custom component
image: custom_component:latest

consumes:
  image:
    type: binary

produces:
  caption:
    type: utf8

args:
  argument1:
    description: An argument passed to the component at runtime
    type: str
  argument2:
    description: Another argument passed to the component at runtime
    type: str

Once you have your component specification, all you need to do is implement a constructor and a single .transform method and Fondant will do the rest. You will get the data defined in your specification partition by partition as a Pandas dataframe.

component/src/main.py

import pandas as pd
from fondant.component import PandasTransformComponent


class ExampleComponent(PandasTransformComponent):

    def __init__(self, *, argument1, argument2, **kwargs) -> None:
        """
        Args:
            argumentX: An argument passed to the component
        """
        # Initialize your component here based on the arguments

    def transform(self, dataframe: pd.DataFrame) -> pd.DataFrame:
        """Implement your custom logic in this single method
        Args:
            dataframe: A Pandas dataframe containing the data
        Returns:
            A pandas dataframe containing the transformed data
        """

For more advanced use cases, you can use the DaskTransformComponent instead.

Running your pipeline

Once you have a pipeline you can easily run (and compile) it by using the built-in CLI:

fondant run local pipeline.py

To see all available runner and arguments you can check the fondant CLI help pages

fondant --help

Or for a subcommand:

fondant <subcommand> --help

(back to top)

👭 Contributing

We welcome contributions of different kinds:

Issues If you encounter any issue or bug, please submit them as a Github issue. You can also submit a pull request directly to fix any clear bugs.
Suggestions and feedback Our roadmap and priorities are defined based on community feedback. To provide input, you can join our discord or submit an idea in our Github Discussions.
Framework code contributions If you want to help with the development of the Fondant framework, have a look at the issues marked with the good first issue label. If you want to add additional functionality, please submit an issue for it first.
Reusable components Extending our library of reusable components is a great way to contribute. If you built a component which would be useful for other users, please submit a PR adding them to the components/ directory. You can find a list of possible contributable components here or your own ideas are also welcome!

For a detailed view on the roadmap and day to day development, you can check our github project board.

You can also check out our architecture page to familiarize yourself with the Fondant architecture and repository structure.

Environment setup

We use poetry and pre-commit to enable a smooth developer flow. Run the following commands to set up your development environment:

pip install poetry
poetry install --all-extras
pre-commit install

(back to top)

About

Large-scale data processing made easy and reusable

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 94.1%
  • Dockerfile 3.7%
  • Shell 1.7%
  • Makefile 0.5%