Differential D5228 Diff 26378 python_modules/dagster/dagster_tests/api_tests/api_tests_repo.py

Changeset View

Standalone View

python_modules/dagster/dagster_tests/api_tests/api_tests_repo.py

import string		import string

from dagster import (		from dagster import (
InputDefinition,		InputDefinition,
Int,		Int,
OutputDefinition,		OutputDefinition,
PartitionSetDefinition,		PartitionSetDefinition,
ScheduleDefinition,		ScheduleDefinition,
lambda_solid,		lambda_solid,
pipeline,		pipeline,
repository,		repository,
solid,		solid,
usable_as_dagster_type,		usable_as_dagster_type,
)		)
from dagster.core.definitions.decorators.sensor import sensor		from dagster.core.definitions.decorators.sensor import sensor
from dagster.core.definitions.sensor import SensorRunParams		from dagster.core.definitions.sensor import RunRequest


@lambda_solid		@lambda_solid
def do_something():		def do_something():
return 1		return 1


@lambda_solid		@lambda_solid
▲ Show 20 Lines • Show All 105 Lines • ▼ Show 20 Lines	return {
run_config_fn_for_partition=lambda partition: {},		run_config_fn_for_partition=lambda partition: {},
tags_fn_for_partition=error_partition_tags_fn,		tags_fn_for_partition=error_partition_tags_fn,
),		),
}		}


@sensor(pipeline_name="foo_pipeline")		@sensor(pipeline_name="foo_pipeline")
def sensor_foo(_):		def sensor_foo(_):
yield SensorRunParams(execution_key=None, run_config={"foo": "FOO"}, tags={"foo": "foo_tag"})		yield RunRequest(run_key=None, run_config={"foo": "FOO"}, tags={"foo": "foo_tag"})
yield SensorRunParams(execution_key=None, run_config={"foo": "FOO"})		yield RunRequest(run_key=None, run_config={"foo": "FOO"})


@sensor(pipeline_name="foo_pipeline")		@sensor(pipeline_name="foo_pipeline")
def sensor_error(_):		def sensor_error(_):
raise Exception("womp womp")		raise Exception("womp womp")


@repository		@repository
Show All 11 Lines