#0: Add pytest import to vit tests because they didn't exist before a…

…nd so it died on import error
tenstorrent · Apr 19, 2024 · d844ddf · d844ddf
1 parent 7155821
commit d844ddf
Show file tree

Hide file tree

Showing 9 changed files with 18 additions and 0 deletions.
diff --git a/models/experimental/vit/tests/test_vit_attention.py b/models/experimental/vit/tests/test_vit_attention.py
@@ -2,6 +2,8 @@
 
 # SPDX-License-Identifier: Apache-2.0
 
+import pytest
+
 import torch
 
 from loguru import logger

diff --git a/models/experimental/vit/tests/test_vit_embeddings.py b/models/experimental/vit/tests/test_vit_embeddings.py
@@ -2,6 +2,8 @@
 
 # SPDX-License-Identifier: Apache-2.0
 
+import pytest
+
 import torch
 
 from loguru import logger

diff --git a/models/experimental/vit/tests/test_vit_encoder.py b/models/experimental/vit/tests/test_vit_encoder.py
@@ -2,6 +2,8 @@
 
 # SPDX-License-Identifier: Apache-2.0
 
+import pytest
+
 import torch
 
 from loguru import logger

diff --git a/models/experimental/vit/tests/test_vit_intermediate.py b/models/experimental/vit/tests/test_vit_intermediate.py
@@ -2,6 +2,8 @@
 
 # SPDX-License-Identifier: Apache-2.0
 
+import pytest
+
 import torch
 
 from loguru import logger

diff --git a/models/experimental/vit/tests/test_vit_layer.py b/models/experimental/vit/tests/test_vit_layer.py
@@ -2,6 +2,8 @@
 
 # SPDX-License-Identifier: Apache-2.0
 
+import pytest
+
 import torch
 
 from loguru import logger

diff --git a/models/experimental/vit/tests/test_vit_model.py b/models/experimental/vit/tests/test_vit_model.py
@@ -2,6 +2,8 @@
 
 # SPDX-License-Identifier: Apache-2.0
 
+import pytest
+
 import torch
 
 from loguru import logger

diff --git a/models/experimental/vit/tests/test_vit_output.py b/models/experimental/vit/tests/test_vit_output.py
@@ -2,6 +2,8 @@
 
 # SPDX-License-Identifier: Apache-2.0
 
+import pytest
+
 import torch
 
 from loguru import logger

diff --git a/models/experimental/vit/tests/test_vit_selfattention.py b/models/experimental/vit/tests/test_vit_selfattention.py
@@ -2,6 +2,8 @@
 
 # SPDX-License-Identifier: Apache-2.0
 
+import pytest
+
 import torch
 
 from loguru import logger

diff --git a/models/experimental/vit/tests/test_vit_selfoutput.py b/models/experimental/vit/tests/test_vit_selfoutput.py
@@ -2,6 +2,8 @@
 
 # SPDX-License-Identifier: Apache-2.0
 
+import pytest
+
 import torch
 
 from loguru import logger