reset tests

tenstorrent · Feb 28, 2025 · 6f70217 · 6f70217
1 parent 15c386e
commit 6f70217
Showing 1 changed file with 19 additions and 19 deletions.
diff --git a/.github/workflows/batch-experiment.yaml b/.github/workflows/batch-experiment.yaml
@@ -40,29 +40,29 @@ jobs:
 
           # Array of tests to exclude
           exclude_tests=(
-            "models/mnist/test_mnist.py"
-            "models/MobileNetV2/test_MobileNetV2.py"
-            "models/openpose/test_openpose_v2.py"
-            "models/resnet/test_resnet.py"
-            "models/resnet50/test_resnet50.py"
-            "models/roberta/test_roberta.py"
-            #"models/hand_landmark/test_hand_landmark.py"
-            "models/squeeze_bert/test_squeeze_bert.py"
+            #"models/mnist/test_mnist.py"
+            #"models/MobileNetV2/test_MobileNetV2.py"
+            #"models/openpose/test_openpose_v2.py"
+            #"models/resnet/test_resnet.py"
+            #"models/resnet50/test_resnet50.py"
+            #"models/roberta/test_roberta.py"
+            ##"models/hand_landmark/test_hand_landmark.py"
+            #"models/squeeze_bert/test_squeeze_bert.py"
             #"models/llama/test_llama.py"
             #"models/timm/test_timm_image_classification.py"
-            "models/torchvision/test_torchvision_image_classification.py"
+            #"models/torchvision/test_torchvision_image_classification.py"
             #"models/unet/test_unet.py"
-            "models/albert/test_albert_question_answering.py"
-            "models/albert/test_alert_sequence_classification.py"
-            "models/albert/test_albert_token_classification.py"
+            #"models/albert/test_albert_question_answering.py"
+            #"models/albert/test_alert_sequence_classification.py"
+            #"models/albert/test_albert_token_classification.py"
             #"models/albert/test_albert_masked_lm.py"
-            "models/unet_carvana/test_unet_carvana.py"
-            "models/unet_brain/test_unet_brain.py"
-            "models/autoencoder_linear/test_autoencoder_linear.py"
+            #"models/unet_carvana/test_unet_carvana.py"
+            #"models/unet_brain/test_unet_brain.py"
+            #"models/autoencoder_linear/test_autoencoder_linear.py"
             #"models/perceiver_io/test_perceiver_io.py"
-            "models/dpr/test_dpr.py"
-            "models/bloom/test_bloom.py"
-            "models/distilbert/test_distilbert.py"
+            #"models/dpr/test_dpr.py"
+            #"models/bloom/test_bloom.py"
+            #"models/distilbert/test_distilbert.py"
 
           )
 
@@ -123,7 +123,7 @@ jobs:
               fi
               local batch_size_to_test=$(( (batch_range[0] + batch_range[1]) / 2 ))
               if (( batch_size_to_test % 2 != 0)); then
-                batch_size_to_test=(( $batch_size_to_test-1 ))
+                (( batch_size_to_test -= 1 ))
               fi
               
               echo "Testing with batch size $batch_size_to_test"