Rename check > _check (#7808)

examples/hacker_news/hacker_news/ops/id_range_for_time.py

@@ -1,6 +1,8 @@
 from datetime import datetime, timezone
 
-from dagster import Out, Output, Tuple, check, op
+from dagster import Out, Output, Tuple
+from dagster import _check as check
+from dagster import op
 
 
 def binary_search_nearest_left(get_value, start, end, min_target):

examples/hacker_news/hacker_news/resources/parquet_io_manager.py

@@ -4,7 +4,9 @@
 import pandas
 import pyspark
 
-from dagster import AssetKey, Field, IOManager, MetadataEntry, OutputContext, check, io_manager
+from dagster import AssetKey, Field, IOManager, MetadataEntry, OutputContext
+from dagster import _check as check
+from dagster import io_manager
 from dagster.seven.temp_dir import get_system_temp_directory
 
 

examples/hacker_news/hacker_news/resources/snowflake_io_manager.py

@@ -11,7 +11,7 @@
 from snowflake.sqlalchemy import URL  # pylint: disable=no-name-in-module,import-error
 from sqlalchemy import create_engine
 
-import dagster.check as check
+import dagster._check as check
 from dagster import AssetKey, IOManager, InputContext, MetadataEntry, OutputContext, io_manager
 
 

examples/hacker_news/hacker_news_tests/test_schedules.py

@@ -3,7 +3,7 @@
 
 from hacker_news.jobs.hacker_news_api_download import download_prod_job, download_staging_job
 
-import dagster.check as check
+import dagster._check as check
 from dagster import Partition
 from dagster.core.definitions import JobDefinition
 from dagster.core.execution.api import create_execution_plan

examples/hacker_news_assets/hacker_news_assets/core/assets/id_range_for_time.py

@@ -2,7 +2,9 @@
 
 from hacker_news_assets.partitions import hourly_partitions
 
-from dagster import Output, asset, check
+from dagster import Output
+from dagster import _check as check
+from dagster import asset
 
 
 def binary_search_nearest_left(get_value, start, end, min_target):

examples/hacker_news_assets/hacker_news_assets/resources/duckdb_parquet_io_manager.py

@@ -3,7 +3,9 @@
 import duckdb
 import pandas as pd
 
-from dagster import Field, check, io_manager
+from dagster import Field
+from dagster import _check as check
+from dagster import io_manager
 from dagster.seven.temp_dir import get_system_temp_directory
 
 from .parquet_io_manager import PartitionedParquetIOManager

examples/hacker_news_assets/hacker_news_assets/resources/parquet_io_manager.py

@@ -4,7 +4,9 @@
 import pandas
 import pyspark
 
-from dagster import Field, IOManager, MetadataEntry, OutputContext, check, io_manager
+from dagster import Field, IOManager, MetadataEntry, OutputContext
+from dagster import _check as check
+from dagster import io_manager
 from dagster.seven.temp_dir import get_system_temp_directory
 
 

examples/hacker_news_assets/hacker_news_assets/resources/snowflake_io_manager.py

@@ -12,7 +12,9 @@
 from snowflake.sqlalchemy import URL  # pylint: disable=no-name-in-module,import-error
 from sqlalchemy import create_engine
 
-from dagster import IOManager, InputContext, MetadataEntry, OutputContext, check, io_manager
+from dagster import IOManager, InputContext, MetadataEntry, OutputContext
+from dagster import _check as check
+from dagster import io_manager
 
 SNOWFLAKE_DATETIME_FORMAT = "%Y-%m-%d %H:%M:%S"
 

examples/modern_data_stack_assets/modern_data_stack_assets/pandas_io_manager.py

@@ -1,6 +1,8 @@
 import pandas as pd
 
-from dagster import IOManager, check, io_manager
+from dagster import IOManager
+from dagster import _check as check
+from dagster import io_manager
 
 
 class PandasIOManager(IOManager):

examples/modern_data_stack_assets/modern_data_stack_assets/setup_airbyte.py

@@ -12,7 +12,7 @@
 from dagster_airbyte import AirbyteResource
 from dagster_postgres.utils import get_conn_string
 
-from dagster import check
+import dagster._check as check
 
 from .constants import PG_DESTINATION_CONFIG, PG_SOURCE_CONFIG
 

examples/software_defined_assets/software_defined_assets/spark_weather_assets_group.py

@@ -17,7 +17,7 @@
 from pyspark.sql import DataFrame as SparkDF
 from pyspark.sql import SparkSession
 
-from dagster import AssetGroup, AssetKey, IOManager, IOManagerDefinition, check
+from dagster import AssetGroup, AssetKey, IOManager, IOManagerDefinition, _check as check
 
 # io_manager_start
 class LocalFileSystemIOManager(IOManager):

integration_tests/python_modules/dagster-k8s-test-infra/dagster_k8s_test_infra/cluster.py

@@ -14,7 +14,7 @@
 from dagster_postgres import PostgresEventLogStorage, PostgresRunStorage, PostgresScheduleStorage
 from dagster_test.test_project import build_and_tag_test_image, get_test_project_docker_image
 
-from dagster import check
+import dagster._check as check
 from dagster.cli.debug import export_run
 from dagster.core.instance import DagsterInstance, InstanceType
 from dagster.core.instance.ref import InstanceRef

integration_tests/python_modules/dagster-k8s-test-infra/dagster_k8s_test_infra/helm.py

@@ -13,7 +13,7 @@
 import yaml
 from dagster_k8s.utils import wait_for_pod
 
-from dagster import check
+import dagster._check as check
 from dagster.utils import find_free_port, git_repository_root, merge_dicts
 
 from .integration_utils import IS_BUILDKITE, check_output, get_test_namespace, image_pull_policy

integration_tests/python_modules/dagster-k8s-test-infra/dagster_k8s_test_infra/kind.py

@@ -10,7 +10,7 @@
 import kubernetes
 from kubernetes.client import Configuration
 
-from dagster import check
+import dagster._check as check
 from dagster.utils import safe_tempfile_path
 
 from .cluster import ClusterConfig

integration_tests/test_suites/k8s-integration-test-suite/tests/test_executor.py

@@ -29,7 +29,7 @@
 )
 from dagster_test.test_project.test_pipelines.repo import define_memoization_pipeline
 
