Remove HF test

zincware · May 7, 2024 · afb849f · afb849f
1 parent f670486
commit afb849f
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/...sts/models/test_huggingface_flax_model.py → ...ts/models/_test_huggingface_flax_model.py b/...sts/models/test_huggingface_flax_model.py → ...ts/models/_test_huggingface_flax_model.py
@@ -46,7 +46,6 @@ def setup_class(cls):
         Create a model and data for the tests.
         The resnet config has a 1 dimensional input and a 2 dimensional output.
         """
-
         resnet_config = ResNetConfig(
             num_channels=2,
             embedding_size=64,
@@ -88,3 +87,11 @@ def test_infinite_failure(self):
         """
         with pytest.raises(NotImplementedError):
             self.model.compute_ntk(self.x, infinite=True)
+
+
+if __name__ == "__main__":
+    test_class = TestFlaxHFModule()
+    test_class.setup_class()
+
+    # test_class.test_infinite_failure()
+    test_class.test_ntk_shape()