Spaces:
Running
Running
| #!/usr/bin/env python3 | |
| """ | |
| Test data persistence in Trackio Space | |
| """ | |
| import requests | |
| import json | |
| import time | |
| import re | |
| def test_persistence(): | |
| """Test if experiment data persists across restarts""" | |
| print("🔍 Testing Data Persistence") | |
| print("=" * 50) | |
| # Test creating an experiment via API | |
| url = 'https://tonic-test-trackio-test.hf.space/gradio_api/call/create_experiment_interface' | |
| payload = {'data': ['test_persistence', 'Testing data persistence']} | |
| response = requests.post(url, json=payload) | |
| if response.status_code == 200: | |
| data = response.json() | |
| if 'event_id' in data: | |
| event_id = data['event_id'] | |
| print(f'✅ Experiment created with event ID: {event_id}') | |
| # Get the result | |
| get_url = f'{url}/{event_id}' | |
| time.sleep(2) | |
| get_response = requests.get(get_url) | |
| if get_response.status_code == 200: | |
| result = get_response.text | |
| print(f'✅ Experiment creation result: {result[:200]}...') | |
| # Extract experiment ID | |
| match = re.search(r'exp_\d{8}_\d{6}', result) | |
| if match: | |
| experiment_id = match.group() | |
| print(f'📋 Experiment ID: {experiment_id}') | |
| # Test logging metrics | |
| metrics_url = 'https://tonic-test-trackio-test.hf.space/gradio_api/call/log_metrics_interface' | |
| metrics_payload = { | |
| 'data': [experiment_id, '{"loss": 1.5, "accuracy": 0.8}', '100'] | |
| } | |
| metrics_response = requests.post(metrics_url, json=metrics_payload) | |
| if metrics_response.status_code == 200: | |
| print('✅ Metrics logged successfully') | |
| # Test getting experiment details | |
| details_url = 'https://tonic-test-trackio-test.hf.space/gradio_api/call/get_experiment_details' | |
| details_payload = {'data': [experiment_id]} | |
| details_response = requests.post(details_url, json=details_payload) | |
| if details_response.status_code == 200: | |
| details_data = details_response.json() | |
| if 'event_id' in details_data: | |
| details_event_id = details_data['event_id'] | |
| # Get details result | |
| details_get_url = f'{details_url}/{details_event_id}' | |
| time.sleep(2) | |
| details_get_response = requests.get(details_get_url) | |
| if details_get_response.status_code == 200: | |
| details_result = details_get_response.text | |
| print(f'✅ Experiment details retrieved: {details_result[:200]}...') | |
| if 'metrics' in details_result.lower(): | |
| print('✅ Found metrics in experiment details') | |
| else: | |
| print('❌ No metrics found in experiment details') | |
| else: | |
| print(f'❌ Failed to get details result: {details_get_response.status_code}') | |
| else: | |
| print('❌ No event_id in details response') | |
| else: | |
| print(f'❌ Failed to get experiment details: {details_response.status_code}') | |
| else: | |
| print(f'❌ Failed to log metrics: {metrics_response.status_code}') | |
| else: | |
| print('❌ Could not extract experiment ID') | |
| else: | |
| print(f'❌ Failed to get result: {get_response.status_code}') | |
| else: | |
| print('❌ No event_id in response') | |
| else: | |
| print(f'❌ Failed to create experiment: {response.status_code}') | |
| print("\n" + "=" * 50) | |
| print("🎯 Next Steps:") | |
| print("1. Check the Trackio Space: https://tonic-test-trackio-test.hf.space") | |
| print("2. Go to '📊 Visualizations' tab") | |
| print("3. Enter the experiment ID above") | |
| print("4. Test if the visualization shows data") | |
| print("=" * 50) | |
| if __name__ == "__main__": | |
| test_persistence() |