#26 use get_args

FaradayInstitution · Oct 5, 2023 · f116efd · f116efd
1 parent fc38062
commit f116efd
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/bpx/schema.py b/bpx/schema.py
@@ -1,4 +1,4 @@
-from typing import List, Literal, Union, Dict
+from typing import List, Literal, Union, Dict, get_args
 from pydantic import BaseModel, Field, Extra, root_validator
 from bpx import Function, InterpolatedTable
 from warnings import warn
@@ -288,7 +288,7 @@ def __init__(Self, **data):
     @root_validator(pre=True)
     def validate_extra_fields(cls, values):
         for k, v in values.items():
-            if not isinstance(v, (float, Function, InterpolatedTable)):
+            if not isinstance(v, get_args(FloatFunctionTable)):
                 raise TypeError(f"{k} must be of type 'FloatFunctionTable'")
         return values