Moves over h_async to hamiton.async_driver

Its time we support this.

Decisions:
1. Did not want to keep it in hamilton.driver, that's getting too
   bloated and there's name collision
2. Moved over docs to point to the right place (as much as I could)
3. Left it backwards compatible with a warning to update the import

.ci/setup.sh

@@ -21,11 +21,6 @@ if [[ ${TASK} != "pre-commit" ]]; then
         -r requirements-test.txt
 fi
 
-if [[ ${TASK} == "async" ]]; then
-    pip install \
-        -r plugin_tests/h_async/requirements-test.txt
-fi
-
 if [[ ${TASK} == "pyspark" ]]; then
     if [[ ${OPERATING_SYSTEM} == "Linux" ]]; then
         sudo apt-get install \

.circleci/config.yml

@@ -155,27 +155,3 @@ workflows:
           name: integrations-py312
           python-version: '3.12'
           task: integrations
-      - test:
-          requires:
-            - check_for_changes
-          name: asyncio-py39
-          python-version: '3.9'
-          task: async
-      - test:
-          requires:
-            - check_for_changes
-          name: asyncio-py310
-          python-version: '3.10'
-          task: async
-      - test:
-          requires:
-            - check_for_changes
-          name: asyncio-py311
-          python-version: '3.11'
-          task: async
-      - test:
-          requires:
-            - check_for_changes
-          name: asyncio-py312
-          python-version: '3.12'
-          task: async

docs/code-comparisons/langchain_snippets/hamilton_async.py

@@ -38,9 +38,9 @@ async def joke_response(
     import hamilton_async
 
     from hamilton import base
-    from hamilton.experimental import h_async
+    from hamilton import async_driver
 
-    dr = h_async.AsyncDriver(
+    dr = async_driver.AsyncDriver(
         {},
         hamilton_async,
         result_builder=base.DictResult()

examples/async/fastapi_example.py

@@ -7,13 +7,13 @@
 from aiohttp import client_exceptions
 from hamilton_sdk import adapters
 
-from hamilton.experimental import h_async
+from hamilton import async_driver
 
 logger = logging.getLogger(__name__)
 
 # can instantiate a driver once for the life of the app:
-dr_with_tracking: h_async.AsyncDriver = None
-dr_without_tracking: h_async.AsyncDriver = None
+dr_with_tracking: async_driver.AsyncDriver = None
+dr_without_tracking: async_driver.AsyncDriver = None
 
 
 async def _tracking_server_running():
@@ -36,7 +36,7 @@ async def lifespan(app: fastapi.FastAPI):
     """
     global dr_with_tracking
     global dr_without_tracking
-    builder = h_async.Builder().with_modules(async_module)
+    builder = async_driver.Builder().with_modules(async_module)
     is_server_running = await _tracking_server_running()
     dr_without_tracking = await builder.build()
     dr_with_tracking = (
@@ -66,8 +66,6 @@ async def lifespan(app: fastapi.FastAPI):
 async def call_without_tracker(request: fastapi.Request) -> dict:
     """Handler for pipeline call -- this does not track in the Hamilton UI"""
     input_data = {"request": request}
-    # Can instantiate a driver within a request as well:
-    # dr = h_async.AsyncDriver({}, async_module, result_builder=base.DictResult())
     result = await dr_without_tracking.execute(["pipeline"], inputs=input_data)
     # dr.visualize_execution(["pipeline"], "./pipeline.dot", {"format": "png"}, inputs=input_data)
     return result
@@ -77,8 +75,6 @@ async def call_without_tracker(request: fastapi.Request) -> dict:
 async def call_with_tracker(request: fastapi.Request) -> dict:
     """Handler for pipeline call -- this does track in the Hamilton UI."""
     input_data = {"request": request}
-    # Can instantiate a driver within a request as well:
-    # dr = h_async.AsyncDriver({}, async_module, result_builder=base.DictResult())
     if dr_with_tracking is None:
         raise ValueError(
             "Tracking driver not initialized -- you must have the tracking server running at app startup to use this endpoint."

examples/feature_engineering/feature_engineering_multiple_contexts/scenario_1/fastapi_server.py

@@ -17,8 +17,7 @@
 import pandas as pd
 import pydantic
 
-from hamilton import base
-from hamilton.experimental import h_async
+from hamilton import async_driver, base
 
 app = fastapi.FastAPI()
 
@@ -56,7 +55,7 @@ def fake_model_predict(df: pd.DataFrame) -> pd.Series:
 
 # We instantiate an async driver once for the life of the app. We use the AsyncDriver here because under the hood
 # FastAPI is async. If you were using Flask, you could use the regular Hamilton driver without issue.
-dr = h_async.AsyncDriver(
+dr = async_driver.AsyncDriver(
     {},  # no config/invariant inputs in this example.
     features,  # the module that contains the common feature definitions.
     result_builder=base.SimplePythonDataFrameGraphAdapter(),