refactor dataclasses init

whylabs · Jul 28, 2023 · a658a0d · a658a0d
1 parent f8b3e1e
commit a658a0d
Showing 1 changed file with 4 additions and 16 deletions.
diff --git a/whylabs_toolkit/helpers/schema.py b/whylabs_toolkit/helpers/schema.py
@@ -13,26 +13,14 @@
 
 @dataclass
 class ColumnsClassifiers:
- inputs: List[str] = field(default=None) # type: ignore
- outputs: List[str] = field(default=None) # type: ignore
-
- def __post_init__(self) -> None:
- if self.inputs is None:
- self.inputs = []
- if self.outputs is None:
- self.outputs = []
+ inputs: List[str] = field(default_factory=list) # type: ignore
+ outputs: List[str] = field(default_factory=list) # type: ignore
 
 
 @dataclass
 class ColumnsDiscreteness:
- discrete: List[str] = field(default=None) # type: ignore
- continuous: List[str] = field(default=None) # type: ignore
-
- def __post_init__(self) -> None:
- if self.discrete is None:
- self.discrete = []
- if self.continuous is None:
- self.continuous = []
+ discrete: List[str] = field(default_factory=list) # type: ignore
+ continuous: List[str] = field(default_factory=list) # type: ignore
 
 
 class UpdateEntity(ABC):