Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Reorganize common and wf manager #922

Open
wants to merge 4 commits into
base: develop
Choose a base branch
from
Open
Show file tree
Hide file tree
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
4 changes: 2 additions & 2 deletions beeflow/client/bee_client.py
Original file line number Diff line number Diff line change
Expand Up @@ -22,9 +22,9 @@
import yaml

from beeflow.common import config_driver
from beeflow.common.cli import NaturalOrderGroup
from beeflow.common.util.cli import NaturalOrderGroup
from beeflow.common.connection import Connection
from beeflow.common import paths
from beeflow.common.util import paths
from beeflow.common.parser import CwlParser
from beeflow.common.wf_data import generate_workflow_id
from beeflow.client import core
Expand Down
2 changes: 1 addition & 1 deletion beeflow/client/core.py
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,7 @@
from beeflow.client import bee_client
from beeflow.common.config_driver import BeeConfig as bc
from beeflow.common import cli_connection
from beeflow.common import paths
from beeflow.common.util import paths
from beeflow.wf_manager.resources import wf_utils

from beeflow.common.deps import container_manager
Expand Down
2 changes: 1 addition & 1 deletion beeflow/common/build/container_drivers.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@
import subprocess
import tempfile
from beeflow.common.config_driver import BeeConfig as bc
from beeflow.common import log as bee_logging
from beeflow.common.util import log as bee_logging
from beeflow.common.build.utils import ContainerBuildError
from beeflow.common.build.build_driver import BuildDriver
from beeflow.common.crt.charliecloud_driver import CharliecloudDriver as crt_driver
Expand Down
2 changes: 1 addition & 1 deletion beeflow/common/build_interfaces.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@
from subprocess import CalledProcessError
from beeflow.common.build.container_drivers import CharliecloudBuildDriver
from beeflow.common.config_driver import BeeConfig as bc
from beeflow.common import log as bee_logging
from beeflow.common.util import log as bee_logging
from beeflow.common.build.utils import arg2task, ContainerBuildError


Expand Down
6 changes: 3 additions & 3 deletions beeflow/common/config_driver.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,9 +11,9 @@
import typer

from beeflow.common.config_validator import ConfigValidator
from beeflow.common.cli import NaturalOrderGroup
from beeflow.common import validation
from beeflow.common.tab_completion import filepath_completion
from beeflow.common.util.cli import NaturalOrderGroup
from beeflow.common.util import validation
from beeflow.common.util.tab_completion import filepath_completion


# System specific path set up
Expand Down
2 changes: 1 addition & 1 deletion beeflow/common/config_validator.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
"""Config validation code."""

from beeflow.common import log as bee_logging
from beeflow.common.util import log as bee_logging

log = bee_logging.setup(__name__)

Expand Down
2 changes: 1 addition & 1 deletion beeflow/common/crt/charliecloud_driver.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@
from beeflow.common.config_driver import BeeConfig as bc
from beeflow.common.build.utils import task2arg
from beeflow.common.container_path import convert_path
from beeflow.common import log as bee_logging
from beeflow.common.util import log as bee_logging


log = bee_logging.setup(__name__)
Expand Down
2 changes: 1 addition & 1 deletion beeflow/common/db/tm_db.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
import jsonpickle

from beeflow.common.db import bdb
from beeflow.common import log as bee_logging
from beeflow.common.util import log as bee_logging
from beeflow.common.wf_data import TaskStateUpdate

log = bee_logging.setup(__name__)
Expand Down
2 changes: 1 addition & 1 deletion beeflow/common/deps/container_manager.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,7 @@
import subprocess

from beeflow.common.config_driver import BeeConfig as bc
from beeflow.common import paths
from beeflow.common.util import paths
from celery import shared_task #noqa pylama can't find celery


Expand Down
4 changes: 2 additions & 2 deletions beeflow/common/deps/neo4j_manager.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,11 +9,11 @@
from beeflow.wf_manager.resources import wf_utils
from beeflow.common.db import wfm_db

from beeflow.common import paths
from beeflow.common.util import paths
from beeflow.common.deps import container_manager

from beeflow.common.config_driver import BeeConfig as bc
from beeflow.common import log as bee_logging
from beeflow.common.util import log as bee_logging

# Define directories within module scope
bee_workdir = paths.workdir()
Expand Down
2 changes: 1 addition & 1 deletion beeflow/common/deps/redis_manager.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@
import os
import subprocess

from beeflow.common import paths
from beeflow.common.util import paths


def start(log):
Expand Down
2 changes: 1 addition & 1 deletion beeflow/common/gdb/gdb_driver.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
"""Abstract base class for the handling of workflow DAGs."""

from abc import ABC, abstractmethod
from beeflow.common import log as bee_logging
from beeflow.common.util import log as bee_logging

log = bee_logging.setup(__name__)

