schrodinger.application.matsci.flywheel.base_model module

Base model class for flywheel model components.

class schrodinger.application.matsci.flywheel.base_model.BaseModel(device='cpu', logger=None)

Bases: object

Base class for lazily-loaded model components.

Provides shared infrastructure for device management, logging, and ONNX provider selection.

__init__(device='cpu', logger=None)
Parameters:

  • device (str) – Compute device (‘cpu’, ‘cuda’, ‘mps’)

  • logger – Logger instance for debug output

getOnnxProviders()

Get ONNX Runtime execution providers for device.

ONNX Runtime has no MPS provider; MPS falls back to CPU.

Return type:

list(str)

Returns:

Ordered list of execution providers

logMsg(msg)

Log a message via textlogger if a logger is set.

Parameters:

msg (str) – Message to log