PY3 migration. Autograph compatible.

PiperOrigin-RevId: 287619398

official/modeling/hyperparams/params_dict.py

@@ -111,10 +111,10 @@ class ParamsDict(object):
       the value of the key.
 
     Raises:
-      KeyError: if k is not defined in the ParamsDict.
+      AttributeError: if k is not defined in the ParamsDict.
     """
     if k not in self.__dict__.keys():
-      raise KeyError('The key `{}` does not exist. '.format(k))
+      raise AttributeError('The key `{}` does not exist. '.format(k))
     return self.__dict__[k]
 
   def __contains__(self, key):

official/modeling/hyperparams/params_dict_test.py

@@ -27,7 +27,7 @@ class ParamsDictTest(tf.test.TestCase):
 
   def test_init_from_an_empty_dict(self):
     params = params_dict.ParamsDict()
-    with self.assertRaises(KeyError):
+    with self.assertRaises(AttributeError):
       _ = params.a
 
     with self.assertRaises(KeyError):

official/vision/detection/dataloader/input_reader.py

@@ -64,7 +64,7 @@ class InputFn(object):
         self._input_sharding = params.train.input_sharding
       else:
         self._input_sharding = params.eval.input_sharding
-    except KeyError:
+    except AttributeError:
       pass
 
   def __call__(self, ctx=None, batch_size: int = None):

official/vision/detection/modeling/base_model.py

@@ -39,7 +39,7 @@ class OptimizerFactory(object):
       nesterov = False
       try:
         nesterov = params.nesterov
-      except KeyError:
+      except AttributeError:
         pass
       self._optimizer = functools.partial(
           tf.keras.optimizers.SGD,