[FIX] Allow abritrary types in Model as Layers

wbenbihi · Aug 23, 2022 · 8040817 · 8040817
1 parent 84cc940
commit 8040817
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/hourglass_tensorflow/types/config/model.py b/hourglass_tensorflow/types/config/model.py
@@ -26,6 +26,9 @@ class HTFModelAsLayers(BaseModel):
     outputs: keras.layers.Layer
     model: keras.models.Model
 
+    class Config:
+        arbitrary_types_allowed = True
+
 
 class HTFModelParams(HTFConfigField):
     name: str = "HourglassNetwork"