oxirs-embed 0.2.4

Knowledge graph embeddings with TransE, ComplEx, and custom models
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86

//! Auto-generated module
//!
//! 🤖 Generated with [SplitRS](https://github.com/cool-japan/splitrs)

#[cfg(test)]
mod tests {
    use super::super::{
        AWSSageMakerService, CloudIntegrationConfig, CloudIntegrationManager, CloudService,
        DeploymentConfig, DeploymentStatus, NetworkingConfig, ResourceRequirements,
    };
    use std::collections::HashMap;
    #[tokio::test]
    async fn test_cloud_integration_manager_creation() {
        let config = CloudIntegrationConfig::default();
        let manager = CloudIntegrationManager::new(config);
        let providers = manager.providers.read().await;
        assert!(providers.is_empty());
    }
    #[tokio::test]
    async fn test_aws_sagemaker_service() {
        let service = AWSSageMakerService::new(
            "us-east-1".to_string(),
            "test_key".to_string(),
            "test_secret".to_string(),
            None,
        );
        let config = DeploymentConfig {
            model_name: "test-model".to_string(),
            model_version: "1.0".to_string(),
            instance_type: "ml.m5.large".to_string(),
            initial_instance_count: 1,
            auto_scaling_enabled: true,
            environment_variables: HashMap::new(),
            resource_requirements: ResourceRequirements {
                cpu_cores: 2.0,
                memory_gb: 8.0,
                gpu_count: 0,
                storage_gb: 50,
            },
            networking: NetworkingConfig {
                vpc_config: None,
                enable_network_isolation: false,
                custom_security_groups: vec![],
            },
            data_capture: None,
        };
        let result = service.deploy_model(&config).await.expect("should succeed");
        assert!(matches!(result.status, DeploymentStatus::Creating));
        assert!(result.cost_estimate.is_some());
    }
    #[tokio::test]
    async fn test_cost_estimation() {
        let service = AWSSageMakerService::new(
            "us-east-1".to_string(),
            "test_key".to_string(),
            "test_secret".to_string(),
            None,
        );
        let config = DeploymentConfig {
            model_name: "test-model".to_string(),
            model_version: "1.0".to_string(),
            instance_type: "ml.m5.large".to_string(),
            initial_instance_count: 2,
            auto_scaling_enabled: true,
            environment_variables: HashMap::new(),
            resource_requirements: ResourceRequirements {
                cpu_cores: 2.0,
                memory_gb: 8.0,
                gpu_count: 0,
                storage_gb: 50,
            },
            networking: NetworkingConfig {
                vpc_config: None,
                enable_network_isolation: false,
                custom_security_groups: vec![],
            },
            data_capture: None,
        };
        let estimate = service
            .estimate_costs(&config, 24)
            .await
            .expect("should succeed");
        assert!(estimate.hourly_cost_usd > 0.0);
        assert!(estimate.estimated_monthly_cost_usd > 0.0);
    }
}