def METHOD_NAME(self):
    with self.assertRaisesRegex(
        EnvironmentError, "bert-base is not a local folder and is not a valid model identifier"
    ):
        _ = FlaxAutoModel.from_pretrained("bert-base")