-from dagster import check
+import dagster._check as check
 from dagster.core.events import DagsterEventType
 from dagster.core.storage.pipeline_run import PipelineRunStatus
 from dagster.core.storage.tags import DOCKER_IMAGE_TAG

integration_tests/test_suites/k8s-integration-test-suite/tests/test_integration.py

@@ -17,7 +17,8 @@
     get_test_project_environments_path,
 )
 
-from dagster import DagsterEventType, check
+from dagster import DagsterEventType
+from dagster import _check as check
 from dagster.core.storage.pipeline_run import PipelineRunStatus
 from dagster.core.storage.tags import DOCKER_IMAGE_TAG
 from dagster.utils import load_yaml_from_path, merge_dicts

python_modules/automation/automation/docker/cli.py

@@ -2,7 +2,7 @@
 
 import click
 
-from dagster import check
+import dagster._check as check
 
 from .dagster_docker import DagsterDockerImage
 from .ecr import ensure_ecr_login

python_modules/automation/automation/docker/dagster_docker.py

@@ -4,7 +4,7 @@
 
 import yaml
 
-from dagster import check
+import dagster._check as check
 
 from .ecr import ecr_image, get_aws_account_id, get_aws_region
 from .utils import (

python_modules/automation/automation/docker/ecr.py

@@ -1,7 +1,7 @@
 import os
 import subprocess
 
-from dagster import check
+import dagster._check as check
 
 # We default to using the ECR region here
 DEFAULT_AWS_ECR_REGION = "us-west-2"

python_modules/automation/automation/docker/image_defs.py

@@ -6,7 +6,7 @@
 
 from automation.git import git_repo_root
 
-from dagster import check
+import dagster._check as check
 
 from .dagster_docker import DagsterDockerImage
 

python_modules/automation/automation/docker/utils.py

@@ -4,7 +4,7 @@
 from datetime import datetime, timezone
 from typing import Dict, Optional
 
-from dagster import check
+import dagster._check as check
 
 
 def execute_docker_build(

python_modules/dagit/dagit/app.py

@@ -1,6 +1,7 @@
 from starlette.applications import Starlette
 
-from dagster import DagsterInstance, check
+from dagster import DagsterInstance
+from dagster import _check as check
 from dagster.cli.workspace.cli_target import get_workspace_process_context_from_kwargs
 from dagster.core.execution.compute_logs import warn_if_compute_logs_disabled
 from dagster.core.telemetry import log_workspace_stats

python_modules/dagit/dagit/cli.py

@@ -5,7 +5,7 @@
 import click
 import uvicorn
 
-from dagster import check
+import dagster._check as check
 from dagster.cli.utils import get_instance_for_service
 from dagster.cli.workspace import (
     get_workspace_process_context_from_kwargs,

python_modules/dagit/dagit/debug.py

@@ -2,7 +2,8 @@
 
 import click
 
-from dagster import DagsterInstance, check
+from dagster import DagsterInstance
+from dagster import _check as check
 from dagster.cli.debug import DebugRunPayload
 from dagster.core.workspace import WorkspaceProcessContext
 from dagster.serdes import deserialize_json_to_dagster_namedtuple

python_modules/dagit/dagit/graphql.py

@@ -20,7 +20,7 @@
 from starlette.routing import BaseRoute
 from starlette.websockets import WebSocket, WebSocketDisconnect, WebSocketState
 
-from dagster import check
+import dagster._check as check
 from dagster.seven import json
 
 

python_modules/dagit/dagit/webserver.py

@@ -25,8 +25,8 @@
 from starlette.staticfiles import StaticFiles
 from starlette.types import Message
 
+import dagster._check as check
 from dagster import __version__ as dagster_version
-from dagster import check
 from dagster.core.debug import DebugRunPayload
 from dagster.core.storage.compute_log_manager import ComputeIOType
 from dagster.core.workspace.context import BaseWorkspaceRequestContext, IWorkspaceProcessContext

python_modules/dagit/dagit_tests/stress/dag_gen.py

@@ -9,8 +9,8 @@
     OutputDefinition,
     PipelineDefinition,
     SolidDefinition,
-    check,
 )
+from dagster import _check as check
 
 
 def generate_solid(solid_id, num_inputs, num_outputs, num_cfg):

python_modules/dagster-graphql/dagster_graphql/cli.py

@@ -4,8 +4,9 @@
 import requests
 from graphql import graphql
 
+import dagster._check as check
+import dagster.seven as seven
 from dagster import __version__ as dagster_version
-from dagster import check, seven
 from dagster.cli.workspace import workspace_target_argument
 from dagster.cli.workspace.cli_target import (
     WORKSPACE_TARGET_WARNING,

python_modules/dagster-graphql/dagster_graphql/client/client.py

@@ -6,7 +6,7 @@
 from gql.transport import Transport
 from gql.transport.requests import RequestsHTTPTransport
 
-from dagster import check
+import dagster._check as check
 from dagster.core.definitions.utils import validate_tags
 from dagster.core.storage.pipeline_run import PipelineRunStatus
 from dagster.utils.backcompat import experimental_class_warning

python_modules/dagster-graphql/dagster_graphql/implementation/events.py

@@ -2,7 +2,8 @@
 
 from dagster_graphql.schema.table import GrapheneTable, GrapheneTableSchema
 
-from dagster import check, seven
+import dagster._check as check
+import dagster.seven as seven
 from dagster.core.definitions.metadata import (
     BoolMetadataValue,
     DagsterAssetMetadataValue,

python_modules/dagster-graphql/dagster_graphql/implementation/execution/__init__.py

@@ -3,7 +3,7 @@
 from graphql.execution.base import ResolveInfo
 from rx import Observable
 
-from dagster import check
+import dagster._check as check
 from dagster.core.events import DagsterEventType, EngineEventData
 from dagster.core.instance import DagsterInstance
 from dagster.core.storage.compute_log_manager import ComputeIOType

python_modules/dagster-graphql/dagster_graphql/implementation/execution/backfill.py

@@ -1,6 +1,6 @@
 import pendulum
 
-from dagster import check
+import dagster._check as check
 from dagster.core.execution.backfill import (
     BulkActionStatus,
     PartitionBackfill,

python_modules/dagster-graphql/dagster_graphql/implementation/execution/launch_execution.py

@@ -1,6 +1,6 @@
 from graphql.execution.base import ResolveInfo
 
-from dagster import check
+import dagster._check as check
 from dagster.core.execution.plan.resume_retry import ReexecutionStrategy
 from dagster.core.host_representation.selector import PipelineSelector
 from dagster.core.instance import DagsterInstance

python_modules/dagster-graphql/dagster_graphql/implementation/execution/run_lifecycle.py

@@ -1,6 +1,6 @@
 from graphql.execution.base import ResolveInfo
 
-from dagster import check
+import dagster._check as check
 from dagster.core.execution.plan.resume_retry import get_retry_steps_from_parent_run
 from dagster.core.execution.plan.state import KnownExecutionState
 from dagster.core.storage.pipeline_run import PipelineRunStatus

python_modules/dagster-graphql/dagster_graphql/implementation/external.py

@@ -2,7 +2,7 @@
 
 from graphql.execution.base import ResolveInfo
 
-from dagster import check
+import dagster._check as check
 from dagster.config.validate import validate_config_from_snap
 from dagster.core.host_representation import ExternalPipeline, PipelineSelector, RepositorySelector
 from dagster.core.workspace.context import BaseWorkspaceRequestContext

python_modules/dagster-graphql/dagster_graphql/implementation/fetch_assets.py

@@ -2,7 +2,9 @@
 
 from dagster_graphql.implementation.loader import CrossRepoAssetDependedByLoader
 
-from dagster import AssetKey, DagsterEventType, EventRecordsFilter, check, seven
+import dagster.seven as seven
+from dagster import AssetKey, DagsterEventType, EventRecordsFilter
+from dagster import _check as check
 from dagster.core.events import ASSET_EVENTS
 
 from .utils import capture_error

python_modules/dagster-graphql/dagster_graphql/implementation/fetch_instigators.py

@@ -1,4 +1,4 @@
-from dagster import check
+import dagster._check as check
 from dagster.core.definitions.run_request import InstigatorType
 from dagster.core.host_representation import InstigatorSelector
 from dagster.core.scheduler.instigation import InstigatorStatus

python_modules/dagster-graphql/dagster_graphql/implementation/fetch_partition_sets.py

@@ -1,7 +1,7 @@
 import yaml
 from graphql.execution.base import ResolveInfo
 
-from dagster import check
+import dagster._check as check
 from dagster.core.host_representation import (
     ExternalPartitionSet,
     RepositoryHandle,

python_modules/dagster-graphql/dagster_graphql/implementation/fetch_pipelines.py

@@ -1,6 +1,6 @@
 from graphql.execution.base import ResolveInfo
 
-from dagster import check
+import dagster._check as check
 from dagster.core.storage.pipeline_run import PipelineRun
 
 from .external import get_external_pipeline_or_raise, get_full_external_pipeline_or_raise

python_modules/dagster-graphql/dagster_graphql/implementation/fetch_runs.py

@@ -3,7 +3,8 @@
 
 from graphql.execution.base import ResolveInfo
 
-from dagster import AssetKey, PipelineDefinition, PipelineRunStatus, check
+from dagster import AssetKey, PipelineDefinition, PipelineRunStatus
+from dagster import _check as check
 from dagster.config.validate import validate_config
 from dagster.core.definitions import create_run_config_schema
 from dagster.core.errors import DagsterRunNotFoundError

python_modules/dagster-graphql/dagster_graphql/implementation/fetch_schedules.py

@@ -1,6 +1,6 @@
 from graphql.execution.base import ResolveInfo
 
-from dagster import check
+import dagster._check as check
 from dagster.core.definitions.run_request import InstigatorType
 from dagster.core.host_representation import PipelineSelector, RepositorySelector, ScheduleSelector
 from dagster.seven import get_current_datetime_in_utc, get_timestamp_from_utc_datetime

python_modules/dagster-graphql/dagster_graphql/implementation/fetch_sensors.py

@@ -1,6 +1,6 @@
 from graphql.execution.base import ResolveInfo
 
-from dagster import check
+import dagster._check as check
 from dagster.core.definitions.run_request import InstigatorType
 from dagster.core.host_representation import PipelineSelector, RepositorySelector, SensorSelector
 from dagster.core.scheduler.instigation import InstigatorState, SensorInstigatorData

python_modules/dagster-graphql/dagster_graphql/implementation/fetch_solids.py

@@ -1,6 +1,6 @@
 from collections import OrderedDict, defaultdict
 
-from dagster import check
+import dagster._check as check
 from dagster.core.host_representation import ExternalRepository
 
 from .utils import GraphSelector, capture_error