fix group_name in to_source_assets (#8279)

dagster-io · Jun 9, 2022 · c52c203 · c52c203
1 parent 079bdd9
commit c52c203
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/python_modules/dagster/dagster/core/asset_defs/assets.py b/python_modules/dagster/dagster/core/asset_defs/assets.py
@@ -381,6 +381,7 @@ def to_source_assets(self) -> Sequence[SourceAsset]:
                     description=output_def.description,
                     resource_defs=self.resource_defs,
                     partitions_def=self.partitions_def,
+                    group_name=self.group_names[asset_key],
                 )
             )
 

diff --git a/python_modules/dagster/dagster_tests/core_tests/asset_defs_tests/test_asset_group.py b/python_modules/dagster/dagster_tests/core_tests/asset_defs_tests/test_asset_group.py
@@ -1263,7 +1263,7 @@ def asset2():
 
 
 def test_to_source_assets():
-    @asset
+    @asset(group_name="abc")
     def my_asset():
         ...
 
@@ -1282,6 +1282,7 @@ def my_multi_asset():
             AssetKey(["my_asset"]),
             io_manager_key="io_manager",
             resource_defs={"io_manager": fs_io_manager},
+            group_name="abc",
         ),
         SourceAsset(
             AssetKey(["my_asset_name"]),