regen snippet

dagster-io · Feb 20, 2025 · 244e693 · 244e693
1 parent d962b98
commit 244e693
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/...pets/docs_beta_snippets/guides/components/shell-script-component/2-shell-command-empty.py b/...pets/docs_beta_snippets/guides/components/shell-script-component/2-shell-command-empty.py
@@ -1,5 +1,4 @@
-import dagster as dg
-
+from dagster import Definitions
 from dagster_components import (
     Component,
     ComponentLoadContext,
@@ -26,6 +25,6 @@ def get_schema(cls):
     def get_scaffolder(cls) -> DefaultComponentScaffolder:
         return DefaultComponentScaffolder()
 
-    def build_defs(self, load_context: ComponentLoadContext) -> dg.Definitions:
+    def build_defs(self, load_context: ComponentLoadContext) -> Definitions:
         # Add definition construction logic here.
-        return dg.Definitions()
+        return Definitions()
diff --git a/examples/docs_snippets/docs_snippets_tests/conftest.py b/examples/docs_snippets/docs_snippets_tests/conftest.py
@@ -6,3 +6,8 @@
 @pytest.fixture
 def docs_snippets_folder():
     return file_relative_path(__file__, "../docs_snippets/")
+
+
+@pytest.fixture
+def mock_s3_bucket(mock_s3_resource):
+    yield mock_s3_resource.create_bucket(Bucket="test-bucket")