feat: ✨ add tps transformation grid

noshita · Mar 10, 2024 · ae87df5 · ae87df5
1 parent d295a4f
commit ae87df5
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/ktch/landmark/_plot/_tps.py b/ktch/landmark/_plot/_tps.py
@@ -14,7 +14,6 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import matplotlib.pyplot as plt
 import numpy as np
 
 from .._Procrustes_analysis import _thin_plate_spline_2d, _tps_2d
@@ -46,6 +45,8 @@ def tps_grid_2d_plot(
         Matplotlib axes.
     """
 
+    import matplotlib.pyplot as plt
+
     W, c, A = _thin_plate_spline_2d(x_reference, x_target)
 
     if ax is None: