Updating integration test

sdv-dev · Dec 11, 2024 · 562aaa3 · 562aaa3
1 parent 3c38070
commit 562aaa3
Show file tree

Hide file tree

Showing 4 changed files with 0 additions and 26 deletions.
diff --git a/.github/workflows/integration.yml b/.github/workflows/integration.yml
@@ -15,14 +15,8 @@ jobs:
         include:
           - os: macos-latest
             python-version: '3.8'
-            env:
-              PYTORCH_ENABLE_MPS_FALLBACK: "1"
-              PYTORCH_MPS_HIGH_WATERMARK_RATIO: 0.0
           - os: macos-latest
             python-version: '3.12'
-            env:
-              PYTORCH_ENABLE_MPS_FALLBACK: "1"
-              PYTORCH_MPS_HIGH_WATERMARK_RATIO: 0.0
     steps:
     - uses: actions/checkout@v4
     - name: Set up Python ${{ matrix.python-version }}

diff --git a/.github/workflows/minimum.yml b/.github/workflows/minimum.yml
@@ -17,15 +17,8 @@ jobs:
         include:
           - os: macos-13
             python-version: '3.8'
-            env:
-              PYTORCH_ENABLE_MPS_FALLBACK: "1"
-              PYTORCH_MPS_HIGH_WATERMARK_RATIO: 0.0
           - os: macos-latest
             python-version: '3.12'
-            env:
-              PYTORCH_ENABLE_MPS_FALLBACK: "1"
-              PYTORCH_MPS_HIGH_WATERMARK_RATIO: 0.0
-
     steps:
     - uses: actions/checkout@v4
     - name: Set up Python ${{ matrix.python-version }}

diff --git a/.github/workflows/unit.yml b/.github/workflows/unit.yml
@@ -18,12 +18,8 @@ jobs:
         include:
           - os: macos-latest
             python-version: '3.8'
-            env:
-              PYTORCH_MPS_HIGH_WATERMARK_RATIO: 0.0
           - os: macos-latest
             python-version: '3.12'
-            env:
-              PYTORCH_MPS_HIGH_WATERMARK_RATIO: 0.0
     steps:
     - uses: actions/checkout@v4
     - name: Set up Python ${{ matrix.python-version }}

diff --git a/tests/integration/test_benchmark.py b/tests/integration/test_benchmark.py
@@ -69,7 +69,6 @@ def test_benchmark_single_table_realtabformer_no_metrics():
     assert sample_time >= 0
 
 
-@pytest.mark.skipif(sys.platform.startswith('darwin'), reason='Test not supported on github MacOS')
 def test_benchmark_single_table_no_metrics():
     """Test it without metrics."""
     # Run
@@ -83,14 +82,6 @@ def test_benchmark_single_table_no_metrics():
     assert not output.empty
     assert 'Train_Time' in output
     assert 'Sample_Time' in output
-
-    train_time = output['Train_Time'][0]
-    sample_time = output['Sample_Time'][0]
-    assert isinstance(train_time, (int, float, complex)), 'Train_Time is not numerical'
-    assert isinstance(sample_time, (int, float, complex)), 'Sample_Time is not numerical'
-    assert train_time >= 0
-    assert sample_time >= 0
-
     # Expect no metric columns.
     assert len(output.columns) == 10