From 6cf6192e0e51260a05fa966ab2af39793c5ac8bc Mon Sep 17 00:00:00 2001 From: larsevj Date: Thu, 30 Jan 2025 10:47:16 +0100 Subject: [PATCH] Resolve polars asof warning --- src/ert/data/_measured_data.py | 5 +++-- src/ert/storage/local_ensemble.py | 5 +++-- 2 files changed, 6 insertions(+), 4 deletions(-) diff --git a/src/ert/data/_measured_data.py b/src/ert/data/_measured_data.py index 0fa7444b532..fcb6fa76a09 100644 --- a/src/ert/data/_measured_data.py +++ b/src/ert/data/_measured_data.py @@ -128,11 +128,12 @@ def _get_data( ) if "time" in pivoted: - joined = observations_for_type.join_asof( - pivoted, + joined = observations_for_type.sort(by="time").join_asof( + pivoted.sort(by="time"), by=["response_key", *response_cls.primary_key], on="time", tolerance="1s", + strategy="nearest", ) else: joined = observations_for_type.join( diff --git a/src/ert/storage/local_ensemble.py b/src/ert/storage/local_ensemble.py index b0f9d71671d..dd6a0368d53 100644 --- a/src/ert/storage/local_ensemble.py +++ b/src/ert/storage/local_ensemble.py @@ -981,11 +981,12 @@ def get_observations_and_responses( ) ) elif "time" in pivoted: - joined = observations_for_type.join_asof( - pivoted, + joined = observations_for_type.sort(by="time").join_asof( + pivoted.sort(by="time"), by=["response_key", *response_cls.primary_key], on="time", tolerance="1s", + strategy="nearest", ) else: joined = observations_for_type.join(