Skip to content

Commit

Permalink
Merge pull request #29 from punch-mission/quick-fix
Browse files Browse the repository at this point in the history
add type hint, update gitignore
  • Loading branch information
jmbhughes authored Jan 10, 2024
2 parents d0160a1 + 8569129 commit 7dceb40
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 3 deletions.
3 changes: 3 additions & 0 deletions .gitignore
Original file line number Diff line number Diff line change
Expand Up @@ -2,3 +2,6 @@ scripts/credentials.py
/docs/build/
/venv/
/venv-docs/
build/
punchpipe.egg-info
.idea
4 changes: 1 addition & 3 deletions punchpipe/controlsegment/launcher.py
Original file line number Diff line number Diff line change
Expand Up @@ -49,7 +49,7 @@ def filter_for_launchable_flows(planned_flows, running_flow_count, max_flows_run


@task
async def launch_ready_flows(session: Session, flow_ids: List[int]):
async def launch_ready_flows(session: Session, flow_ids: List[int]) -> List:
"""Given a list of ready-to-launch flow_ids, this task creates flow runs in Prefect for them.
These flow runs are automatically marked as scheduled in Prefect and will be picked up by a work queue and
agent as soon as possible.
Expand Down Expand Up @@ -96,10 +96,8 @@ def launcher_flow(pipeline_configuration_path="config.yaml"):
"""
logger = get_run_logger()

# load pipeline configuration
pipeline_config = load_pipeline_configuration(pipeline_configuration_path)

# Get database credentials and establish a connection
logger.info("Establishing database connection")
session = get_database_session()

Expand Down

0 comments on commit 7dceb40

Please sign in to comment.