inline argument validation

mmore500 · Feb 15, 2025 · 5eec37c · 5eec37c
1 parent 8838962
commit 5eec37c
Showing 1 changed file with 4 additions and 8 deletions.
diff --git a/teeplot/teeplot.py b/teeplot/teeplot.py
@@ -380,13 +380,6 @@ def teed(*args, **kwargs):
         saveit()
 
 
-def validate_teewrap_kwargs(teeplot_kwargs: dict[str, object]):
-    if not all(k.startswith("teeplot_") for k in teeplot_kwargs):
-        raise ValueError(
-            "The `teewrap` decorator only accepts teeplot_* keyword arguments"
-        )
-
-
 def teewrap(
     **teeplot_kwargs: object,
 ):
@@ -398,7 +391,10 @@ def teewrap(
     the same across function calls. For printing attributes on a per-call basis, 
     see `teeplot_outinclude` in `teeplot.tee`.
     """
-    validate_teewrap_kwargs(teeplot_kwargs)
+    if not all(k.startswith("teeplot_") for k in teeplot_kwargs):
+        raise ValueError(
+            "The `teewrap` decorator only accepts teeplot_* keyword arguments"
+        )
 
     def decorator(f: typing.Callable):
         @functools.wraps(f)