Merge branch 'testing' of https://github.com/unifyai/demos into testing

.github/workflows/_demo-test.yml

@@ -13,7 +13,7 @@ jobs:
     strategy:
       fail-fast: false
       matrix:
-        modules: [alexnet_demo, bert_demo, image_segmentation_with_ivy_unet, mmpretrain_to_jax, resnet_demo, torch_to_jax]
+        modules: [torch_to_jax]
 
     steps:
         - name: Clean repository

examples_and_demos/torch_to_jax.ipynb

@@ -50,9 +50,9 @@
         }
       ],
       "source": [
-        "!pip install -q ivy\n",
-        "!pip install -q transformers\n",
-        "!pip install -q dm-haiku"
+        "!pip install ivy\n",
+        "!pip install transformers\n",
+        "!pip install dm-haiku"
       ]
     },
     {

tests/main.py

@@ -17,13 +17,18 @@ def __init__(self, *args, **kwargs):
 
     @classmethod
     def setUp(cls):
+        print("starting kernel manager")
         cls.km = KernelManager()
         cls.km.start_kernel(
             extra_arguments=["--pylab=inline"], stderr=open(os.devnull, "w")
         )
+        print("started kernel manager")
         cls.kc = cls.km.blocking_client()
+        print("created blocking client")
         cls.kc.start_channels()
+        print("started channels")
         cls.kc.execute_interactive("import os;os.environ['IVY_ROOT']='.ivy'")
+        print("executed first cell")
 
     @classmethod
     def tearDown(cls):
@@ -39,6 +44,7 @@ def invert_config(self):
         self.config = inverted_config
 
     def test_notebook(self):
+        print("inside test_notebook")
         cells_to_ignore = []
         for cell_number, cell_content in enumerate(self.file.cells):
             outputs = []
@@ -141,6 +147,7 @@ def startTest(self, test):
 
     suite = unittest.TestLoader().loadTestsFromTestCase(NotebookTest)
     runner = IterativeTestRunner(verbosity=2)
+    print("REACHED BEFORE RUN")
     result = runner.run(suite)
 
     if result.wasSuccessful():