From 7cdc4b1a2f0ca1c183f4bb06a4d163058f0030c0 Mon Sep 17 00:00:00 2001
From: "Nikita.Ashihmin" <nikita.ashihmin@jetbrains.com>
Date: Fri, 28 Jun 2024 19:28:03 +0400
Subject: [PATCH] PY-73353 Jupyter(fix): Fix remove table output

GitOrigin-RevId: 95db9ee363524d3aa2b546fa29c06798bbdf6c2b
---
 .../plugins/notebooks/visualization/NotebookCellLines.kt        | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/notebooks/visualization/src/org/jetbrains/plugins/notebooks/visualization/NotebookCellLines.kt b/notebooks/visualization/src/org/jetbrains/plugins/notebooks/visualization/NotebookCellLines.kt
index 5a5d9ec95bb4..6b159a3a7115 100644
--- a/notebooks/visualization/src/org/jetbrains/plugins/notebooks/visualization/NotebookCellLines.kt
+++ b/notebooks/visualization/src/org/jetbrains/plugins/notebooks/visualization/NotebookCellLines.kt
@@ -65,6 +65,8 @@ interface NotebookCellLines {
 
   fun intervalsIterator(startLine: Int = 0): ListIterator<Interval>
 
+  fun getCell(line: Int) = intervals.firstOrNull { it.lines.contains(line) }
+
   val intervals: List<Interval>
 
   val intervalListeners: EventDispatcher<IntervalListener>