Commit d76bc437 authored by patil-suraj's avatar patil-suraj
Browse files

add skeleton for model tests

parent 8c1f5197
...@@ -118,6 +118,24 @@ class ModelTesterMixin(unittest.TestCase): ...@@ -118,6 +118,24 @@ class ModelTesterMixin(unittest.TestCase):
assert image is not None, "Make sure output is not None" assert image is not None, "Make sure output is not None"
def test_save_load(self):
pass
def test_determinism(self):
pass
def test_output(self):
pass
def test_forward_signature(self):
pass
def test_model_from_config(self):
pass
def test_training(self):
pass
class PipelineTesterMixin(unittest.TestCase): class PipelineTesterMixin(unittest.TestCase):
def test_from_pretrained_save_pretrained(self): def test_from_pretrained_save_pretrained(self):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment