suppress errors in executorch

Differential Revision: D47423118

fbshipit-source-id: 79088266b904e67a8db8ae359bb261a1ab78c306

Author: generatedunixname89002005307016

backends/test/test_backends.py

@@ -279,6 +279,7 @@ def forward(self, a, x, b):
 
         # pyre-ignore[16]: Module `executorch.pybindings` has no attribute `portable`.
         executorch_module = _load_for_executorch_from_buffer(buff)
+        # pyre-fixme[16]: Module `pytree` has no attribute `tree_flatten`.
         inputs_flattened, _ = tree_flatten(model_inputs)
         model_output = executorch_module.run_method("forward", tuple(inputs_flattened))
         ref_output = add_mul_module(*model_inputs)
@@ -857,6 +858,7 @@ def forward(self, a, x, b):
 
         # pyre-ignore[16]: Module `executorch.pybindings` has no attribute `portable`.
         executorch_module = _load_for_executorch_from_buffer(executorch_prog.buffer)
+        # pyre-fixme[16]: Module `pytree` has no attribute `tree_flatten`.
         inputs_flattened, _ = tree_flatten(inputs)
         model_output = executorch_module.run_method("forward", tuple(inputs_flattened))
         ref_output = m(*inputs)

backends/vulkan/test/test_vulkan_delegate.py

@@ -86,6 +86,7 @@ def forward(self, *args):
         # Test the model with executor
         # pyre-ignore
         executorch_module = _load_for_executorch_from_buffer(buffer)
+        # pyre-fixme[16]: Module `pytree` has no attribute `tree_flatten`.
         inputs_flattened, _ = tree_flatten(sample_inputs)
 
         model_output = executorch_module.run_method("forward", tuple(inputs_flattened))

backends/xnnpack/test/test_xnnpack_utils.py

@@ -220,6 +220,7 @@ def forward(self, *args):
         # Test the model with executor
         # pyre-ignore
         executorch_module = _load_for_executorch_from_buffer(buffer)
+        # pyre-fixme[16]: Module `pytree` has no attribute `tree_flatten`.
         inputs_flattened, _ = tree_flatten(sample_inputs)
 
         model_output = executorch_module.run_method("forward", tuple(inputs_flattened))
@@ -422,6 +423,7 @@ def forward(self, x):
         buffer = serialize_to_flatbuffer(program)
         # pyre-ignore
         executorch_module = _load_for_executorch_from_buffer(buffer)
+        # pyre-fixme[16]: Module `pytree` has no attribute `tree_flatten`.
         inputs_flattened, _ = tree_flatten(example_inputs)
 
         model_output = executorch_module.run_method("forward", tuple(inputs_flattened))

bundled_program/config.py

