upate usage of selected_asset_keys

dagster-io · Sep 26, 2023 · fb6cac4 · fb6cac4
1 parent fcabda0
commit fb6cac4
Show file tree

Hide file tree

Showing 4 changed files with 12 additions and 12 deletions.
diff --git 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
@@ -1585,7 +1585,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(
@@ -1882,7 +1882,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
@@ -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/...dagster_tests/definitions_tests/decorators_tests/test_asset_decorator_with_check_specs.py b/...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