Removed import torch from header

Author: joacorapela

src/lds/inference.py

@@ -1,7 +1,6 @@
 
 import math
 import numpy as np
-import torch
 
 
 class OnlineKalmanFilter:
@@ -248,6 +247,7 @@ def filterLDS_SS_withMissingValues_torch(y, B, Q, m0, V0, Z, R):
 
     """
 
+    import torch
     if torch.any(torch.isnan(y[:, 0])) or torch.any(torch.isnan(y[:, -1])):
         raise ValueError("The first or last observation cannot contain nan")
 

src/lds/learning.py

@@ -3,7 +3,6 @@
 import time
 import numpy as np
 import scipy.optimize
-import torch
 import warnings
 import copy
 
@@ -176,6 +175,7 @@ def torch_lbfgs_optimize_SS_tracking_diagV0(y, B, sigma_a0, Qe, Z,
                                                 "R": True, "m0": True, "V0": True},
                                             disp=True):
 
+    import torch
     def log_likelihood_fn():
         V0 = torch.diag(sqrt_diag_V0**2)
         R = torch.diag(sqrt_diag_R**2)
@@ -289,6 +289,7 @@ def torch_adam_optimize_SS_tracking_diagV0(y, B, sigma_a0, Qe, Z,
                                                "V0": True},
                                           ):
 
+    import torch
     def log_likelihood_fn():
         V0 = torch.diag(sqrt_diag_V0**2)
         R = torch.diag(sqrt_diag_R**2)