Merge pull request #54 from probml/test-fix

Remove unnecessary path munging.
probml · May 17, 2022 · 76d557e · 76d557e
2 parents 25c6f68 + 157afaf
commit 76d557e
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/jsl/lds/kalman_filter_test.py b/jsl/lds/kalman_filter_test.py
@@ -1,15 +1,11 @@
-import os
-import sys
-sys.path.insert(0, os.path.abspath(os.path.dirname(__file__)))
-
 from jax import random
 from jax import numpy as jnp
 import numpy as np
 import tensorflow as tf
 import tensorflow_probability as tfp
 
 tfd = tfp.distributions
-from kalman_filter import LDS, kalman_filter
+from jsl.lds.kalman_filter import LDS, kalman_filter
 
 def tfp_filter(timesteps, A, transition_noise_scale, C, observation_noise_scale, mu0, x_hist):
     """ Perform filtering using tensorflow probability """