Use setter instead of set_classes.

tryolabs · Aug 24, 2018 · 41b5960 · 41b5960
1 parent b5f932f
commit 41b5960
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/luminoth/tools/dataset/readers/object_detection/object_detection_reader.py b/luminoth/tools/dataset/readers/object_detection/object_detection_reader.py
@@ -57,10 +57,14 @@ def classes(self):
             self._classes = self.get_classes()
         return self._classes
 
+    @classes.setter
+    def classes(self, classes):
+        self._classes = classes
+
     @abc.abstractmethod
     def get_total(self):
         """
-        Returns the total amount of records in the dataset.
+        Returns the total number of records in the dataset.
         """
 
     @abc.abstractmethod
@@ -129,6 +133,3 @@ def iterate(self):
                     xmax (int): x value for bottom-right point.
                     ymax (int): y value for bottom-right point.
         """
-
-    def set_classes(self, classes):
-        self._classes = classes
diff --git a/luminoth/tools/dataset/transform.py b/luminoth/tools/dataset/transform.py
@@ -75,7 +75,7 @@ def transform(dataset_reader, data_dir, output_dir, splits, only_classes,
                 classes = split_reader.classes
             else:
                 # Overwrite classes after first split for consistency.
-                split_reader.set_classes(classes)
+                split_reader.classes = classes
 
             # We assume we are saving object detection objects, but it should
             # be easy to modify once we have different types of objects.