From 8037e49164bb342ac1d6df529a6c1e32396277e8 Mon Sep 17 00:00:00 2001 From: kostastsa Date: Wed, 1 Nov 2023 11:53:45 +0100 Subject: [PATCH] aux --- dynamax/slds/__init__.py | 2 ++ dynamax/slds/inference_test.py | 5 +++-- 2 files changed, 5 insertions(+), 2 deletions(-) create mode 100644 dynamax/slds/__init__.py diff --git a/dynamax/slds/__init__.py b/dynamax/slds/__init__.py new file mode 100644 index 00000000..072feffb --- /dev/null +++ b/dynamax/slds/__init__.py @@ -0,0 +1,2 @@ +from dynamax.slds.inference import DiscreteParamsSLDS, LGParamsSLDS, ParamsSLDS, rbpfilter, rbpfilter_optimal +from dynamax.slds.models import SLDS diff --git a/dynamax/slds/inference_test.py b/dynamax/slds/inference_test.py index 1602c5d1..a2ff7f69 100644 --- a/dynamax/slds/inference_test.py +++ b/dynamax/slds/inference_test.py @@ -1,8 +1,9 @@ +import sys +sys.path.append(r'/Users/kostastsampourakis/Desktop/code/Python/projects/dynamax') import jax.numpy as jnp import jax.random as jr -from dynamax.slds.inference import DiscreteParamsSLDS, LGParamsSLDS, ParamsSLDS, rbpfilter, rbpfilter_optimal +from dynamax.slds import SLDS, DiscreteParamsSLDS, LGParamsSLDS, ParamsSLDS, rbpfilter, rbpfilter_optimal from functools import partial -from dynamax.slds.models import SLDS import matplotlib.pyplot as plt import mixture_kalman_filter as kflib from functools import partial