use app.screen

Author: willmcgugan

src/textual/layouts/grid.py

@@ -60,7 +60,7 @@ def arrange(
                     table_size_columns -= 1
 
         table_size_rows = styles.grid_size_rows
-        viewport = parent.screen.size
+        viewport = parent.app.size
         keyline_style, _keyline_color = styles.keyline
         offset = (0, 0)
         gutter_spacing: Spacing | None

src/textual/screen.py

@@ -512,7 +512,7 @@ def get_maximize_widgets(maximized: Widget) -> list[Widget]:
                 else self._nodes
             ),
             size,
-            self.screen.size,
+            self.app.size,
             False,
         )
 

src/textual/widget.py

@@ -1263,7 +1263,7 @@ def _arrange(self, size: Size, optimal: bool = False) -> DockArrangeResult:
             return cached_result
 
         arrangement = self._arrangement_cache[cache_key] = arrange(
-            self, self._nodes, size, self.screen.size, optimal=optimal
+            self, self._nodes, size, self.app.size, optimal=optimal
         )
 
         return arrangement