From e919b6ee36837dea9178c4c22f719b03445087e6 Mon Sep 17 00:00:00 2001 From: colecitrenbaum Date: Mon, 5 Jan 2026 15:18:16 -0800 Subject: [PATCH 1/2] fixed off by one in input driven HMM --- dynamax/hidden_markov_model/models/abstractions.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/dynamax/hidden_markov_model/models/abstractions.py b/dynamax/hidden_markov_model/models/abstractions.py index defc70762..1dd6b946b 100644 --- a/dynamax/hidden_markov_model/models/abstractions.py +++ b/dynamax/hidden_markov_model/models/abstractions.py @@ -295,7 +295,7 @@ def collect_suff_stats(self, PyTree of sufficient statistics for updating the transition distribution """ - return posterior.trans_probs, pytree_slice(inputs, slice(1, None)) + return posterior.trans_probs, inputs def initialize_m_step_state(self, params: ParameterSet, props:PropertySet) -> Any: """Initialize any required state for the M step. From 14cb0ce663b634cf8567156da82e475ba0a48901 Mon Sep 17 00:00:00 2001 From: colecitrenbaum Date: Mon, 5 Jan 2026 15:35:03 -0800 Subject: [PATCH 2/2] add ipython to dependencies --- pyproject.toml | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/pyproject.toml b/pyproject.toml index bd7e910fe..31891f388 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -70,7 +70,8 @@ test = [ "coverage", "pytest>=3.9", "pytest-cov", - "interrogate>=1.5.0" + "interrogate>=1.5.0", + "ipython" ] dev = [ @@ -91,7 +92,8 @@ dev = [ "coverage", "pytest>=3.9", "pytest-cov", - "interrogate>=1.5.0" + "interrogate>=1.5.0", + "ipython" ] [tool.setuptools.packages.find]