add @beartype to GigaConfig

delphi-suite · Mar 7, 2024 · a061950 · a061950
1 parent 6dfa4bd
commit a061950
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/delphi/train/gigaconfig.py b/src/delphi/train/gigaconfig.py
@@ -1,9 +1,12 @@
 from dataclasses import dataclass
 from datetime import datetime
 
+from beartype import beartype
+
 from delphi.train.architectures import ModelTypes
 
 
+@beartype
 @dataclass
 class GigaConfig:
     """This is a terrible hack to get usable config objects to pass around