i%SSKJr SSKJrJrJrJrJr SSKrSSK J r SSK J r \\R\\\4r\ \S'\\r\ \S'\\Rr\ \S'\\\\4r\ \S '"S S 5r\"S S 55rg)) dataclass)get_argsListOptionalSequenceUnionN) tree_flatten) TypeAlias ConfigValueMethodInputTypeMethodOutputType DataContainercR\rSrSrSrS S\S\\SS4SjjrS\ S\ \ 4S jr S r g) MethodTestCase1zTest case with inputs and expected outputs The expected_outputs are optional and only required if the user wants to verify model outputs after execution. Ninputsexpected_outputsreturncrURU5Ul/UlUbURU5Ulgg)aSingle test case for verifying specific method Args: inputs: All inputs required by eager_model with specific inference method for one-time execution. It is worth mentioning that, although both bundled program and ET runtime apis support setting input other than `torch.tensor` type, only the input in `torch.tensor` type will be actually updated in the method, and the rest of the inputs will just do a sanity check if they match the default value in method. expected_outputs: Expected output of given input for verification. It can be None if user only wants to use the test case for profiling. Returns: self N)_flatten_and_sanity_checkrr)selfrrs i/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/executorch/devtools/bundled_program/config.py__init__MethodTestCase.__init__6s=**.)G)G)O 35  '$($B$BCS$TD ! (unflatten_datac[U5up#UHqn[U[[55(d SR U[ U555e[U[ S55(dM]SR U55e U$)zFlat the given data and check its legality Args: unflatten_data: Data needs to be flatten. Returns: flatten_data: Flatten data with legal type. z4The type of input {} with type {} is not supported. Nz)The input {} should not be in null type. )r isinstancerr formattype)rr flatten_data_datas rr(MethodTestCase._flatten_and_sanity_checkQs'~6  D% GMMd4j  "T  I<BB4H I!r)rr)N)__name__ __module__ __qualname____firstlineno____doc__r rr rrrr r__static_attributes__rrrr1sQ8<UU##34U  U6+ k rrc4\rSrSr%Sr\\S'\\\S'Sr g)MethodTestSuitenzAll test info related to verify method Attributes: method_name: Name of the method to be verified. test_cases: All test cases for verifying the method. method_name test_casesr+N) r%r&r'r(r)str__annotations__rrr*r+rrr-r-ns((rr-) dataclassesrtypingrrrrrtorchtorch.utils._pytreer typing_extensionsr Tensorintboolfloatr r2r r listtupledictrrr-r+rrr?s"<< ,' LL   Y&k22'u||4)4!ud!23 y3::z  ) )  )r