From f7019df477e7504e641b4cdc1c0af5f0916e7ab4 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Daniel=20Alves=20R=C3=B6sel?=
 <60182044+velocitatem@users.noreply.github.com>
Date: Fri, 12 Dec 2025 12:10:20 +0100
Subject: [PATCH] Update experiments/procesing/steps/session.py

Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
---
 experiments/procesing/steps/session.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/experiments/procesing/steps/session.py b/experiments/procesing/steps/session.py
index cc9d99e..0c5f078 100644
--- a/experiments/procesing/steps/session.py
+++ b/experiments/procesing/steps/session.py
@@ -53,7 +53,9 @@ class TemporalFeatureStep(BaseContextStep):
             (agg['total_interactions'] / agg['session_duration_sec']) * 60, 0)
 
         vel = df.set_index('ts_dt').groupby('sessionId').resample(self.velocity_window, include_groups=False).size()
-        agg = agg.merge(vel.groupby('sessionId').max().rename('max_velocity_5min'),on='sessionId', how='left').fillna({'max_velocity_5min': 0}) # warns but its a series so whatevs
+        max_velocity = vel.groupby('sessionId').max().rename('max_velocity_5min')
+        agg = agg.merge(max_velocity, on='sessionId', how='left')
+        agg['max_velocity_5min'] = agg['max_velocity_5min'].fillna(0)
         return agg