Simplify Flores to have 2 configs instead of 2 classes.

PiperOrigin-RevId: 239313949

Author: adarob

tensorflow_datasets/translate/__init__.py

@@ -15,9 +15,8 @@
 
 """Translation datasets."""
 
+from tensorflow_datasets.translate.flores import Flores
 from tensorflow_datasets.translate.flores import FloresConfig
-from tensorflow_datasets.translate.flores_neen import FloresTranslateNeen
-from tensorflow_datasets.translate.flores_sien import FloresTranslateSien
 from tensorflow_datasets.translate.ted_hrlr import TedHrlrTranslate
 from tensorflow_datasets.translate.ted_multi import TedMultiTranslate
 from tensorflow_datasets.translate.wmt import WMTConfig

tensorflow_datasets/translate/flores.py

@@ -19,7 +19,6 @@
 from __future__ import division
 from __future__ import print_function
 
-import abc
 import collections
 
 import tensorflow as tf
@@ -92,13 +91,19 @@ def __init__(self,
     self.language_pair = language_pair
 
 
-class FloresTranslate(tfds.core.GeneratorBasedBuilder):
+class Flores(tfds.core.GeneratorBasedBuilder):
   """FLoRes machine translation dataset."""
 
-  @abc.abstractproperty
-  def is_abstract(self):
-    """Forces base class to be abstract."""
-    return True
+  BUILDER_CONFIGS = [
+      FloresConfig(
+          language_pair=("ne", "en"),
+          version="0.0.3",
+      ),
+      FloresConfig(
+          language_pair=("si", "en"),
+          version="0.0.3",
+      ),
+  ]
 
   def _info(self):
     source, target = self.builder_config.language_pair