Skip to content

Commit

Permalink
Update training module to have super class methods accessible (#7082)
Browse files Browse the repository at this point in the history
Update training module to have super class methods accessible (#7082)

Summary:

This is needed so the training module has access to non-training methods (e.g. constant string return methods).

Reviewed By: JacobSzwejbka

Differential Revision: D66419247
  • Loading branch information
georgehong authored Nov 27, 2024
1 parent 1c7d94e commit 9b29b4b
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion extension/training/module/training_module.h
Original file line number Diff line number Diff line change
Expand Up @@ -26,7 +26,8 @@ namespace training {
* A facade class for loading programs for on-device training and executing
* methods within them.
*/
class ET_EXPERIMENTAL TrainingModule final : executorch::extension::Module {
class ET_EXPERIMENTAL TrainingModule final
: public executorch::extension::Module {
public:
explicit TrainingModule(
std::unique_ptr<runtime::DataLoader> data_loader,
Expand Down

0 comments on commit 9b29b4b

Please sign in to comment.