Data Manager

Module for the Data Manager factory class.

class _DataManager(**kwargs)

Bases: object

Factory class for data manager implementations.

This class dispatches object creation to either _TensorDataManager or _GraphDataManager depending on the types of the provided keyword arguments.

Create the appropriate data manager implementation based on the provided keyword arguments.

If all values in kwargs are instances of torch.Tensor, LabelTensor, or BaseEquation, an instance of _TensorDataManager is created. Otherwise, an instance of _GraphDataManager is created.

Parameters:

kwargs (dict) – The keyword arguments for the data manager.

Returns:

A concrete data manager instance.

Return type:

_TensorDataManager | _GraphDataManager