Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 3 additions & 3 deletions tests/integration/test_enhance_integration.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@

import pytest
from roex_python.client import RoExClient
from roex_python.models import MixEnhanceRequest, MusicalStyle
from roex_python.models import MixEnhanceRequest, EnhanceMusicalStyle
from roex_python.utils import upload_file


Expand All @@ -29,7 +29,7 @@ def test_mix_enhancement_workflow(self, requires_api_key, integration_audio_file
# Create enhancement request
enhance_request = MixEnhanceRequest(
audio_file_location=track_url,
musical_style=MusicalStyle.POP,
musical_style=EnhanceMusicalStyle.POP,
is_master=False,
fix_clipping_issues=True,
fix_loudness_issues=True,
Expand All @@ -46,7 +46,7 @@ def test_mix_enhancement_workflow(self, requires_api_key, integration_audio_file
# Retrieve enhanced track (with polling)
enhanced = client.enhance.retrieve_enhanced_track(
task.mixrevive_task_id,
max_attempts=50,
timeout=500,
poll_interval=10
)
assert enhanced is not None
Expand Down