fix test

dagster-io · May 6, 2024 · fceb4be · fceb4be
1 parent c1f8316
commit fceb4be
Showing 1 changed file with 2 additions and 7 deletions.
diff --git a/python_modules/dagster/dagster_tests/asset_defs_tests/test_asset_defs_source_metadata.py b/python_modules/dagster/dagster_tests/asset_defs_tests/test_asset_defs_source_metadata.py
@@ -3,7 +3,6 @@
 
 from dagster import AssetsDefinition, load_assets_from_modules
 from dagster._core.definitions.metadata import (
-    DEFAULT_SOURCE_FILE_KEY,
     LocalFileCodeReference,
     with_source_code_references,
 )
@@ -74,16 +73,12 @@ def test_asset_code_origins() -> None:
                     )
 
                 assert isinstance(
-                    asset.metadata_by_key[key]["dagster/code_references"].code_references[
-                        DEFAULT_SOURCE_FILE_KEY
-                    ],
+                    asset.metadata_by_key[key]["dagster/code_references"].code_references[-1],
                     LocalFileCodeReference,
                 )
                 meta = cast(
                     LocalFileCodeReference,
-                    asset.metadata_by_key[key]["dagster/code_references"].code_references[
-                        DEFAULT_SOURCE_FILE_KEY
-                    ],
+                    asset.metadata_by_key[key]["dagster/code_references"].code_references[-1],
                 )
 
                 assert meta.file_path == expected_file_path