From 58b6e2002b4fb0c4df9238cd3c70f5e95e742c06 Mon Sep 17 00:00:00 2001 From: Douglas Raillard Date: Tue, 9 Jul 2024 11:24:37 +0100 Subject: [PATCH] lisa.trace: Fix polars exception import FIX From 1.0.0, polars exception types are not exported from the top-level polars package but from polars.exceptions instead. --- lisa/trace.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lisa/trace.py b/lisa/trace.py index b0b5a3cec..5f0f42a5f 100644 --- a/lisa/trace.py +++ b/lisa/trace.py @@ -4328,7 +4328,7 @@ class _TraceCache(Loggable): try: data.sink_parquet(path, **kwargs) # Some LazyFrame cannot be sunk lazily to a parquet file - except pl.InvalidOperationError: + except polars.exceptions.InvalidOperationError: data.collect().write_parquet(path, **kwargs) else: data.to_parquet(path, **kwargs) -- GitLab