Merge pull request #524 from colibrisson/feature/reading_order

Fix small bugs in the Feature/reading order branch
mittagessen · Jul 17, 2023 · 584ed68 · 584ed68
2 parents 20a869a + 8f34a9a
commit 584ed68
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/kraken/containers.py b/kraken/containers.py
@@ -29,6 +29,9 @@ class BaselineLine:
     split: Optional[Literal['train', 'validation', 'test']] = None
     regions: Optional[List[str]] = None
 
+    def __getitem__(self, item):
+        return getattr(self, item)
+
 @dataclass
 class BBoxLine:
     """

diff --git a/kraken/lib/train.py b/kraken/lib/train.py
@@ -33,7 +33,7 @@
 from pytorch_lightning.callbacks import Callback, EarlyStopping, BaseFinetuning, LearningRateMonitor
 
 from kraken.lib import models, vgsl, default_specs, progress
-from kraken.lib.xml import preparse_xml_data
+# from kraken.lib.xml import preparse_xml_data
 from kraken.lib.util import make_printable
 from kraken.lib.codec import PytorchCodec
 from kraken.lib.dataset import (ArrowIPCRecognitionDataset, BaselineSet,