diff --git a/logai/dataloader/data_model.py b/logai/dataloader/data_model.py index 4be68d2..775a1ef 100755 --- a/logai/dataloader/data_model.py +++ b/logai/dataloader/data_model.py @@ -33,21 +33,19 @@ class LogRecordObject: whether each line is anomalous or not). :param _index: The indices of the log data. """ - - timestamp: pd.DataFrame = pd.DataFrame() - attributes: pd.DataFrame = pd.DataFrame() - resource: pd.DataFrame = pd.DataFrame() - trace_id: pd.DataFrame = pd.DataFrame() - span_id: pd.DataFrame = pd.DataFrame() - severity_text: pd.DataFrame = pd.DataFrame() - severity_number: pd.DataFrame = pd.DataFrame() - body: pd.DataFrame = pd.DataFrame() - labels: pd.DataFrame = pd.DataFrame() + timestamp: pd.DataFrame = field(default_factory=pd.DataFrame) + attributes: pd.DataFrame = field(default_factory=pd.DataFrame) + resource: pd.DataFrame = field(default_factory=pd.DataFrame) + trace_id: pd.DataFrame = field(default_factory=pd.DataFrame) + span_id: pd.DataFrame = field(default_factory=pd.DataFrame) + severity_text: pd.DataFrame = field(default_factory=pd.DataFrame) + severity_number: pd.DataFrame = field(default_factory=pd.DataFrame) + body: pd.DataFrame = field(default_factory=pd.DataFrame) + labels: pd.DataFrame = field(default_factory=pd.DataFrame) _index: np.array = field(init=False) def __post_init__(self): self._index = pd.DataFrame(self.body.index.values) - for field in self.__dataclass_fields__: field_content = getattr(self, field) if not field_content.empty: