use field to set default empty array (#390)

Unstructured-IO · Oct 10, 2024 · e411862 · e411862
1 parent 7122199
commit e411862
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 4 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,3 +1,7 @@
+## 0.7.39
+
+* fix: Correctly assign mutable default value to variable in `LayoutElements` class
+
 ## 0.7.38
 
 * fix: Correctly assign mutable default value to variable in `TextRegions` class

diff --git a/unstructured_inference/__version__.py b/unstructured_inference/__version__.py
@@ -1 +1 @@
-__version__ = "0.7.38"  # pragma: no cover
+__version__ = "0.7.39"  # pragma: no cover
diff --git a/unstructured_inference/inference/layoutelement.py b/unstructured_inference/inference/layoutelement.py
@@ -1,6 +1,6 @@
 from __future__ import annotations
 
-from dataclasses import dataclass
+from dataclasses import dataclass, field
 from typing import Any, Collection, Iterable, List, Optional
 
 import numpy as np
@@ -30,8 +30,8 @@
 
 @dataclass
 class LayoutElements(TextRegions):
-    element_probs: np.ndarray = np.array([])
-    element_class_ids: np.ndarray = np.array([])
+    element_probs: np.ndarray = field(default_factory=lambda: np.array([]))
+    element_class_ids: np.ndarray = field(default_factory=lambda: np.array([]))
     element_class_id_map: dict[int, str] | None = None
 
     def __post_init__(self):