pytest fix (#642)

mosaicml · Mar 25, 2024 · 9781727 · 9781727
1 parent 508346d
commit 9781727
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -68,7 +68,7 @@
     'docformatter>=1.4',
     'jupyter==1.0.0',
     'pre-commit>=2.18.1,<4',
-    'pytest==7.4.4',
+    'pytest==8.1.1',
     'pytest_codeblocks==0.17.0',
     'pytest-cov>=4,<5',
     'toml==0.10.2',

diff --git a/tests/test_util.py b/tests/test_util.py
@@ -111,6 +111,9 @@ def test_number_abbrev_to_int_Exception():
 
 
 def test_clean_stale_shared_memory():
+    # Clean up the stale shared memory
+    clean_stale_shared_memory()
+
     # Create a leaked shared memory
     name = _get_path(0, RESUME)
     _ = BuiltinSharedMemory(name, True, 64)