Rm all_problems from basic_test and autoencoders_test

PiperOrigin-RevId: 193612418

Author: Ryan Sepassi

tensor2tensor/models/basic_test.py

@@ -23,7 +23,7 @@
 
 import numpy as np
 
-from tensor2tensor.data_generators import all_problems  # pylint: disable=unused-import
+from tensor2tensor.data_generators import mnist  # pylint: disable=unused-import
 from tensor2tensor.models import basic
 from tensor2tensor.utils import trainer_lib
 

tensor2tensor/models/research/autoencoders_test.py

@@ -23,7 +23,7 @@
 
 import numpy as np
 
-from tensor2tensor.data_generators import all_problems  # pylint: disable=unused-import
+from tensor2tensor.data_generators import mnist  # pylint: disable=unused-import
 from tensor2tensor.models.research import autoencoders  # pylint: disable=unused-import
 from tensor2tensor.utils import registry
 from tensor2tensor.utils import trainer_lib

tensor2tensor/utils/t2t_model.py

@@ -32,6 +32,7 @@
 from tensor2tensor.data_generators import text_encoder
 from tensor2tensor.data_generators.problem import problem_hparams_to_features
 from tensor2tensor.layers import common_layers
+from tensor2tensor.layers import modalities  # pylint: disable=unused-import
 from tensor2tensor.utils import beam_search
 from tensor2tensor.utils import decoding
 from tensor2tensor.utils import expert_utils as eu