Update gdl module to cleanup base interf name

plstcharles · Nov 6, 2019 · 6281f1e · 6281f1e
1 parent a369448
commit 6281f1e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/thelper/data/geo/gdl.py b/thelper/data/geo/gdl.py
@@ -16,12 +16,12 @@
 import numpy as np
 
 import thelper.nn.coordconv
-from thelper.data.parsers import SegmentationDataset
+from thelper.data.parsers import SegmentationDataset as BaseSegmentationDataset
 
 logger = logging.getLogger(__name__)
 
 
-class SegmentationDataset(SegmentationDataset):
+class SegmentationDataset(BaseSegmentationDataset):
     """Semantic segmentation dataset interface for GDL-based HDF5 parsing."""
 
     def __init__(self, class_names, work_folder, dataset_type, max_sample_count=None,