From 74d56e52ff1fd459c5fbe93a3349ee8c7e319e91 Mon Sep 17 00:00:00 2001
From: Arbion Halili <99731180+ToxicPine@users.noreply.github.com>
Date: Mon, 7 Jul 2025 20:22:27 +0100
Subject: [PATCH] fix: Improve naming

---
 shared/types/states/master.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/shared/types/states/master.py b/shared/types/states/master.py
index 51659334..e1233b11 100644
--- a/shared/types/states/master.py
+++ b/shared/types/states/master.py
@@ -70,7 +70,7 @@ class MasterState(SharedState):
     cache_policy: CachePolicy[CachePolicyType]
 
 
-def get_inference_plan(
+def get_shard_assignments(
     inbox: Queue[ExternalCommand],
     outbox: Queue[ExternalCommand],
     resource_graph: ResourceGraph,