use non deprecated methods in tests

dagster-io · Jan 29, 2024 · 8e4e057 · 8e4e057
1 parent ee98bb2
commit 8e4e057
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python_modules/dagster/dagster_tests/asset_defs_tests/test_materialize.py b/python_modules/dagster/dagster_tests/asset_defs_tests/test_materialize.py
@@ -268,12 +268,12 @@ def multi_asset_with_internal_deps(thing):
 def test_materialize_tags():
     @asset
     def the_asset(context):
-        assert context.get_tag("key1") == "value1"
+        assert context.run.tags.get("key1") == "value1"
 
     with instance_for_test() as instance:
         result = materialize([the_asset], instance=instance, tags={"key1": "value1"})
         assert result.success
-        assert result.dagster_run.tags == {"key1": "value1"}
+        assert result.run.tags == {"key1": "value1"}
 
 
 def test_materialize_partition_key():