File tree Expand file tree Collapse file tree 2 files changed +9
-1
lines changed Expand file tree Collapse file tree 2 files changed +9
-1
lines changed Original file line number Diff line number Diff line change 1313try :
1414 from executorch .extension .pybindings .portable_lib import ( # @manual
1515 _load_for_executorch_from_buffer ,
16+ Verification ,
1617 )
1718except ImportError as e :
1819 logger .warning (f"{ e = } " )
@@ -39,7 +40,9 @@ def graph_module(self) -> None:
3940
4041 def run_artifact (self , inputs ):
4142 inputs_flattened , _ = tree_flatten (inputs )
42- executorch_module = _load_for_executorch_from_buffer (self .buffer )
43+ executorch_module = _load_for_executorch_from_buffer (
44+ self .buffer , program_verification = Verification .Minimal
45+ )
4346 executorch_output = copy .deepcopy (
4447 executorch_module .run_method ("forward" , tuple (inputs_flattened ))
4548 )
Original file line number Diff line number Diff line change 33import re
44import time
55import unittest
6+ import warnings
67
78from datetime import timedelta
89from typing import Any
@@ -249,6 +250,10 @@ def build_test_filter(args: argparse.Namespace) -> TestFilter:
249250def runner_main ():
250251 args = parse_args ()
251252
253+ # Suppress deprecation warnings for export_for_training, as it generates a
254+ # lot of log spam. We don't really need the warning here.
255+ warnings .simplefilter ("ignore" , category = FutureWarning )
256+
252257 begin_test_session (args .report )
253258
254259 if len (args .suite ) > 1 :
You can’t perform that action at this time.
0 commit comments