diff --git a/pocomc/flow.py b/pocomc/flow.py index ec4e00b..206c815 100644 --- a/pocomc/flow.py +++ b/pocomc/flow.py @@ -1,4 +1,4 @@ -from typing import Union, Optional, Tuple, Dict, List +from typing import Tuple import numpy as np import copy diff --git a/pocomc/mcmc.py b/pocomc/mcmc.py index 30c91c6..3ae9931 100644 --- a/pocomc/mcmc.py +++ b/pocomc/mcmc.py @@ -1,8 +1,7 @@ import numpy as np import torch -from .tools import numpy_to_torch, torch_to_numpy, flow_numpy_wrapper -from .student import fit_mvstud +from .tools import flow_numpy_wrapper @torch.no_grad() def preconditioned_pcn(state_dict: dict, diff --git a/pocomc/particles.py b/pocomc/particles.py index b96656c..08d213d 100644 --- a/pocomc/particles.py +++ b/pocomc/particles.py @@ -1,4 +1,3 @@ -from typing import Any import numpy as np class Particles: