Reestablish caching of testdata_yaml[f]

RagnarGrootKoerkamp · Mar 8, 2024 · 2b3d8af · 2b3d8af
1 parent dd6de68
commit 2b3d8af
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/bin/problem.py b/bin/problem.py
@@ -202,7 +202,11 @@ def get_testdata_yaml(p, path, key, name=None) -> str | None:
             f = dir / 'testdata.yaml'
 
             if f.is_file():
-                flags = read_yaml(f, plain=True)
+                if f in p._testdata_yamls:
+                    flags = p._testdata_yamls[f]
+                else:
+                    p._testdata_yamls[f] = flags = read_yaml(f, plain=True)
+
 
                 # Validate the flags
                 for key in flags:
@@ -226,10 +230,6 @@ def get_testdata_yaml(p, path, key, name=None) -> str | None:
                             warn(f'Unknown testdata.yaml key: {key}')
 
                 # Store testdata.yaml files in a cache.
-                if f not in p._testdata_yamls:
-                    p._testdata_yamls[f] = read_yaml(f, plain=True)
-
-                # flags = p._testdata_yamls[f]
                 if key in flags:
                     if isinstance(flags[key], str):
                         return flags[key]