Expand Down
2 changes: 1 addition & 1 deletion beeflow/common/gdb/neo4j_cypher.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
"""Neo4j/Cypher transaction functions used by the Neo4jDriver class."""

from re import fullmatch
from beeflow.common import log as bee_logging
from beeflow.common.util import log as bee_logging

log = bee_logging.setup(__name__)

Expand Down
2 changes: 1 addition & 1 deletion beeflow/common/gdb/neo4j_driver.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,7 @@
from beeflow.common.gdb import neo4j_cypher as tx
from beeflow.common.wf_data import (Workflow, Task, Requirement, Hint,
InputParameter, OutputParameter, StepInput, StepOutput)
from beeflow.common import log as bee_logging
from beeflow.common.util import log as bee_logging

log = bee_logging.setup(__name__)

Expand Down
File renamed without changes.
File renamed without changes.
File renamed without changes.
File renamed without changes.
File renamed without changes.
File renamed without changes.
File renamed without changes.
2 changes: 1 addition & 1 deletion beeflow/common/wf_interface.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
"""

import re
from beeflow.common import log as bee_logging
from beeflow.common.util import log as bee_logging
from beeflow.common.gdb.neo4j_driver import Neo4jDriver

log = bee_logging.setup(__name__)
Expand Down
2 changes: 1 addition & 1 deletion beeflow/common/worker/flux_worker.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@

import io
import os
from beeflow.common import log as bee_logging
from beeflow.common.util import log as bee_logging
from beeflow.common.worker.worker import Worker

log = bee_logging.setup(__name__)
Expand Down
4 changes: 2 additions & 2 deletions beeflow/common/worker/slurm_worker.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,9 +11,9 @@
import requests_unixsocket
import requests

from beeflow.common import log as bee_logging
from beeflow.common.util import log as bee_logging
from beeflow.common.worker.worker import (Worker, WorkerError)
from beeflow.common import validation
from beeflow.common.util import validation
from beeflow.common.worker.utils import get_state_sacct
from beeflow.common.worker.utils import parse_key_val

Expand Down
2 changes: 1 addition & 1 deletion beeflow/common/worker/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@

import subprocess
from beeflow.common.worker.worker import WorkerError
from beeflow.common import log as bee_logging
from beeflow.common.util import log as bee_logging


log = bee_logging.setup(__name__)
Expand Down
2 changes: 1 addition & 1 deletion beeflow/common/worker/worker.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@

from abc import ABC, abstractmethod
import os
from beeflow.common import log as bee_logging
from beeflow.common.util import log as bee_logging
from beeflow.common.crt_interface import ContainerRuntimeInterface


Expand Down
2 changes: 1 addition & 1 deletion beeflow/remote/remote.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,7 @@
import uvicorn

from beeflow.common import cli_connection
from beeflow.common import paths
from beeflow.common.util import paths
from beeflow.client import bee_client
from beeflow.common.config_driver import BeeConfig as bc

Expand Down
2 changes: 1 addition & 1 deletion beeflow/scheduler/scheduler.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,7 @@
from beeflow.scheduler import resource_allocation
from beeflow.common.config_driver import BeeConfig as bc
from beeflow.common.db import sched_db
from beeflow.common import log as bee_logging
from beeflow.common.util import log as bee_logging

from beeflow.common.db.bdb import connect_db

Expand Down
2 changes: 1 addition & 1 deletion beeflow/task_manager/background.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,7 @@
import traceback
import jsonpickle
from beeflow.task_manager import utils
from beeflow.common import log as bee_logging
from beeflow.common.util import log as bee_logging
from beeflow.common.build.utils import ContainerBuildError
from beeflow.common.build_interfaces import build_main
from beeflow.common.worker import WorkerError
Expand Down
2 changes: 1 addition & 1 deletion beeflow/task_manager/task_actions.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@
import traceback
from flask import jsonify, make_response
from flask_restful import Resource
from beeflow.common import log as bee_logging
from beeflow.common.util import log as bee_logging
from beeflow.task_manager import utils

log = bee_logging.setup(__name__)
Expand Down
2 changes: 1 addition & 1 deletion beeflow/task_manager/task_submit.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@
from flask import jsonify, make_response
from flask_restful import Resource, reqparse
import jsonpickle
from beeflow.common import log as bee_logging
from beeflow.common.util import log as bee_logging
from beeflow.task_manager import utils

log = bee_logging.setup(__name__)
Expand Down
2 changes: 1 addition & 1 deletion beeflow/task_manager/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,7 @@
from beeflow.common.db import tm_db
from beeflow.common.db import bdb
from beeflow.common import worker
from beeflow.common import paths
from beeflow.common.util import paths
from beeflow.common.connection import Connection
from beeflow.common.worker_interface import WorkerInterface

Expand Down
Loading