Nested Submodels

ricardoV94 · March 30, 2023, 7:18am

You could implement an extend_model function, which would update the parent model attributes with the new variables.

These are the things you need to populate:

pymc-devs/pymc/blob/ae9fcac57415fed0d4b121b0a6b88aa715be7295/pymc/model.py#L575-L586


      
          self.named_vars = treedict()
          self.named_vars_to_dims = treedict()
          self.values_to_rvs = treedict()
          self.rvs_to_values = treedict()
          self.rvs_to_transforms = treedict()
          self.rvs_to_initial_values = treedict()
          self.free_RVs = treelist()
          self.observed_RVs = treelist()
          self.deterministics = treelist()
          self.potentials = treelist()
          self._coords = {}
          self._dim_lengths = {}

You can add new items to treedicts (just a subclass of dict) and append new values to treelists (just a subclass of lists)

Topic		Replies	Views
Composing models Questions	2	984	August 6, 2017
How to extend an existing model Questions	3	721	December 22, 2019
A meteorological model writed by Fortran Questions development	1	658	September 24, 2017
Zoo module in 3.2 for storing predefined model sketches Development	5	704	June 27, 2017
Pass a Model defined in a config file to a later file Questions	2	483	September 9, 2019

Nested Submodels

Related topics