diff --git a/.github/workflows/ci-cloud.yml b/.github/workflows/ci-cloud.yml index 04e229a..24c29cc 100644 --- a/.github/workflows/ci-cloud.yml +++ b/.github/workflows/ci-cloud.yml @@ -8,7 +8,7 @@ defaults: shell: bash jobs: - pytester: + integration: runs-on: ${{ matrix.os }} strategy: fail-fast: false @@ -17,14 +17,13 @@ jobs: python-version: ["3.10"] # Timeout: https://stackoverflow.com/a/59076067/4521646 - timeout-minutes: 35 + timeout-minutes: 15 env: TORCH_URL: "https://download.pytorch.org/whl/cpu/torch_stable.html" steps: - uses: actions/checkout@v4 - - name: Set up Python ${{ matrix.python-version }} - uses: actions/setup-python@v5 + - uses: actions/setup-python@v5 with: python-version: ${{ matrix.python-version }} cache: "pip" diff --git a/tests/test_io_cloud.py b/tests/test_io_cloud.py index e13e30f..1230359 100644 --- a/tests/test_io_cloud.py +++ b/tests/test_io_cloud.py @@ -9,10 +9,18 @@ from torch.nn import Module -@pytest.mark.parametrize("name", ["org/model", "model-name", "/too/many/slashes"]) -def test_wrong_model_name(name): +@pytest.mark.parametrize( + "name", ["/too/many/slashes", "org/model"] +) # todo: add back "model-name" after next SDk release +def test_upload_wrong_model_name(name): with pytest.raises(ValueError, match=r".*organization/teamspace/model.*"): upload_model_files(path="path/to/checkpoint", name=name) + + +@pytest.mark.parametrize( + "name", ["/too/many/slashes", "org/model"] +) # todo: add back "model-name" after next SDk release +def test_download_wrong_model_name(name): with pytest.raises(ValueError, match=r".*organization/teamspace/model.*"): download_model(name=name)