Refactor the function get_results (#20999)

examples/flax/test_flax_examples.py

@@ -60,14 +60,11 @@ def get_setup_file():
 
 
 def get_results(output_dir, split="eval"):
-    results = {}
     path = os.path.join(output_dir, f"{split}_results.json")
     if os.path.exists(path):
         with open(path, "r") as f:
-            results = json.load(f)
-    else:
+            return json.load(f)
     raise ValueError(f"can't find {path}")
-    return results
 
 
 stream_handler = logging.StreamHandler(sys.stdout)