From c8eda15c56c31c2ea05240258920db9b5c18f3ba Mon Sep 17 00:00:00 2001 From: Justin Xu Date: Thu, 26 Sep 2024 02:11:48 +0100 Subject: [PATCH] Fixes polars update which modifies enable_string_cache and broke ACES dataloading --- EventStream/data/dataset_polars.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/EventStream/data/dataset_polars.py b/EventStream/data/dataset_polars.py index 9be90299..5a0dd1f9 100644 --- a/EventStream/data/dataset_polars.py +++ b/EventStream/data/dataset_polars.py @@ -33,7 +33,7 @@ from .vocabulary import Vocabulary # We need to do this so that categorical columns can be reliably used via category names. -pl.enable_string_cache(True) +pl.enable_string_cache() @dataclasses.dataclass(frozen=True)