Add base init on default obj

MLSTRUCT · Aug 14, 2024 · 971194c · 971194c
1 parent c279910
commit 971194c
Show file tree

Hide file tree

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

diff --git a/MLStructFP/db/image/_base.py b/MLStructFP/db/image/_base.py
@@ -141,3 +141,9 @@ def get_file_id(self, filename) -> int:
         if filename not in self._names:
             raise ValueError(f'File <{filename}> have not been processed yet')
         return self._names.index(filename)
+
+    def init(self) -> 'BaseImage':
+        """
+        Init the object.
+        """
+        return self