@@ -151,6 +151,7 @@ def _tree_flatten(unflatten_data: Any) -> List[ConfigValue]:
         Returns:
             flatten_data: Flatten data with legal type.
         """
+        # pyre-fixme[16]: Module `pytree` has no attribute `tree_flatten`.
         flatten_data, _ = tree_flatten(unflatten_data)
 
         for data in flatten_data:

bundled_program/tests/test_config.py

@@ -128,6 +128,7 @@ def test_create_config_from_eager_model(self) -> None:
 
                 model_outputs = eager_model(*ri)
                 if isinstance(model_outputs, get_args(DataContainer)):
+                    # pyre-fixme[16]: Module `pytree` has no attribute `tree_flatten`.
                     flatten_eager_model_outputs = tree_flatten(model_outputs)
                 else:
                     flatten_eager_model_outputs = [

exir/emit/_emit_program.py

@@ -33,6 +33,7 @@ def _emit_prim_getters(prim_getters: Dict[str, Any]) -> List[ExecutionPlan]:
     plans = []
     # flatten any structures
     for method, vals in prim_getters.items():
+        # pyre-fixme[16]: Module `pytree` has no attribute `tree_flatten`.
         flattened_output, spec = ex_pytree.tree_flatten(vals)
         spec = spec.to_str()
         chain = Chain(

exir/emit/_emitter.py

@@ -1258,6 +1258,7 @@ def create_container_str(spec: pytree.TreeSpec) -> str:
             # pyre-fixme[16]: `TreeSpec` has no attribute `num_leaves`.
             dummy_leaves = [0] * spec.num_leaves
             tree = torch.utils._pytree.tree_unflatten(dummy_leaves, spec)
+            # pyre-fixme[16]: Module `pytree` has no attribute `tree_flatten`.
             _, tree = ex_pytree.tree_flatten(tree)
             return tree.to_str()
 

exir/tracer.py

@@ -325,11 +325,14 @@ def __torch_dispatch__(
 
         tensor_args = [
             x
+            # pyre-fixme[16]: Module `pytree` has no attribute `tree_flatten`.
             for x in ex_pytree.tree_flatten(args)[0] + ex_pytree.tree_flatten(kwargs)[0]
             if isinstance(x, torch.Tensor)
         ]
 
+        # pyre-fixme[16]: Module `pytree` has no attribute `tree_map`.
         proxy_args = ex_pytree.tree_map(unwrap_proxy, args)
+        # pyre-fixme[16]: Module `pytree` has no attribute `tree_map`.
         proxy_kwargs = ex_pytree.tree_map(unwrap_proxy, kwargs)
 
         # Get the output of the function

exir/verification/interpreter.py

@@ -362,6 +362,7 @@ def run(self, *raw_args: torch.Tensor) -> PyTree:
         Outputs after completing all computations
         """
 
+        # pyre-fixme[16]: Module `pytree` has no attribute `tree_flatten`.
         args, pytree = ex_pytree.tree_flatten(raw_args)
 
         if pytree.to_str() != self.container_metatype.encoded_inp_str:
@@ -412,5 +413,7 @@ def run(self, *raw_args: torch.Tensor) -> PyTree:
             ip += 1
 
         ret = [self._value_list[i] for i in self.execution_plan.outputs]
+        # pyre-fixme[16]: Module `pytree` has no attribute `from_str`.
         treespec = ex_pytree.from_str(self.container_metatype.encoded_out_str)
+        # pyre-fixme[16]: Module `pytree` has no attribute `tree_unflatten`.
         return ex_pytree.tree_unflatten(ret, treespec)

profiler/test/test_profiler_e2e.py

@@ -45,6 +45,7 @@ def setUpClass(cls) -> None:
         # pyre-ignore: Undefined attribute [16]: Module `executorch.pybindings` has no attribute `portable`.
         cls.module = _load_for_executorch_from_buffer(cls.flatbuff_without_stacktrace)
 
+        # pyre-fixme[16]: Module `pytree` has no attribute `tree_flatten`.
         cls.inputs_flattened, _ = tree_flatten(inputs)
         cls.module.run_method("forward", tuple(cls.inputs_flattened))
         # pyre-ignore: Undefined attribute [16]: Module `executorch.pybindings` has no attribute `portable`.

pytree/test/test.py

@@ -17,6 +17,7 @@
 
 # pyre-fixme[11]: Annotation `TreeSpec` is not defined as a type.
 def _spec(o: Any) -> TreeSpec:
+    # pyre-fixme[16]: Module `pytree` has no attribute `tree_flatten`.
     _, spec = tree_flatten(o)
     return spec
 

test/end2end/test_end2end.py

@@ -562,8 +562,10 @@ def wrapper(self: unittest.TestCase) -> None:
                     expected = getattr(module.eager_module, method)(*inputs)
 
                 if do_tree_flatten:
+                    # pyre-fixme[16]: Module `pytree` has no attribute `tree_flatten`.
                     flatten_inputs, inputs_spec = pytree.tree_flatten(*inputs)
                     executorch_result = executorch_module.forward([*flatten_inputs])
+                    # pyre-fixme[16]: Module `pytree` has no attribute `TreeSpec`.
                     executorch_result_unflatten = pytree.TreeSpec.from_str(
                         program.execution_plan[0].container_meta_type.encoded_out_str
                     ).tree_unflatten(executorch_result)