Skip to content

Commit

Permalink
Trying to get debug prints working
Browse files Browse the repository at this point in the history
  • Loading branch information
AidanHilt committed Jan 26, 2024
1 parent 4cae3c5 commit b2566fc
Showing 1 changed file with 14 additions and 14 deletions.
28 changes: 14 additions & 14 deletions src/argowrapper/engine/argo_engine.py
Original file line number Diff line number Diff line change
@@ -1,43 +1,42 @@
import string
import traceback
from datetime import datetime
from typing import Dict, List, Literal

import argo_workflows
from argo_workflows.api import (
archived_workflow_service_api,
workflow_service_api,
artifact_service_api,
workflow_service_api,
)
from argo_workflows.exceptions import NotFoundException
from argo_workflows.model.io_argoproj_workflow_v1alpha1_retry_archived_workflow_request import (
IoArgoprojWorkflowV1alpha1RetryArchivedWorkflowRequest,
)
from argo_workflows.model.io_argoproj_workflow_v1alpha1_workflow_create_request import (
IoArgoprojWorkflowV1alpha1WorkflowCreateRequest,
)
from argo_workflows.model.io_argoproj_workflow_v1alpha1_workflow_terminate_request import (
IoArgoprojWorkflowV1alpha1WorkflowTerminateRequest,
)
from argo_workflows.model.io_argoproj_workflow_v1alpha1_workflow_retry_request import (
IoArgoprojWorkflowV1alpha1WorkflowRetryRequest,
)
from argo_workflows.model.io_argoproj_workflow_v1alpha1_retry_archived_workflow_request import (
IoArgoprojWorkflowV1alpha1RetryArchivedWorkflowRequest,
from argo_workflows.model.io_argoproj_workflow_v1alpha1_workflow_terminate_request import (
IoArgoprojWorkflowV1alpha1WorkflowTerminateRequest,
)
from argo_workflows.exceptions import NotFoundException

from argowrapper import logger
from argowrapper.constants import (
ARGO_HOST,
ARGO_NAMESPACE,
WORKFLOW,
GEN3_USER_METADATA_LABEL,
GEN3_SUBMIT_TIMESTAMP_LABEL,
GEN3_TEAM_PROJECT_METADATA_LABEL,
GEN3_USER_METADATA_LABEL,
GEN3_WORKFLOW_PHASE_LABEL,
GEN3_SUBMIT_TIMESTAMP_LABEL,
WORKFLOW,
)
from argowrapper.engine.helpers import argo_engine_helper
from argowrapper.engine.helpers.workflow_factory import WorkflowFactory
from argowrapper.workflows.argo_workflows.gwas import GWAS

from datetime import datetime


class ArgoEngine:
"""
Expand Down Expand Up @@ -71,7 +70,6 @@ def __init__(self, dry_run: bool = False):
self.artifact_api_instance = artifact_service_api.ArtifactServiceApi(api_client)

def _get_workflow_details_dict(self, workflow_name: str) -> Dict:

return self.api_instance.get_workflow(
namespace=ARGO_NAMESPACE,
name=workflow_name,
Expand Down Expand Up @@ -418,7 +416,9 @@ def get_user_workflows_for_current_month(self, auth_header: str) -> List[Dict]:

def get_workflows_for_label_selector(self, label_selector: str) -> List[Dict]:
try:
logger.info(f"ARGO_NAMESPACE: {ARGO_NAMESPACE}")
logger.info(
f"ARGO_NAMESPACE for get_workflows_for_label_selector: {ARGO_NAMESPACE}"
)
workflow_list_return = self.api_instance.list_workflows(
namespace=ARGO_NAMESPACE,
list_options_label_selector=label_selector,
Expand Down

0 comments on commit b2566fc

Please sign in to comment.