remove task for installing sdv_enterprise

Author: R-Palazzo

.github/workflows/run_benchmark_multi_table.yaml

@@ -31,8 +31,7 @@ jobs:
         source venv/bin/activate
 
         python -m pip install --upgrade pip
-        python -m pip install invoke
-        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

.github/workflows/run_benchmark_single_table.yml

@@ -28,8 +28,7 @@ jobs:
         source venv/bin/activate
 
         python -m pip install --upgrade pip
-        python -m pip install invoke
-        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

tasks.py

@@ -219,22 +219,3 @@ def notify_sdgym_benchmark_uploaded(c, folder_name, commit_url=None, modality='s
 
     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."""
-    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,26 +209,3 @@ def test__resolve_version_conflicts_pointing_to_branch():
         'rdt==1.1.2',
         'copulas==0.12.0',
     ])
-
-
-@patch('tasks.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_sdv_install_cmd.return_value = 'install command'
-    mock_context = Mock()
-
-    # Run
-    install_sdv_enterprise(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')
-    assert False