move scipy rotation import to grid_functions

thoglu · Dec 27, 2023 · 10b792c · 10b792c
1 parent 6d1ca31
commit 10b792c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git 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
@@ -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