change to actor_cache

unionai · Jan 9, 2025 · 9b91961 · 9b91961
1 parent f2a09f2
commit 9b91961
Show file tree

Hide file tree

Showing 6 changed files with 6 additions and 6 deletions.
diff --git a/user_guide/core_concepts/actors/byoc/caching_basic.py b/user_guide/core_concepts/actors/byoc/caching_basic.py
@@ -15,7 +15,7 @@
 )
 
 
-@actor.cache
+@union.actor_cache
 def load_model(state: int) -> callable:
     sleep(4)  # simulate model loading
     return lambda value: state + value

diff --git a/user_guide/core_concepts/actors/byoc/caching_custom_object.py b/user_guide/core_concepts/actors/byoc/caching_custom_object.py
@@ -26,7 +26,7 @@ def __eq__(self, other):
         return self.state == other.state
 
 
-@actor.cache
+@union.actor_cache
 def get_state(obj: MyObj) -> int:
     sleep(2)
     return obj.state

diff --git a/user_guide/core_concepts/actors/byoc/caching_map_task.py b/user_guide/core_concepts/actors/byoc/caching_map_task.py
@@ -35,7 +35,7 @@ def create_model_state() -> union.FlyteFile:
     return model_state_path
 
 
-@actor.cache
+@union.actor_cache
 def load_model(model_state_path: union.FlyteFile) -> MyModel:
     # Simulate model loading time. This can take a long time
     # because the FlyteFile download is large, or when the  

diff --git a/user_guide/core_concepts/actors/serverless/caching_basic.py b/user_guide/core_concepts/actors/serverless/caching_basic.py
@@ -9,7 +9,7 @@
 )
 
 
-@actor.cache
+@union.actor_cache
 def load_model(state: int) -> callable:
     sleep(4)  # simulate model loading
     return lambda value: state + value

diff --git a/user_guide/core_concepts/actors/serverless/caching_custom_object.py b/user_guide/core_concepts/actors/serverless/caching_custom_object.py
@@ -20,7 +20,7 @@ def __eq__(self, other):
         return self.state == other.state
 
 
-@actor.cache
+@union.actor_cache
 def get_state(obj: MyObj) -> int:
     sleep(2)
     return obj.state

diff --git a/user_guide/core_concepts/actors/serverless/caching_map_task.py b/user_guide/core_concepts/actors/serverless/caching_map_task.py
@@ -28,7 +28,7 @@ def create_model_state() -> union.FlyteFile:
     return model_state_path
 
 
-@actor.cache
+@union.actor_cache
 def load_model(model_state_path: union.FlyteFile) -> MyModel:
     # Simulate model loading time. This can take a long time
     # because the FlyteFile download is large, or when the