Как проверить, работает ли мой код в среде SLURM?

Question

Я работаю над рабочим процессом с использованием Snakemake, который должен быть переносимым на любую систему на базе Linux, но в основном разработан для работы на hp c с использованием SLURM. Для оптимизации при использовании SLURM я хотел бы проверить, работает ли код в среде SLURM, а затем немного изменить задачи, чтобы улучшить управление ресурсами.

Моя первая идея заключалась в том, чтобы просто попытаться разрешить переменную среды $ SLURM_JOB_ID через os.path.expandvars, но, на мой взгляд, это немного грязно, так есть ли чистый способ просто проверить среду?

Answer 1

Проверка переменной окружения - это путь к go. В Python вы бы сделали это так:

import os
if "SLURM_JOB_ID" in os.environ:
    print("Running in Slurm")
else:
    print("NOT running in Slurm")