diff --git a/extension/pybindings/portable_lib.py b/extension/pybindings/portable_lib.py index d8813539b45..6ae74a60c96 100644 --- a/extension/pybindings/portable_lib.py +++ b/extension/pybindings/portable_lib.py @@ -88,6 +88,7 @@ _load_for_executorch, # noqa: F401 _load_for_executorch_from_buffer, # noqa: F401 _load_for_executorch_from_bundled_program, # noqa: F401 + _load_for_executorch_from_data_loader, # noqa: F401 _load_program, # noqa: F401 _load_program_from_buffer, # noqa: F401 _reset_profile_results, # noqa: F401 @@ -98,6 +99,7 @@ ExecuTorchModule, # noqa: F401 ExecuTorchProgram, # noqa: F401 MethodMeta, # noqa: F401 + TensorInfo, # noqa: F401 Verification, # noqa: F401 )