diff --git a/sdk/aqueduct/integrations/dynamic_k8s_integration.py b/sdk/aqueduct/integrations/dynamic_k8s_integration.py index 3286485a0..9c5119895 100644 --- a/sdk/aqueduct/integrations/dynamic_k8s_integration.py +++ b/sdk/aqueduct/integrations/dynamic_k8s_integration.py @@ -1,7 +1,7 @@ from typing import Dict, Union +from aqueduct.integrations.validation import validate_is_connected from aqueduct import globals -from aqueduct.integrations.integration_decorators import validate_is_connected from aqueduct.constants.enums import K8sClusterActionType, K8sClusterStatusType from aqueduct.error import InvalidIntegrationException, InvalidUserArgumentException diff --git a/sdk/aqueduct/integrations/google_sheets_integration.py b/sdk/aqueduct/integrations/google_sheets_integration.py index b44d0e1fe..acaef59f1 100644 --- a/sdk/aqueduct/integrations/google_sheets_integration.py +++ b/sdk/aqueduct/integrations/google_sheets_integration.py @@ -1,6 +1,6 @@ from typing import Optional -from aqueduct.integrations.integration_decorators import validate_is_connected +from aqueduct.integrations.validation import validate_is_connected from aqueduct.artifacts.base_artifact import BaseArtifact from aqueduct.artifacts.table_artifact import TableArtifact diff --git a/sdk/aqueduct/integrations/mongodb_integration.py b/sdk/aqueduct/integrations/mongodb_integration.py index 58ba70534..f8504102c 100644 --- a/sdk/aqueduct/integrations/mongodb_integration.py +++ b/sdk/aqueduct/integrations/mongodb_integration.py @@ -1,7 +1,7 @@ import json from typing import Any, Dict, List, Optional -from aqueduct.integrations.integration_decorators import validate_is_connected +from aqueduct.integrations.validation import validate_is_connected from aqueduct import globals from aqueduct.artifacts import preview as artifact_utils diff --git a/sdk/aqueduct/integrations/s3_integration.py b/sdk/aqueduct/integrations/s3_integration.py index 29090866c..a7fcc573a 100644 --- a/sdk/aqueduct/integrations/s3_integration.py +++ b/sdk/aqueduct/integrations/s3_integration.py @@ -1,7 +1,7 @@ import json from typing import List, Optional, Union -from aqueduct.integrations.integration_decorators import validate_is_connected +from aqueduct.integrations.validation import validate_is_connected from aqueduct import globals from aqueduct.artifacts import preview as artifact_utils diff --git a/sdk/aqueduct/integrations/salesforce_integration.py b/sdk/aqueduct/integrations/salesforce_integration.py index 634731e14..b00edffda 100644 --- a/sdk/aqueduct/integrations/salesforce_integration.py +++ b/sdk/aqueduct/integrations/salesforce_integration.py @@ -1,7 +1,7 @@ import uuid from typing import Optional -from aqueduct.integrations.integration_decorators import validate_is_connected +from aqueduct.integrations.validation import validate_is_connected from aqueduct.artifacts.base_artifact import BaseArtifact from aqueduct.artifacts.table_artifact import TableArtifact diff --git a/sdk/aqueduct/integrations/sql_integration.py b/sdk/aqueduct/integrations/sql_integration.py index dd9b77d89..06a3e7f0c 100644 --- a/sdk/aqueduct/integrations/sql_integration.py +++ b/sdk/aqueduct/integrations/sql_integration.py @@ -1,7 +1,7 @@ from typing import List, Optional, Union import pandas as pd -from aqueduct.integrations.integration_decorators import validate_is_connected +from aqueduct.integrations.validation import validate_is_connected from aqueduct.artifacts.base_artifact import BaseArtifact from aqueduct.artifacts.preview import preview_artifact diff --git a/sdk/aqueduct/integrations/integration_decorators.py b/sdk/aqueduct/integrations/validation.py similarity index 100% rename from sdk/aqueduct/integrations/integration_decorators.py rename to sdk/aqueduct/integrations/validation.py diff --git a/sdk/aqueduct/utils/integration_validation.py b/sdk/aqueduct/utils/integration_validation.py index b8350744a..1d4bc6094 100644 --- a/sdk/aqueduct/utils/integration_validation.py +++ b/sdk/aqueduct/utils/integration_validation.py @@ -6,7 +6,6 @@ from aqueduct.models.execution_state import ExecutionState -# TODO: maybe move this to artifacts/? def validate_integration_is_connected(name: str, exec_state: Optional[ExecutionState]) -> None: """Method used to determine if this integration was successfully connected to or not. If not successfully connected (or pending), we will raise an Exception. @@ -17,7 +16,7 @@ def validate_integration_is_connected(name: str, exec_state: Optional[ExecutionS if exec_state.status == ExecutionStatus.FAILED: raise IntegrationFailedToConnect( - "Cannot use integration %s because it has not been successfully connected to:" + "Cannot use integration %s because it has not been successfully connected to: " "%s\n%s\n\n Please see the /integrations page on the UI for more details." % ( name, exec_state.error.tip, exec_state.error.context, ) @@ -25,7 +24,7 @@ def validate_integration_is_connected(name: str, exec_state: Optional[ExecutionS else: raise IntegrationConnectionInProgress( "Cannot use integration %s because it is still in the process of connecting." - "Please see the /integrations page oen the UI for more details." + "Please see the /integrations page on the UI for more details." % name ) \ No newline at end of file