models/file.py: Update some signatures

1 files changed, 7 insertions, 7 deletions

diff --git a/kodereviewer/models/file.py b/kodereviewer/models/file.py
index cc3f8be..2e00755 100644
--- a/kodereviewer/models/file.py
+++ b/kodereviewer/models/file.py
@@ -158,21 +158,21 @@ class TreeFileModel(QAbstractItemModel):
             return QModelIndex()
         return self.createIndex(parent.row(), 0, parent)
 
-    def rowCount(self, index: QModelIndex = QModelIndex()) -> int:
+    def rowCount(self, parent: QModelIndex | QPersistentModelIndex = QModelIndex()) -> int:
         """Returns the number of rows under the given parent
 
         When the parent is valid it means that rowCount is returning
         the number of children of parent. """
 
-        parent: FileItem
-        if index.isValid():
-            parent = index.internalPointer()
+        node: FileItem
+        if parent.isValid():
+            node = parent.internalPointer()
         else:
-            parent = self.root_node
+            node = self.root_node
 
-        return len(parent.children)
+        return len(node.children)
 
-    def columnCount(self, parent: QModelIndex = QModelIndex()) -> int:
+    def columnCount(self, parent: QModelIndex | QPersistentModelIndex = QModelIndex()) -> int:
         return 1
 
     def data(self, index: QModelIndex | QPersistentModelIndex, role: int = Qt.ItemDataRole.DisplayRole) -> Any: