pyhazards.utils package

Submodules

pyhazards.utils.hardware module

pyhazards.utils.hardware.auto_device(prefer=None)[source]

Choose a device automatically. Respects PYHAZARDS_DEVICE and prefer flag.

Return type:

device

pyhazards.utils.hardware.get_device()[source]
Return type:

device

pyhazards.utils.hardware.num_devices()[source]
Return type:

int

pyhazards.utils.hardware.set_device(device_str)[source]
Return type:

None

pyhazards.utils.common module

pyhazards.utils.common.get_logger(name=None)[source]
Return type:

Logger

pyhazards.utils.common.seed_all(seed=42, deterministic=False)[source]
Return type:

None

Module contents

pyhazards.utils.auto_device(prefer=None)[source]

Choose a device automatically. Respects PYHAZARDS_DEVICE and prefer flag.

Return type:

device

pyhazards.utils.get_device()[source]
Return type:

device

pyhazards.utils.get_logger(name=None)[source]
Return type:

Logger

pyhazards.utils.num_devices()[source]
Return type:

int

pyhazards.utils.seed_all(seed=42, deterministic=False)[source]
Return type:

None

pyhazards.utils.set_device(device_str)[source]
Return type:

None