fix type

alexgiving · Nov 29, 2024 · 91f549c · 91f549c
1 parent d694bd5
commit 91f549c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lkmeans/examples/main.py b/lkmeans/examples/main.py
@@ -35,7 +35,7 @@ class ExperimentArguments(Tap):
 
 
 def get_clustering_algorithm(clustering_type: ClusteringAlgorithmType) -> Clustering:
-    clustering_map: Dict[clustering_type, Clustering] = {
+    clustering_map: Dict[ClusteringAlgorithmType, Clustering] = {
         ClusteringAlgorithmType.LKMEANS: LKMeans,
         ClusteringAlgorithmType.SOFT_SS_LKMEANS: SoftSSLKMeans,
         ClusteringAlgorithmType.HARD_SS_LKMEANS: HardSSLKMeans