armi.context module

Module containing global constants that reflect the executing context of ARMI.

ARMI’s global state information: operating system information, environment data, user data, memory parallelism, temporary storage locations, and if operational mode (interactive, gui, or batch).

class armi.context.Mode(value)

Bases: Enum

Mode represents different run types possible in ARMI.

The modes can be Batch, Interactive, or GUI. Mode is generally auto-detected based on your terminal. It can also be set in various CLI entry points. Each entry point has a --batch command line argument that can force Batch mode.

BATCH = 1

INTERACTIVE = 2

GUI = 4

classmethod setMode(mode)

Set the run mode of the current ARMI case.

armi.context.activateLocalFastPath() → None

Specify a local temp directory to be the fast path.

FAST_PATH is often a local hard drive on a cluster node. It should be a high-performance scratch space. Different processors on the same node should have different fast paths.

Notes

This path will be obliterated when the job ends.

This path is set at import time, so if a series of unit tests come through that instantiate one operator after the other, the path will already exist the second time. The directory is created in the Operator constructor.

armi.context.getFastPath() → str

Callable to get the current FAST_PATH.

Notes

This exists because it’s dangerous to use FAST_PATH directly. as it can change between import and runtime.

armi.context.cleanTempDirs(olderThanDays=None)

Clean up temporary files after a run.

Some Windows HPC systems send a SIGBREAK signal when the user cancels a job, which is NOT handled by atexit. Notably, SIGBREAK does not exist outside Windows. For the SIGBREAK signal to work with a Windows HPC, the TaskCancelGracePeriod option must be configured to be non- zero. This sets the period between SIGBREAK and SIGTERM/SIGINT. To do cleanups in this case, we must use the signal module. Actually, even then it does not work because MS mpiexec does not pass signals through.

Parameters:

olderThanDays (int, optional) – If provided, deletes other ARMI directories if they are older than the requested time.

armi.context.cleanAllArmiTempDirs(olderThanDays: int) → None

Delete all ARMI-related files from other unrelated runs after olderThanDays days (in case this failed on earlier runs).

This is a useful utility in HPC environments when some runs crash sometimes.

Warning

This will break any concurrent runs that are still running.

armi.context.disconnectAllHdfDBs() → None

Forcibly disconnect all instances of HdfDB objects.

Notes

This is a hack to help ARMI exit gracefully when the garbage collector and h5py have issues destroying objects. The root cause for why this was having issues was never identified. It appears that when several HDF5 files are open in the same run (e.g. when calling armi.init() multiple times from a post-processing script), when these h5py File objects were closed, the garbage collector would raise an exception related to the repr’ing the object. We get around this by using the garbage collector to manually disconnect all open HdfDBs.