dagster-io
diff --git a/‎js_modules/dagster-ui/packages/ui-core/src/graphql/possibleTypes.generated.json
+1-1 b/‎js_modules/dagster-ui/packages/ui-core/src/graphql/possibleTypes.generated.json
+1-1
diff --git a/‎js_modules/dagster-ui/packages/ui-core/src/graphql/schema.graphql
+9 b/‎js_modules/dagster-ui/packages/ui-core/src/graphql/schema.graphql
+9
diff --git a/‎js_modules/dagster-ui/packages/ui-core/src/graphql/types.ts
+31 b/‎js_modules/dagster-ui/packages/ui-core/src/graphql/types.ts
+31
diff --git a/‎python_modules/dagster-graphql/dagster_graphql_tests/graphql/components/test_dg_docs_json.py
+3-1 b/‎python_modules/dagster-graphql/dagster_graphql_tests/graphql/components/test_dg_docs_json.py
+3-1
diff --git a/‎python_modules/libraries/dagster-dg/dagster_dg_tests/utils_tests/test_sample_yaml.py
+1-1 b/‎python_modules/libraries/dagster-dg/dagster_dg_tests/utils_tests/test_sample_yaml.py
+1-1
@@ -5,10 +5,12 @@
 from dagster._core.definitions.repository_definition.repository_definition import (
     RepositoryDefinition,
 )
-from dagster._core.test_utils import instance_for_test
+from dagster._core.test_utils import ensure_dagster_tests_import, instance_for_test
 from dagster_graphql.test.utils import define_out_of_process_context, execute_dagster_graphql
 from dagster_shared.serdes.objects.package_entry import PluginObjectKey
 
+ensure_dagster_tests_import()
+
 GET_DOCS_JSON_QUERY = """
 query GetDocsJson {
   locationDocsJsonOrError(repositorySelector: {repositoryLocationName: "test_location", repositoryName: "__repository__"}) {
 
@@ -2,7 +2,7 @@
 from typing import Annotated
 
 from dagster.components import Model, Resolvable, ResolvableFieldInfo
-from dagster_dg.docs import generate_sample_yaml
+from dagster_shared.yaml_utils.sample_yaml import generate_sample_yaml
 
 
 class SampleSubModel(Model):