diff --git a/python_modules/dagster-test/dagster_test/toys/asset_checks.py b/python_modules/dagster-test/dagster_test/toys/asset_checks.py index e258f12b3e90c..343d630847fe4 100644 --- a/python_modules/dagster-test/dagster_test/toys/asset_checks.py +++ b/python_modules/dagster-test/dagster_test/toys/asset_checks.py @@ -144,10 +144,10 @@ def random_fail_check_on_partitioned_asset(): can_subset=True, ) def multi_asset_1_and_2(context): - if AssetKey("multi_asset_piece_1") in context.selected_asset_keys: + if AssetKey("multi_asset_piece_1") in context.asset_keys: yield Output(1, output_name="one") yield AssetCheckResult(success=True, metadata={"foo": "bar"}) - if AssetKey("multi_asset_piece_2") in context.selected_asset_keys: + if AssetKey("multi_asset_piece_2") in context.asset_keys: yield Output(1, output_name="two") diff --git a/python_modules/dagster/dagster_tests/asset_defs_tests/test_assets.py b/python_modules/dagster/dagster_tests/asset_defs_tests/test_assets.py index 8fce3dc997ae2..2d0410f1373fc 100644 --- a/python_modules/dagster/dagster_tests/asset_defs_tests/test_assets.py +++ b/python_modules/dagster/dagster_tests/asset_defs_tests/test_assets.py @@ -1584,7 +1584,7 @@ def my_function(): can_subset=True, ) def subset(context: AssetExecutionContext): - # ...use context.selected_asset_keys materialize subset of assets without IO manager + # ...use context.asset_keys materialize subset of assets without IO manager pass with pytest.raises( @@ -1881,7 +1881,7 @@ def basic_deps(): can_subset=True, ) def basic_subset(context: AssetExecutionContext): - for key in context.selected_asset_keys: + for key in context.asset_keys: yield MaterializeResult(asset_key=key) mats = _exec_asset(basic_subset, ["table_A"]) diff --git a/python_modules/dagster/dagster_tests/asset_defs_tests/test_assets_job.py b/python_modules/dagster/dagster_tests/asset_defs_tests/test_assets_job.py index 30bc8a0b49336..af5bc3d7a0172 100644 --- a/python_modules/dagster/dagster_tests/asset_defs_tests/test_assets_job.py +++ b/python_modules/dagster/dagster_tests/asset_defs_tests/test_assets_job.py @@ -2738,10 +2738,10 @@ def test_subset_cycle_resolution_basic(): can_subset=True, ) def foo(context, s, a_prime): - context.log.info(context.selected_asset_keys) - if AssetKey("a") in context.selected_asset_keys: + context.log.info(context.asset_keys) + if AssetKey("a") in context.asset_keys: yield Output(s + 1, "a") - if AssetKey("b") in context.selected_asset_keys: + if AssetKey("b") in context.asset_keys: yield Output(a_prime + 1, "b") @multi_asset( @@ -2753,10 +2753,10 @@ def foo(context, s, a_prime): can_subset=True, ) def foo_prime(context, a, b): - context.log.info(context.selected_asset_keys) - if AssetKey("a_prime") in context.selected_asset_keys: + context.log.info(context.asset_keys) + if AssetKey("a_prime") in context.asset_keys: yield Output(a + 1, "a_prime") - if AssetKey("b_prime") in context.selected_asset_keys: + if AssetKey("b_prime") in context.asset_keys: yield Output(b + 1, "b_prime") job = Definitions( diff --git a/python_modules/dagster/dagster_tests/definitions_tests/decorators_tests/test_asset_decorator_with_check_specs.py b/python_modules/dagster/dagster_tests/definitions_tests/decorators_tests/test_asset_decorator_with_check_specs.py index d8545bc103f50..3b65ed0610213 100644 --- a/python_modules/dagster/dagster_tests/definitions_tests/decorators_tests/test_asset_decorator_with_check_specs.py +++ b/python_modules/dagster/dagster_tests/definitions_tests/decorators_tests/test_asset_decorator_with_check_specs.py @@ -403,10 +403,10 @@ def test_multi_asset_with_check_subset(): can_subset=True, ) def asset_1_and_2(context: AssetExecutionContext): - if AssetKey("asset1") in context.selected_asset_keys: + if AssetKey("asset1") in context.asset_keys: yield Output(None, output_name="one") yield AssetCheckResult(check_name="check1", success=True) - if AssetKey("asset2") in context.selected_asset_keys: + if AssetKey("asset2") in context.asset_keys: yield Output(None, output_name="two") # no selection