Merge pull request #441 from colecitrenbaum/main

Off by one in input driven HMM

Author: slinderman

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.

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]