Way to resolving OS.fork issue?

[cwoolfo1]

I am currently using jax and I wrote unit tests for the code that I am writing. I just recently discovered a cool library that can be used to run unit tests in parallel ( "https://github.com/craigahobbs/unittest-parallel" ). However, I am getting some warnings about the fact this library using os.fork()

/usr/local/miniconda3/lib/python3.12/multiprocessing/popen_fork.py:66: RuntimeWarning: os.fork() was called. os.fork() is incompatible with multithreaded code, and JAX is multithreaded, so this will likely lead to a deadlock.

I would really like to use this library with my jax code, especially as I am starting to write more complex unit tests for my code. Is there a way to ensure my code will work with this library?