clean runnning part

Author: R-Palazzo

.github/workflows/run_benchmark_multi_table.yaml

@@ -4,12 +4,9 @@ on:
   workflow_dispatch:
   schedule:
     - cron: '0 5 1 * *'
-  push:
-    branches:
-      - issue-516-add-workflows
 
 jobs:
-  run-sdgym-benchmark:
+  run-sdgym-multi-table-benchmark:
     runs-on: ubuntu-latest
 
     steps:
@@ -31,15 +28,12 @@ jobs:
         source venv/bin/activate
 
         python -m pip install --upgrade pip
-        python -m pip install invoke tomli packaging
-        invoke install-sdv-enterprise
+        python -m pip install sdv-enterprise --index-url "https://${USERNAME}:${LICENSE_KEY}@pypi.datacebo.com"
         python -m pip install "sdgym[all] @ git+https://github.com/sdv-dev/SDGym.git@issue-516-add-workflows"
 
         echo "VIRTUAL_ENV=$(pwd)/venv" >> $GITHUB_ENV
         echo "$(pwd)/venv/bin" >> $GITHUB_PATH
 
-    - name: Stop workflow after dependency install (temporary)
-      run: exit 0
     - name: Run SDGym Benchmark
       env:
         GCP_SERVICE_ACCOUNT_JSON: ${{ secrets.GCP_SERVICE_ACCOUNT_JSON }}

.github/workflows/run_benchmark_single_table.yml

@@ -6,7 +6,7 @@ on:
     - cron: '0 5 1 * *'
 
 jobs:
-  run-sdgym-benchmark:
+  run-sdgym-single-table-benchmark:
     runs-on: ubuntu-latest
 
     steps:
@@ -28,7 +28,6 @@ jobs:
         source venv/bin/activate
 
         python -m pip install --upgrade pip
-        python -m pip install invoke tomli packaging
         python -m pip install sdv-enterprise --index-url "https://${USERNAME}:${LICENSE_KEY}@pypi.datacebo.com"
         python -m pip install "sdgym[all] @ git+https://github.com/sdv-dev/SDGym.git@issue-516-add-workflows"
 

.github/workflows/upload_benchmark_results_multi_table.yml

@@ -13,7 +13,7 @@ on:
       - issue-516-add-workflows
 
 jobs:
-  upload-sdgym-benchmark:
+  upload-sdgym-multi-table-benchmark:
     runs-on: ubuntu-latest
 
     steps:

.github/workflows/upload_benchmark_results_single_table.yml

@@ -10,7 +10,7 @@ on:
     - cron: '0 6 * * *'
 
 jobs:
-  upload-sdgym-benchmark:
+  upload-sdgym-single-table-benchmark:
     runs-on: ubuntu-latest
 
     steps:

sdgym/run_benchmark/utils.py

@@ -10,9 +10,7 @@
 
 from sdgym.s3 import parse_s3_path
 
-OUTPUT_DESTINATION_AWS = (
-    's3://sdgym-benchmark/Debug/GCP_Github/'  # 's3://sdgym-benchmark/Benchmarks/'
-)
+OUTPUT_DESTINATION_AWS = 's3://sdgym-benchmark/Benchmarks/'
 UPLOAD_DESTINATION_AWS = 's3://sdgym-benchmark/Debug/GCP/'
 DEBUG_SLACK_CHANNEL = 'sdv-alerts-debug'
 SLACK_CHANNEL = 'sdv-alerts'
@@ -104,7 +102,7 @@ def post_slack_message(channel, text):
 
 def post_benchmark_launch_message(date_str, compute_service='AWS', modality='single_table'):
     """Post a message to the SDV Alerts Slack channel when the benchmark is launched."""
-    channel = DEBUG_SLACK_CHANNEL
+    channel = SLACK_CHANNEL
     folder_name = get_result_folder_name(date_str)
     bucket, prefix = parse_s3_path(OUTPUT_DESTINATION_AWS)
     url_link = get_s3_console_link(bucket, f'{prefix}{modality}/{folder_name}/')
@@ -116,7 +114,7 @@ def post_benchmark_launch_message(date_str, compute_service='AWS', modality='sin
 
 def post_benchmark_uploaded_message(folder_name, commit_url=None, modality='single_table'):
     """Post benchmark uploaded message to sdv-alerts slack channel."""
-    channel = DEBUG_SLACK_CHANNEL
+    channel = SLACK_CHANNEL
     bucket, prefix = parse_s3_path(OUTPUT_DESTINATION_AWS)
     modality_text = modality.replace('_', '-')
     url_link = get_s3_console_link(bucket, quote_plus(f'{prefix}{modality}/SDGym Monthly Run.xlsx'))

tasks.py

@@ -218,25 +218,3 @@ def notify_sdgym_benchmark_uploaded(c, folder_name, commit_url=None, modality='s
     from sdgym.run_benchmark.utils import post_benchmark_uploaded_message
 
     post_benchmark_uploaded_message(folder_name, commit_url, modality)
-
-
-@task
-def install_sdv_enterprise(c, username=None, license_key=None):
-    """Install sdv-enterprise using sdv-installer if credentials are available."""
-    # 🔹 Import here to avoid triggering sdgym imports at load time
-    from sdgym._benchmark.credentials_utils import sdv_install_cmd
-
-    username = username or os.getenv("SDV_ENTERPRISE_USERNAME")
-    license_key = license_key or os.getenv("SDV_ENTERPRISE_LICENSE_KEY")
-    credentials = {
-        "sdv": {
-            "username": username,
-            "license_key": license_key,
-        }
-    }
-
-    install_cmd = sdv_install_cmd(credentials)
-    if install_cmd:
-        c.run(install_cmd)
-    else:
-        print("No sdv-enterprise credentials found. Skipping installation.")

tests/test_tasks.py

@@ -1,12 +1,9 @@
 """Tests for the ``tasks.py`` file."""
 
-from unittest.mock import Mock, patch
-
 from tasks import (
     _get_extra_dependencies,
     _get_minimum_versions,
     _resolve_version_conflicts,
-    install_sdv_enterprise,
 )
 
 
@@ -212,38 +209,3 @@ def test__resolve_version_conflicts_pointing_to_branch():
         'rdt==1.1.2',
         'copulas==0.12.0',
     ])
-
-
-@patch('sdgym._benchmark.credentials_utils.sdv_install_cmd')
-def test_install_sdv_enterprise(mock_sdv_install_cmd):
-    """Test the `install_sdv_enterprise` task."""
-    # Setup
-    username = 'test_user'
-    license_key = 'test_license_key'
-    mock_context = Mock()
-    mock_sdv_install_cmd.return_value = 'install command'
-
-    # Run
-    install_sdv_enterprise.body(mock_context, username=username, license_key=license_key)
-
-    # Assert
-    mock_sdv_install_cmd.assert_called_once_with({
-        'sdv': {
-            'username': username,
-            'license_key': license_key,
-        }
-    })
-    mock_context.run.assert_called_once_with('install command')
-
-
-def test_install_sdv_enterprise_no_credentials(capsys):
-    """Test the `install_sdv_enterprise` task without credentials."""
-    # Setup
-    mock_context = Mock()
-
-    # Run
-    install_sdv_enterprise.body(mock_context)
-
-    # Assert
-    captured = capsys.readouterr()
-    assert 'No sdv-enterprise credentials found. Skipping installation.' in captured.out