reduced verbosity

tenstorrent · Feb 26, 2025 · 894b00f · 894b00f
1 parent 5d710ec
commit 894b00f
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/.github/workflows/batch-experiment.yaml b/.github/workflows/batch-experiment.yaml
@@ -47,7 +47,7 @@ jobs:
           #  "models/resnet50/test_resnet50.py"
           #  "models/roberta/test_roberta.py"
             "models/unet/test_unet.py"
-          #  "models/hand_landmark/test_hand_landmark.py"
+            "models/hand_landmark/test_hand_landmark.py"
           )
 
           # Preprocess file paths
@@ -69,7 +69,7 @@ jobs:
           done
 
           TOTAL_TESTS=${#test_ids[@]}
-          TESTS_PER_GROUP=$(( TOTAL_TESTS / TOTAL_GROUPS ))
+          TESTS_PER_GROUP=1
 
           START_INDEX=$(( (CURRENT_GROUP - 1) * TESTS_PER_GROUP ))
           END_INDEX=$(( CURRENT_GROUP * TESTS_PER_GROUP ))

diff --git a/tests/conftest.py b/tests/conftest.py
@@ -203,8 +203,8 @@ def compile_and_run(device, reset_torch_dynamo, request):
                     model_name, option._aten_fx_graphs, option._out_fx_graphs, option._all_inputs
                 )
 
-            if len(option._out_fx_graphs) > 0:
-                option._out_fx_graphs[0].print_tabular()
+            # if len(option._out_fx_graphs) > 0:
+            #    option._out_fx_graphs[0].print_tabular()
 
             if model_name not in ["speecht5-tts", "ssd300_vgg16", "retinanet_resnet50_fpn_v2"]:
                 accuracy = calculate_accuracy(outputs, outputs_after)