---
id: datarobot-drum
version: "1.17.18"
license: Apache License, Version 2.0
license_treatment: permissive
maintenance: active
---

# datarobot-drum — DRUM - develop, test and deploy custom models

License: permissive · Maintenance: active · Downloads: 97.1K/mo

## What it is and what it does

DRUM is a local development and testing tool for custom machine learning models that bridges the gap between model development and DataRobot deployment. It allows you to verify that Python, R, or Java models run correctly and make predictions before uploading them to DataRobot's platform. The tool operates in multiple modes: batch scoring for making predictions on CSV data, performance testing to measure latency and memory usage, and validation checks to test model robustness on edge cases like null values.

The package includes a prediction server mode for real-time inference, support for runtime parameters (credentials, URLs) loaded from YAML files, and Docker integration for containerized testing. It depends on a large ecosystem of standard data science and web libraries (numpy, pandas, flask, docker, gunicorn) plus DataRobot's own client and MLOps packages, making it tightly integrated with the DataRobot ecosystem. DRUM is designed for Linux and macOS; Windows users must use WSL.

Use it for:

- Validate a custom Python, R, or Java model locally before uploading to DataRobot for production deployment
- Measure model latency, throughput, and memory consumption under realistic batch-scoring workloads
- Test model behavior on edge cases and missing values to ensure it meets DataRobot's validation requirements
- Run batch predictions on CSV files locally for development and debugging without deploying to the cloud
- Generate model code templates and scaffolding for new custom models in Python, R, or Java

## Worth the install?

AI-flagged interpretation of the facts above — verify before relying.

DRUM is a command-line tool for developing, testing, and deploying custom machine learning models written in Python, R, or Java before uploading them to DataRobot's platform.

Yes—install if you develop custom models for DataRobot. DRUM is actively maintained, has no known vulnerabilities, uses a permissive license, and supports current Python versions. The substantial dependency footprint is justified by its role as a full-featured model testing and serving tool. Not worth installing if you don't use DataRobot or need a general-purpose model server unrelated to DataRobot deployment.

## Install

pip install datarobot-drum
uv add datarobot-drum
poetry add datarobot-drum

## Installing datarobot-drum

Before you install: Installation is straightforward with low friction; the package is actively maintained with a recent release and supports current Python versions (3.8–3.12). The 30 runtime dependencies are substantial but standard for a model-serving tool.

License in practice: Licensed under Apache License 2.0 (permissive), allowing free use, modification, and distribution with minimal restrictions—suitable for both open-source and commercial projects.

Quickstart:

pip install datarobot-drum

# Score a regression model
drum score --code-dir ./model/ --input data.csv --target-type regression

# Run performance and memory tests
drum perf-test --code-dir ./model/ --input data.csv --target-type binary --positive-class-label yes --negative-class-label no

Linux or macOS required; Windows users must use WSL. Java models require JRE >= 11. R models require Python >= 3.6 and R framework installed.

Verify before relying:

- Exact Python version support boundaries within the 3.8–3.12 range listed in classifiers
- Whether all 30 runtime dependencies are required for basic scoring or if some are optional for specific features
- Performance characteristics and typical latency for batch scoring on different dataset sizes

## Package facts

- License: Apache License, Version 2.0 (permissive)
- Python support: supports_current
- Install friction: low
- Maintenance: active
- Downloads: 97.1K/month (top 15,000 on PyPI)
- Known vulnerabilities: none known

## Tags

custom model testing local development, machine learning model validation, batch scoring predictions, model performance profiling, datarobot model deployment, python r java model runner, model performance memory testing, model-testing, datarobot-integration, batch-scoring

[View on SkillFed](https://skillfed.io/packages/datarobot-drum) · [View on PyPI](https://pypi.org/project/datarobot-drum/)