[dagster-dataform] Fix __init__.py imports (#229)

* [dagster-dataform] Initial README

* [dagster-dataform] Root files

* [dagster-dataform] Integration modules

* [dagster-dataform] Integration unit tests

* [dagster-dataform] Add prereq for freshness checks

* [dagster-dataform] import module objects into __init__.py

* [dagster-dataform] fix import names

* [dagster-dataform] Resolve linting warnings.

* [dagster-dataform] Add # noqa for stubborn type warnings

* [dagster-dataform] Fix; remove non-existent function import

* [dagster-dataform] Modify ignore comments to silence certain pyright errors.

---------

Co-authored-by: 2705815 <Justin.Cohen2@aetna.com>

Author: Jcohen010

libraries/dagster-dataform/dagster_dataform/__init__.py

@@ -2,7 +2,6 @@
 from dagster_dataform.resources import (
     DataformRepositoryResource as DataformRepositoryResource,
     load_dataform_assets as load_dataform_assets,
-    load_dataform_asset_check_specs as load_dataform_asset_check_specs,
 )
 from dagster_dataform.dataform_polling_sensor import (
     create_dataform_workflow_invocation_sensor as create_dataform_workflow_invocation_sensor,

libraries/dagster-dataform/dagster_dataform/resources.py

@@ -148,7 +148,7 @@ def query_workflow_invocation(
 
         if not name:
             self.logger.error("No workflow invocation name available")
-            return []  # noqa
+            return []  # pyright: ignore[reportReturnType]
 
         self.logger.info(f"Querying workflow invocation: {name}")
 
@@ -163,7 +163,7 @@ def query_workflow_invocation(
         # self.logger.info(f"Found {len(response)} workflow invocation actions")
 
         # Handle the response
-        return response  # noqa
+        return response  # pyright: ignore[reportReturnType]
 
     def create_workflow_invocation(
         self, compilation_result_name: str
@@ -213,7 +213,7 @@ def load_dataform_assets(
         try:
             spec = AssetSpec(
                 key=asset.target.name,
-                kinds=["bigquery"],  # noqa
+                kinds=["bigquery"],  # pyright: ignore[reportArgumentType]
                 metadata={
                     "Project ID": asset.target.database,
                     "Dataset": asset.target.schema,

libraries/dagster-dataform/dagster_dataform_tests/test_dataform_orchestration_schedule.py

@@ -14,7 +14,7 @@ def test_dataform_orchestration_schedule_creates_schedule_and_job():
         repository_id="test-repo",
         location="us-central1",
         environment="dev",
-        client=mock_dataform_client,  # noqa
+        client=mock_dataform_client,  # pyright: ignore[reportArgumentType]
     )
 
     schedule = create_dataform_orchestration_schedule(

libraries/dagster-dataform/dagster_dataform_tests/test_dataform_polling_sensor.py

@@ -15,7 +15,7 @@ def test_dataform_polling_sensor_creates_sensor_and_job():
         repository_id="test-repo",
         location="us-central1",
         environment="dev",
-        client=mock_dataform_client,  # noqa
+        client=mock_dataform_client,  # pyright: ignore[reportArgumentType]
     )
 
     sensor = create_dataform_workflow_invocation_sensor(
@@ -36,7 +36,7 @@ def test_dataform_polling_sensor_creates_sensor_and_job_when_passed_job():
         repository_id="test-repo",
         location="us-central1",
         environment="dev",
-        client=mock_dataform_client,  # noqa
+        client=mock_dataform_client,  # pyright: ignore[reportArgumentType]
     )
 
     @dg.job