aux

probml · Nov 1, 2023 · 8037e49 · 8037e49
1 parent f2263f4
commit 8037e49
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/dynamax/slds/__init__.py 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
@@ -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