Update on "[ExecuTorch][#10447] Extend PyBundledModule with extension.BundledModule"

#10447

# Context

This issue is a step of #9638.

In #9638, we want to have `extension.Module` as the single source of implementation in `pybindings`, which means that `pybindings.PyModule` should use `extension.Module` rather than its own `pybindings.Module`.

# Proposal

Now that we have `extension.BundledModule` ready, we want to test it out by having our existing `PyBundledModule` to extend it, and let `verify_result_with_bundled_expected_output` to use it, so that we can test out the whole thing with https://github.com/pytorch/executorch/blob/fb45e19055a92d2a91a4d4b7008e135232cbb14b/devtools/bundled_program/test/test_end2end.py

Differential Revision: [D73564127](https://our.internmc.facebook.com/intern/diff/D73564127/)

[ghstack-poisoned]

Author: zhenyan-zhang-meta

extension/module/test/resources/gen_bundled_program.py

@@ -1,14 +1,13 @@
-# (c) Meta Platforms, Inc. and affiliates. Confidential and proprietary.
-
 import torch
 
-from executorch.exir import to_edge_transform_and_lower
 from executorch.devtools import BundledProgram
 
 from executorch.devtools.bundled_program.config import MethodTestCase, MethodTestSuite
 from executorch.devtools.bundled_program.serialize import (
     serialize_from_bundled_program_to_flatbuffer,
 )
+
+from executorch.exir import to_edge_transform_and_lower
 from torch.export import export, export_for_training
 
 # Step 1: ExecuTorch Program Export
@@ -17,8 +16,8 @@ class SampleModel(torch.nn.Module):
 
     def __init__(self) -> None:
         super().__init__()
-        self.register_buffer('a', 3 * torch.ones(2, 2, dtype=torch.int32))
-        self.register_buffer('b', 2 * torch.ones(2, 2, dtype=torch.int32))
+        self.register_buffer("a", 3 * torch.ones(2, 2, dtype=torch.int32))
+        self.register_buffer("b", 2 * torch.ones(2, 2, dtype=torch.int32))
 
     def forward(self, x: torch.Tensor, q: torch.Tensor) -> torch.Tensor:
         z = x.clone()
@@ -76,7 +75,7 @@ def main() -> None:
             test_cases=[
                 MethodTestCase(
                     inputs=input,
-                    expected_outputs=(getattr(model, method_name)(*input), ),
+                    expected_outputs=(getattr(model, method_name)(*input),),
                 )
                 for input in inputs
             ],