fix tests

Author: R-Palazzo

.github/workflows/run_benchmark_multi_table.yaml

@@ -28,12 +28,21 @@ jobs:
         source venv/bin/activate
 
         python -m pip install --upgrade pip
-        python -m pip install sdv-enterprise --index-url "https://${USERNAME}:${LICENSE_KEY}@pypi.datacebo.com"
+        python -m pip install sdv-installer
+        python -c "
+        from sdv_installer.installation.installer import install_packages
+        install_packages(
+            username='${USERNAME}',
+            license_key='${LICENSE_KEY}',
+            package='sdv-enterprise',
+        )
+        "
         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

@@ -28,7 +28,15 @@ jobs:
         source venv/bin/activate
 
         python -m pip install --upgrade pip
-        python -m pip install sdv-enterprise --index-url "https://${USERNAME}:${LICENSE_KEY}@pypi.datacebo.com"
+        python -m pip install sdv-installer
+        python -c "
+        from sdv_installer.installation.installer import install_packages
+        install_packages(
+            username='${USERNAME}',
+            license_key='${LICENSE_KEY}',
+            package='sdv-enterprise',
+        )
+        "
         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

sdgym/run_benchmark/utils.py

@@ -114,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 = SLACK_CHANNEL
+    channel = DEBUG_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'))

tests/unit/run_benchmark/test_utils.py

@@ -107,7 +107,7 @@ def test_post_benchmark_launch_message(
     mock_get_s3_console_link.assert_called_once_with(
         'my-bucket', f'my-prefix/single_table/{folder_name}/'
     )
-    mock_post_slack_message.assert_called_once_with('sdv-alerts-debug', expected_body)
+    mock_post_slack_message.assert_called_once_with('sdv-alerts', expected_body)
 
 
 @patch('sdgym.run_benchmark.utils.post_slack_message')