lazy import scipy

Signed-off-by: PeaBrane <[email protected]>

Author: PeaBrane

components/src/dynamo/planner/utils/perf_interpolation.py

@@ -20,7 +20,6 @@
 from typing import Optional
 
 import numpy as np
-import scipy
 
 from dynamo.runtime.logging import configure_dynamo_logging
 
@@ -80,6 +79,9 @@ def __init__(
         self.min_isl = min(self.prefill_isl)
         self.max_isl = max(self.prefill_isl)
 
+        # Lazy import scipy only when interpolation is actually needed
+        import scipy.interpolate
+
         # perform 1d interpolation
         self.ttft_interpolator = scipy.interpolate.interp1d(
             self.prefill_isl, self.prefill_ttft, kind="cubic"
@@ -151,6 +153,9 @@ def __init__(
         self.yi = np.linspace(0, max(self.y_context_length), resolution)
         self.X, self.Y = np.meshgrid(self.xi, self.yi)
 
+        # Lazy import scipy only when interpolation is actually needed
+        import scipy.interpolate
+
         # perform 2d interpolation with fallback for NaN values
         self.itl_interpolator = scipy.interpolate.griddata(
             (self.x_kv_usage, self.y_context_length),