undo adding a test

nod-ai · Nov 13, 2023 · 8ecf4ed · 8ecf4ed
1 parent e631023
commit 8ecf4ed
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 4 deletions.
diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
@@ -34,6 +34,7 @@ jobs:
             -r pytorch-cpu-requirements.txt \
             -r torchvision-requirements.txt          
           pip install --upgrade -r requirements.txt
+          pip install --upgrade -r turbine-models-requirements.txt
           pip install -e  .[testing]
 
       - name: Run tests

diff --git a/python/turbine_models/model_builder.py b/python/turbine_models/model_builder.py
@@ -1,9 +1,8 @@
 from transformers import AutoModel, AutoTokenizer, AutoConfig
-from abc import ABC, abstractmethod
 import torch
 import shark_turbine.aot as aot
 
-class HFBuilder():
+class TransformerBuilder():
     """
     A model builder that uses Hugging Face's transformers library to build a PyTorch model.
 

diff --git a/tests/examples/aot_mlp_test.py b/tests/examples/aot_mlp_test.py
@@ -17,15 +17,13 @@ def _run(local_path: str):
     path = REPO_DIR / local_path
     subprocess.check_call([sys.executable, str(path)])
 
-
 class AOTMLPTest(unittest.TestCase):
     def testMLPExportSimple(self):
         _run("examples/aot_mlp/mlp_export_simple.py")
 
     def testMLPExportSimple(self):
         _run("examples/aot_mlp/mlp_export_dynamic.py")
 
-
 if __name__ == "__main__":
     logging.basicConfig(level=logging.DEBUG)
     unittest.main()