19634252007

Committed 24 Nov 2025 12:25PM UTC coverage: 88.22% (-0.04%) from 88.262%

Build # 19634252007

Build Type

Pull #678

github

Committed by

web-flow

Commit Message

Merge c46228df3 into 539258334

Pull Request Pull Request #678: Feature/partial First Pydantic Models for parameter validation

Coverage Stats

540 of 585 new or added lines in 9 files covered. (92.31%)

27 existing lines in 4 files now uncovered.

7234 of 8200 relevant lines covered (88.22%)

5.28 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

90.48

/elephant/schemas/function_validator.py

from functools import wraps
from inspect import signature
from pydantic import BaseModel

skip_validation = False

def validate_with(model_class: type[BaseModel]):
    """
    A decorator that validates the inputs of a function using a Pydantic model.
    Works for both positional and keyword arguments.
    """
    def decorator(func):
        sig = signature(func)

        @wraps(func)
        def wrapper(*args, **kwargs):

            if not skip_validation:
                # Bind args & kwargs to function parameters
                bound = sig.bind_partial(*args, **kwargs)
                bound.apply_defaults()
                data = bound.arguments

                # Validate using Pydantic
                model_class(**data)

            # Call function
            return func(*args, **kwargs)
        return wrapper
    return decorator

def activate_validation():
    skip_validation = False

def deactivate_validation():
    skip_validation = True

1	from functools import wraps	6✔
2	from inspect import signature	6✔
3	from pydantic import BaseModel	6✔
4
5	skip_validation = False	6✔
6
7	def validate_with(model_class: type[BaseModel]):	6✔
8	"""
9	A decorator that validates the inputs of a function using a Pydantic model.
10	Works for both positional and keyword arguments.
11	"""
12	def decorator(func):	6✔
13	sig = signature(func)	6✔
14
15	@wraps(func)	6✔
16	def wrapper(args, *kwargs):	6✔
17
18	if not skip_validation:	6✔
19	# Bind args & kwargs to function parameters
20	bound = sig.bind_partial(args, *kwargs)	6✔
21	bound.apply_defaults()	6✔
22	data = bound.arguments	6✔
23
24	# Validate using Pydantic
25	model_class(**data)	6✔
26
27	# Call function
28	return func(args, *kwargs)	6✔
29	return wrapper	6✔
30	return decorator	6✔
31
32	def activate_validation():	6✔
NEW 33	skip_validation = False	×
34
35	def deactivate_validation():	6✔
NEW 36	skip_validation = True	×

NeuralEnsemble / elephant / 19634252007

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous