Add type hints to ports_allocation

skalenetwork · May 2, 2024 · 471ed0f · 471ed0f
1 parent 6791264
commit 471ed0f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/skale/schain_config/generator.py b/skale/schain_config/generator.py
@@ -19,9 +19,10 @@
 
 
 from skale.skale_manager import SkaleManager
+from skale.types.schain import SchainName
 
 
-def get_nodes_for_schain(skale: SkaleManager, name: str) -> list[int]:
+def get_nodes_for_schain(skale: SkaleManager, name: SchainName) -> list[int]:
     nodes = []
     ids = skale.schains_internal.get_node_ids_for_schain(name)
     for id_ in ids: