notebook-generator

Generate Databricks-ready Python notebooks for testing PySpark classes.

Updated Aug 23, 2026
One-click install
npx skills add https://github.com/talhayilmazai/ai-agent-in-customer-repository --skill notebook-generator
Or copy as Structured Prompt for Agent
Please help me install this Agent Skill.
Skill: notebook-generator
Source: https://github.com/talhayilmazai/ai-agent-in-customer-repository/tree/main/.cursor/skills/notebookGeneration
Command: npx skills add https://github.com/talhayilmazai/ai-agent-in-customer-repository --skill notebook-generator

SYSTEM DOCUMENTATION & REQUIREMENTS

What problem does it solve?

This Skill generates ready-to-run Databricks notebooks to streamline testing of Rocks extension PySpark classes, reducing the need to run the full data pipeline during development.

Core Features & Use Cases

  • Generate a Databricks-ready notebook for a given rocks_extension module, class, or config section.
  • Include environment setup, optional class extension for testing, and metadata injection steps to enable isolated experimentation.
  • Use Case: quickly validate changes to a Rocks class by running a focused notebook in Databricks and inspecting outputs.

Quick Start

Use the notebook generator to create a test notebook for BootsLookupTableGenerator. Save as notebooks/test_[module]_[class_name].ipynb and open it in Databricks to execute.

Frequently Asked Questions about notebook-generator

High-intent search queries and answers about installing and using this skill.

FAQPage Schema
How do I generate a Databricks notebook to test a PySpark class?

To generate a Databricks notebook for testing a PySpark class, this Skill locates the Python implementation, identifies the parent Rocks class, and injects runtime metadata and path overrides to produce a ready-to-run test environment.

Can I test a Rocks extension module in Databricks without running the full data pipeline?

Yes, you can test a Rocks extension module in Databricks without the full pipeline by generating an isolated notebook that includes environment setup, class instantiation, metadata injection, and result verification steps.

What steps are included in a generated Databricks PySpark test notebook?

A generated Databricks PySpark test notebook includes environment setup, optional class extension for testing, class instantiation, metadata injection, execution, and result verification steps.

Do I need to manually configure test data paths when testing Rocks extension classes in Databricks?

You do not need to manually configure test data paths, because the notebook generator automatically locates runtime config sections and injects path overrides for test data into the Databricks notebook.

What is the best way to validate changes to a Rocks class in Databricks?

The best way to validate changes to a Rocks class in Databricks is generating a focused notebook, saving it as test_[module]_[class_name].ipynb, and executing it to inspect the outputs directly.

Does the generated Databricks notebook support metadata injection for isolated experimentation?

The generated Databricks notebook supports metadata injection for isolated experimentation by locating runtime metadata and config sections and applying them during class instantiation and execution.