cleanup

predibase · Apr 3, 2024 · de8f9b2 · de8f9b2
1 parent 988666e
commit de8f9b2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/server/lorax_server/adapters/weights.py b/server/lorax_server/adapters/weights.py
@@ -106,7 +106,7 @@ def ranks(self) -> Set[int]:
         return set(
             rank_data.rank
             for layer_data in self.data.values()
-            for rank_data in layer_data.get(LORA, []).rank_data.values() if LORA in layer_data
+            for rank_data in layer_data.get(LORA).rank_data.values() if LORA in layer_data
         )
 
     @property