paroto.models.registry#

Model registry for managing available models.

Classes#

ModelRegistry

Registry for managing available physics models.

Module Contents#

class paroto.models.registry.ModelRegistry#

Registry for managing available physics models.

This class maintains a registry of all available models for different physical phenomena, allowing users to select models by name.

classmethod register(category, model_class)#

Register a model class.

Parameters:

  • category (str) – Model category (e.g., ‘breakdown_voltage’, ‘ablation’)

  • model_class (Type[BaseModel]) – Model class to register

classmethod get_model(category, model_name)#

Get a model class by category and name.

Parameters:

  • category (str) – Model category

  • model_name (str) – Model name

Returns:

model_class – Model class if found, None otherwise

Return type:

Type[BaseModel] or None

classmethod list_models(category)#

List all models in a category.

Parameters:

category (str) – Model category

Returns:

models – Dictionary of model names to model classes

Return type:

dict

classmethod list_categories()#

List all available model categories.

Returns:

categories – List of category names

Return type:

list

classmethod create_model(category, model_name, parameters=None)#

Create a model instance.

Parameters:

  • category (str) – Model category

  • model_name (str) – Model name

  • parameters (dict, optional) – Model parameters

Returns:

model – Model instance

Return type:

BaseModel

Raises:

ValueError – If model not found