From 10b792c75f732829c50a8767ae8bd9cc784d78ac Mon Sep 17 00:00:00 2001 From: thoglu Date: Wed, 27 Dec 2023 02:20:07 +0100 Subject: [PATCH] move scipy rotation import to grid_functions --- jammy_flows/helper_fns/grid_functions.py | 2 ++ jammy_flows/helper_fns/plotting/general.py | 1 - 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/jammy_flows/helper_fns/grid_functions.py b/jammy_flows/helper_fns/grid_functions.py index 43898c6..ea75d6a 100644 --- a/jammy_flows/helper_fns/grid_functions.py +++ b/jammy_flows/helper_fns/grid_functions.py @@ -1,6 +1,8 @@ import numpy import torch +from scipy.spatial.transform import Rotation as R + def find_bins(trace, percentiles=[5.0,95.0], num_bins=50, use_outlier_binning=False): if(use_outlier_binning): diff --git a/jammy_flows/helper_fns/plotting/general.py b/jammy_flows/helper_fns/plotting/general.py index 1873f3c..543ef8f 100644 --- a/jammy_flows/helper_fns/plotting/general.py +++ b/jammy_flows/helper_fns/plotting/general.py @@ -11,7 +11,6 @@ import copy #import astropy.stats -from scipy.spatial.transform import Rotation as R from .. import contours from ..grid_functions import obtain_bins_and_visualization_regions, get_pdf_on_grid, cartesian_lambert_to_spherical, spherical_to_cartesian_lambert, get_basic_gridlines