test links checker

Author: Ofir Gordon

.github/workflows/nightly.yml

@@ -36,6 +36,6 @@ jobs:
           twine upload --repository pypi dist/* --verbose -u __token__ -p ${{ secrets.PYPI_API_SSI_DEV_NIGHTLY_KEY }}
       - name: Post publish import test
         run: |
-          pip install mct-nightly tensorflow torch
+          pip install mct-nightly tensorflow torch torchvision
           version=$(python -c 'import model_compression_toolkit; print(model_compression_toolkit.__version__)')
           echo $version

tests/doc_tests/test_docs_links.py

@@ -31,6 +31,7 @@ class TestDocsLinks(unittest.TestCase):
     @staticmethod
     def check_link(_url, branch_name):
         if 'sony/model_optimization' in _url:
+            _url = _url.replace('/main/', f'/{branch_name}/')
             index = _url.find(f"/{branch_name}/")
             print(_url)
             print(branch_name)