Proposed fix for TF example now running on safetensors. (#23208)

* Proposed fix for TF example now running on safetensors. * Adding more warnings and returning keys. * Trigger CI * Trigger CI --------- Co-authored-by: Sylvain Gugger <Sylvain.gugger@gmail.com>
2023-05-09 19:04:27 +02:00
parent b4d4d6fe87
commit c34a525d2f
3 changed files with 34 additions and 3 deletions
--- a/examples/tensorflow/test_tensorflow_examples.py
+++ b/examples/tensorflow/test_tensorflow_examples.py
@@ -297,7 +297,6 @@ class ExamplesTests(TestCasePlus):
            result = get_results(tmp_dir)
            self.assertGreaterEqual(result["bleu"], 30)

-    @skip("Fix me Matt")
    def test_run_image_classification(self):
        tmp_dir = self.get_auto_remove_tmp_dir()
        testargs = f"""