LoadTrainedModelUnit
Rebuild a trained model from the run that produced it.
Reads the model component name and the artifact path off the Run row
rather than taking them as configuration, so the model that is restored is
always the one that run actually saved.
load is invoked on the model class, not on an instance, which is what
every concrete model in this codebase expects: they all declare it as a
staticmethod or a classmethod that rebuilds the object from the
file. ExplainerJob instead instantiates the class before calling
load; that extra step has no effect for those models, and preserving
the difference is why LoadRunModelUnit exists separately instead of
this unit growing a flag.
Parameters
- run_id : integer, default=
1 - Identifier of the run whose trained model is restored. Both the model component and the artifact path come from that row.
Methods
execute(self, ctx: DashAI.back.units.context.ExecutionContext) -> None
LoadTrainedModelUnitDo the unit's work, reading from and writing to the context.
Parameters
- ctx : ExecutionContext
- The shared execution context.
get_credential(self, name: str)
ConfigObjectResolve a registered credential component by name.
Parameters
- name : str
- Credential component class name (e.g. "HuggingFaceCredential").
Returns
- BaseCredential
- An instance of the requested credential component.
get_schema(cls) -> dict
ConfigObjectGenerates the component related Json Schema.
Returns
- dict
- Dictionary representing the Json Schema of the component.
validate(self, ctx: DashAI.back.units.context.ExecutionContext) -> None
BaseUnitCheck preconditions without executing the unit.
Parameters
- ctx : ExecutionContext
- The shared execution context.
validate_and_transform(self, raw_data: dict) -> dict
ConfigObjectIt takes the data given by the user to initialize the model and returns it with all the objects that the model needs to work.
Parameters
- raw_data : dict
- A dictionary with the data provided by the user to initialize the model.
Returns
- dict
- A validated dictionary with the necessary objects.