Enhance source pixels copy

MLSTRUCT · Mar 31, 2024 · e09a0cf · e09a0cf
1 parent d74dbd8
commit e09a0cf
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/MLStructFP/__init__.py b/MLStructFP/__init__.py
@@ -9,7 +9,7 @@
 __description__ = 'Machine learning structural floor plan dataset'
 __keywords__ = ['ml', 'ai', 'dataset', 'calc', 'matrix analysis', 'cnn', 'structural analysis', 'structural design']
 __email__ = 'pablo@ppizarror.com'
-__version__ = '0.5.5'
+__version__ = '0.5.6'
 
 # URL
 __url__ = 'https://github.com/MLSTRUCT/MLSTRUCT-FP'

diff --git a/MLStructFP/db/image/_rect_photo.py b/MLStructFP/db/image/_rect_photo.py
@@ -311,7 +311,7 @@ def _parse_image(self, ip: str, mutator_scale_x: float = 1, mutator_scale_y: flo
             sy = int(math.ceil(abs(sy * mutator_scale_y)))
             pixels = cv2.resize(pixels, (sx, sy))
 
-        source_pixels: 'np.ndarray' = pixels
+        source_pixels: Optional['np.ndarray'] = None
         if verbose:
             source_pixels = pixels.copy()
         cy, cx = pixels.shape[:2]  # Source center pixel