metaflow-stubs

What it is and what it does

metaflow-stubs is a type stub package that provides static type information for Metaflow, the ML workflow framework originally developed at Netflix. Type stubs allow Python type checkers like mypy and IDE tools to understand Metaflow's API surface, enabling autocomplete, inline documentation, and type validation during development without requiring changes to Metaflow itself.

The package is a pure-Python wheel with a single runtime dependency on metaflow. It supports Python 3.7 through 3.12 and is maintained actively alongside Metaflow's main releases. Installation is straightforward and adds no runtime overhead—stubs are used only during development and type checking, not at runtime.

Use it for:

• Enable IDE autocomplete and inline documentation when writing Metaflow workflows in VS Code, PyCharm, or other type-aware editors.
• Catch type errors in Metaflow code before runtime using mypy or other static type checkers in CI/CD pipelines.
• Provide type hints for custom Metaflow decorators and step functions to improve code clarity and maintainability.
• Support refactoring of large Metaflow projects by making API changes visible to type checkers and IDEs.
• Document expected parameter and return types for Metaflow API calls without modifying the main package.

Worth the install?

AI-flagged interpretation of the facts on this page — verify before relying

Provides type stubs for the Metaflow package, enabling static type checking and IDE autocompletion for Metaflow workflows and APIs.

Yes, if you use Metaflow and want static type checking or IDE support. Type stubs add no runtime cost and integrate transparently with existing Metaflow code. The package is actively maintained, permissively licensed, and has no known vulnerabilities. Install it alongside metaflow to unlock type safety and better developer experience.

Install

metaflow-stubs on PyPI

pip install metaflow-stubs

uv add metaflow-stubs

poetry add metaflow-stubs

Installing metaflow-stubs

Before you install

Low friction install with a single pure-Python wheel dependency on metaflow itself. Actively maintained with a recent release (3 days old) and strong upstream support from Netflix's Metaflow project.

License in practice

Licensed under Apache Software License (permissive), imposing no restrictions on use, modification, or distribution in proprietary or open-source projects.

Quickstart

pip install metaflow-stubs

# Then use metaflow with type checking:
import metaflow
from metaflow import FlowSpec, step

class MyFlow(FlowSpec):
    @step
    def start(self):
        pass

Requires metaflow to be installed; type stubs alone do not provide runtime functionality.

Verify before relying

• Whether type stubs cover all public Metaflow APIs or only a subset of the full package surface.
• Compatibility with specific type checkers (mypy, pyright, pyre) beyond the general Python 3.7+ support.
• Whether stubs are actively synchronized with Metaflow's release cycle or lag behind.

Package facts

Evidence: metaflow_stubs-2.19.37-py2.py3-none-any.whl

Tags

More Application Frameworks packages