loki2.cl.utils

Utilities for interacting with Loki2 outputs and .pt/.json files.

This module provides utilities for loading and processing Loki2 model outputs, including embeddings and metadata files.

Module Contents

loki2.cl.utils.setup_dummy_modules() → None

Sets up dummy modules for ‘loki2’ and its submodules.

This is required to simulate the module structure that the torch model expects when unpickling objects that reference e.g. ‘loki2.data.dataclass.cell_graph’.

loki2.cl.utils.load_and_print_tensor(file_path: str) → Any

Load a PyTorch object from a given file path.

Parameters:

file_path – Path to the .pt file.

Returns:

The loaded Python object (typically a tensor or model state).

Return type:

Any