From dc00c0b761ed9c27d759359e2a419aed62845af5 Mon Sep 17 00:00:00 2001 From: Anatoly Myachev Date: Sat, 14 Oct 2023 01:53:56 +0200 Subject: [PATCH] fixes Signed-off-by: Anatoly Myachev --- modin/conftest.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/modin/conftest.py b/modin/conftest.py index 87cc0083792..eeeeb49f536 100644 --- a/modin/conftest.py +++ b/modin/conftest.py @@ -175,6 +175,9 @@ def __init__(self, modin_frame): def finalize(self): self._modin_frame.finalize() + def wait_computations(self): + self._modin_frame.wait_computations() + @classmethod def from_pandas(cls, df, data_cls): return cls(data_cls.from_pandas(df))