From 157afafd747e369f48b5e3749625388ba10a7be9 Mon Sep 17 00:00:00 2001 From: gileshd Date: Tue, 17 May 2022 23:40:00 +0100 Subject: [PATCH] Remove unnecessary path munging. --- jsl/lds/kalman_filter_test.py | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) diff --git a/jsl/lds/kalman_filter_test.py b/jsl/lds/kalman_filter_test.py index 0edb84a..79dadd1 100644 --- a/jsl/lds/kalman_filter_test.py +++ b/jsl/lds/kalman_filter_test.py @@ -1,7 +1,3 @@ -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 @@ -9,7 +5,7 @@ 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 """