From 977b746301bc698fced1e94b0aacde7f558bd771 Mon Sep 17 00:00:00 2001
From: JamieDeMaria <jamie@dagsterlabs.com>
Date: Fri, 25 Oct 2024 10:04:31 -0400
Subject: [PATCH] celery changes for new context

---
 .../dagster-celery/example/example_code/definitions.py        | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/python_modules/libraries/dagster-celery/example/example_code/definitions.py b/python_modules/libraries/dagster-celery/example/example_code/definitions.py
index 078c35a3b9fc9..fdd39d698cd28 100644
--- a/python_modules/libraries/dagster-celery/example/example_code/definitions.py
+++ b/python_modules/libraries/dagster-celery/example/example_code/definitions.py
@@ -2,8 +2,8 @@
 import time
 
 from dagster import (
+    AssetExecutionContext,
     Definitions,
-    OpExecutionContext,
     StaticPartitionsDefinition,
     asset,
     define_asset_job,
@@ -20,7 +20,7 @@
 @asset(
     partitions_def=example_partition_def,
 )
-def partitioned_asset(context: OpExecutionContext) -> None:
+def partitioned_asset(context: AssetExecutionContext) -> None:
     """This asset is partitioned."""
     context.log.info("Preparing to greet the world!")
     time.sleep(10)