numpy-pydantic-types is a Python library providing numpy scalar data types compatible with pydantic, allowing for easy validation and enforcement of numerical precision in pydantic models.
pydantic models rely on Python's native types (e.g., int, float, str) for data validation, which can be limiting when working with scientific or numerical applications that require specific numpy types, like float32, uint16, or int8. This library bridges that gap by introducing numpy scalar types as valid pydantic field types, allowing you to define precise, controlled models that integrate with scientific workflows.
- Numpy Scalar Compatibility: Define
pydanticmodels usingnumpyscalar types likefloat32,uint32,int8, etc. - Strict Type Enforcement: Ensures model fields adhere to specific precision and range constraints.
- Simple Integration: Easily import and use
numpytypes in yourpydanticmodels.
pip install numpy_pydantic_types
This library enables the use of numpy scalar types directly within pydantic models. Here’s how to set up a model using numpy types:
from pydantic import BaseModel, ValidationError
from numpy_pydantic_types import Float32, UInt32, UInt16
class ScientificModel(BaseModel):
precision_value: Float32 # Enforces a 32-bit floating point
sample_count: UInt32 # Enforces a 32-bit unsigned integer
sensor_id: UInt16 # Enforces a 16-bit unsigned integer
# Example data
data = {
"precision_value": 1.234567,
"sample_count": 4294967295, # Max value for UInt32
"sensor_id": 65535 # Max value for UInt16
}
# Instantiate the model with the specific numpy types
try:
model = ScientificModel(**data)
print("Model validated successfully:", model)
except ValidationError as e:
print("Validation error:", e)
The library provides a range of numpy scalar types that can be used in pydantic models, including:
- Float Types:
Float32,Float64 - Signed Integer Types:
Int8,Int16,Int32,Int64 - Unsigned Integer Types:
UInt8,UInt16,UInt32,UInt64
Using specific numpy scalar types can help enforce type constraints in applications where precision or memory footprint is crucial, such as scientific computations, data analysis, or embedded systems.
from numpy_pydantic_types import Int8, Float64
class DataProcessingModel(BaseModel):
temperature: Float64
adjustment_factor: Int8
- Precision: Control over numerical precision, essential in scientific or numerical applications.
- Range Enforcement: Ensures values conform to specific data type ranges (e.g.,
UInt8ranges from 0 to 255). - Memory Efficiency: Helps reduce memory consumption by enforcing smaller data types.
MIT License