diff --git a/exo/topology/partitioning_strategy.py b/exo/topology/partitioning_strategy.py
index ab5fe9b06..47e349d64 100644
--- a/exo/topology/partitioning_strategy.py
+++ b/exo/topology/partitioning_strategy.py
@@ -1,8 +1,6 @@
 from abc import ABC, abstractmethod
-from typing import List, Optional
+from typing import List
 from dataclasses import dataclass
-from exo.inference.shard import Shard
-from exo.networking.peer_handle import PeerHandle
 from .topology import Topology
 
 # Partitions shard-space into pieces of contiguous shards, represented by floating point range [start, end) between 0 and 1