squeeze test progress commit [skip ci]

tenstorrent · Dec 27, 2024 · 40a14e8 · 40a14e8
1 parent fce84da
commit 40a14e8
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/forge/test/operators/pytorch/tm/test_squeeze.py b/forge/test/operators/pytorch/tm/test_squeeze.py
@@ -77,8 +77,8 @@ def verify(
             math_fidelity=test_vector.math_fidelity,
             warm_reset=warm_reset,
             value_range=ValueRanges.SMALL,
-            deprecated_verification=False,
-            verify_config=VerifyConfig(value_checker=AllCloseValueChecker()),
+            # deprecated_verification=False,
+            # verify_config=VerifyConfig(value_checker=AllCloseValueChecker()),
         )
 
 
@@ -104,14 +104,14 @@ def generate_kwargs(cls, test_vector: TestVector):
             operators=["squeeze"],
             input_sources=TestCollectionCommon.all.input_sources,
             input_shapes=TestCollectionCommon.all.input_shapes,
-            kwargs=lambda test_vector: TestParamsData.generate_kwargs(test_vector),
+            # kwargs=lambda test_vector: TestParamsData.generate_kwargs(test_vector),
         ),
         # Test Data formats collection:
         TestCollection(
             operators=["squeeze"],
             input_sources=TestCollectionCommon.single.input_sources,
             input_shapes=TestCollectionCommon.single.input_shapes,
-            kwargs=lambda test_vector: TestParamsData.generate_kwargs(test_vector),
+            # kwargs=lambda test_vector: TestParamsData.generate_kwargs(test_vector),
             dev_data_formats=[
                 item
                 for item in TestCollectionCommon.all.dev_data_formats
@@ -124,7 +124,7 @@ def generate_kwargs(cls, test_vector: TestVector):
             operators=["squeeze"],
             input_sources=TestCollectionCommon.single.input_sources,
             input_shapes=TestCollectionCommon.single.input_shapes,
-            kwargs=lambda test_vector: TestParamsData.generate_kwargs(test_vector),
+            # kwargs=lambda test_vector: TestParamsData.generate_kwargs(test_vector),
             dev_data_formats=TestCollectionCommon.single.dev_data_formats,
             math_fidelities=TestCollectionCommon.all.math_fidelities,
         ),