Fix failing test

tskit-dev · Jul 16, 2024 · 0593277 · 0593277
1 parent 2d0a384
commit 0593277
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/test_formats.py b/tests/test_formats.py
@@ -252,7 +252,10 @@ def test_too_small_max_file_size_add(self):
                         genotypes=np.zeros(base_size, dtype=np.int8),
                     )
             # Work around https://github.com/tskit-dev/tsinfer/issues/201
-            small_sample_file.data.store.close()
+            try:
+                small_sample_file.data.store.close()
+            except UnboundLocalError:
+                pass
 
     def test_acceptable_max_file_size(self):
         with tempfile.TemporaryDirectory(prefix="tsinf_format_test") as tempdir: